feat: 积分质押功能

internal/model/mg_servers_model.go

@@ -0,0 +1,57 @@
+package model
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
+
+var _ MgServersModel = (*customMgServersModel)(nil)
+
+type (
+	// MgServersModel is an interface to be customized, add more methods here,
+	// and implement the added methods in customMgServersModel.
+	MgServersModel interface {
+		mgServersModel
+		withSession(session sqlx.Session) MgServersModel
+		GetGameServer(ctx context.Context, agentId, serverId int64) (*GameServer, error)
+	}
+
+	customMgServersModel struct {
+		*defaultMgServersModel
+	}
+
+	GameServer struct {
+		AgentID        int    `db:"agent_id"`         // 代理ID
+		ServerID       int    `db:"server_id"`        // 服务器ID
+		WebPort        int    `db:"web_port"`         // web访问端口
+		ServerIP       string `db:"server_ip"`        // 内网IP
+		PublicServerIP string `db:"public_server_ip"` // 公网IP
+	}
+)
+
+func (m *customMgServersModel) GetGameServer(ctx context.Context, agentId, serverId int64) (*GameServer, error) {
+	query := fmt.Sprintf("SELECT g.agent_id, g.server_id, g.web_port, s.ip AS server_ip, s.public_ip AS public_server_ip FROM ms_game_server AS g LEFT JOIN mg_servers AS s ON s.id = g.server_server_id WHERE g.server_id = ? AND g.agent_id = ?")
+	var result GameServer
+	err := m.conn.QueryRowCtx(ctx, &result, query, serverId, agentId)
+	switch {
+	case err == nil:
+		return &result, nil
+	case errors.Is(err, sqlx.ErrNotFound):
+		return nil, ErrNotFound
+	default:
+		return nil, err
+	}
+}
+
+// NewMgServersModel returns a model for the database table.
+func NewMgServersModel(conn sqlx.SqlConn) MgServersModel {
+	return &customMgServersModel{
+		defaultMgServersModel: newMgServersModel(conn),
+	}
+}
+
+func (m *customMgServersModel) withSession(session sqlx.Session) MgServersModel {
+	return NewMgServersModel(sqlx.NewSqlConnFromSession(session))
+}