feat: 优化CDK服务器冠名逻辑与鉴权

2026-04-08 19:31:44 +08:00
parent 28b6386963
commit 3b35789b47
5 changed files with 248 additions and 184 deletions
--- a/modules/config/service/cdk.go
+++ b/modules/config/service/cdk.go
@@ -160,7 +160,16 @@ func (s *CdkService) Set(id string) bool {
 	return true
 }

-func (s *CdkService) UseServerNamingCDK(ctx context.Context, code string, ownerID, serverID uint32, serverName string) (*model.ServerList, error) {
+type ServerNamingCDKResult struct {
+	ServerID         uint32
+	ServerName       string
+	OwnerID          uint32
+	ServerExpireTime time.Time
+	CDKExpireTime    time.Time
+}
+
+// UseServerNamingCDK 使用服务器冠名类型CDK，并原子化更新服务器归属和到期时间。
+func (s *CdkService) UseServerNamingCDK(ctx context.Context, code string, ownerID, serverID uint32, serverName string) (*ServerNamingCDKResult, error) {
 	if ctx == nil {
 		ctx = context.TODO()
 	}
@@ -190,7 +199,10 @@ func (s *CdkService) UseServerNamingCDK(ctx context.Context, code string, ownerI
 		if err := tx.Model(model.NewServerList()).Where("online_id", serverID).Scan(&server); err != nil {
 			return err
 		}
-		if !serverService.CanUseDonationName(server, now) {
+		if server.OnlineID == 0 {
+			return gerror.New("服务器不存在")
+		}
+		if server.Owner != ownerID && !serverService.CanUseDonationName(server, now) {
 			return gerror.New("服务器不可冠名")
 		}

@@ -203,19 +215,23 @@ func (s *CdkService) UseServerNamingCDK(ctx context.Context, code string, ownerI
 			return gerror.New("cdk已被使用")
 		}

-		baseTime := now
-		if server.ExpireTime.After(now) {
-			baseTime = server.ExpireTime
-		}
 		updated = server
 		updated.Name = serverName
 		updated.Owner = ownerID
-		updated.ExpireTime = baseTime.AddDate(0, 1, 0)
+		if server.Owner != ownerID {
+			updated.CDKExpireTime = now.AddDate(0, 1, 0)
+		} else {
+			baseTime := server.CDKExpireTime
+			if baseTime.IsZero() {
+				baseTime = now
+			}
+			updated.CDKExpireTime = baseTime.AddDate(0, 1, 0)
+		}

 		_, err = tx.Model(model.NewServerList()).Where("online_id", serverID).Data(g.Map{
-			"name":        updated.Name,
-			"owner":       updated.Owner,
-			"expire_time": updated.ExpireTime,
+			"name":            updated.Name,
+			"owner":           updated.Owner,
+			"cdk_expire_time": updated.CDKExpireTime,
 		}).Update()
 		return err
 	})
@@ -225,7 +241,13 @@ func (s *CdkService) UseServerNamingCDK(ctx context.Context, code string, ownerI

 	g.DB(s.Model.GroupName()).GetCore().ClearCache(context.TODO(), s.Model.TableName())
 	g.DB(model.NewServerList().GroupName()).GetCore().ClearCache(context.TODO(), model.NewServerList().TableName())
-	return &updated, nil
+	return &ServerNamingCDKResult{
+		ServerID:         updated.OnlineID,
+		ServerName:       updated.Name,
+		OwnerID:          updated.Owner,
+		ServerExpireTime: updated.ExpireTime,
+		CDKExpireTime:    updated.CDKExpireTime,
+	}, nil
 }

 func (s *CdkService) BatchGenerate(ctx context.Context, count int) (interface{}, error) {
--- a/modules/config/service/server.go
+++ b/modules/config/service/server.go
@@ -131,9 +131,10 @@ func (s *ServerService) GetServerID(OnlineID uint32) model.ServerList {

 }

+// GetDonationAvailableServerIDs 返回当前可被冠名占用的服务器ID列表。
 func (s *ServerService) GetDonationAvailableServerIDs() []uint32 {
 	now := time.Now()
-	builder := dbm_nocache_noenable(s.Model).Builder().Where("owner", 0).WhereOr("expire_time <= ?", now)
+	builder := dbm_nocache_noenable(s.Model).Builder().Where("owner", 0).WhereOr("cdk_expire_time <= ?", now)

 	var rows []struct {
 		OnlineID uint32 `json:"online_id"`
@@ -150,6 +151,7 @@ func (s *ServerService) GetDonationAvailableServerIDs() []uint32 {
 	return ids
 }

+// CanUseDonationName 校验目标服务器在当前时间点是否允许被冠名。
 func (s *ServerService) CanUseDonationName(server model.ServerList, now time.Time) bool {
 	if server.OnlineID == 0 {
 		return false
@@ -157,7 +159,7 @@ func (s *ServerService) CanUseDonationName(server model.ServerList, now time.Tim
 	if server.Owner == 0 {
 		return true
 	}
-	return !server.ExpireTime.After(now)
+	return !server.CDKExpireTime.After(now)
 }

 // 保存版本号
--- a/modules/player/controller/app/cdk.go
+++ b/modules/player/controller/app/cdk.go
@@ -15,6 +15,7 @@ type CdkController struct {
 	*cool.Controller
 }

+// init 注册用户态CDK相关Web接口。
 func init() {
 	controller := &CdkController{
 		&cool.Controller{
@@ -27,33 +28,29 @@ func init() {
 }

 type DonationServerListReq struct {
-	g.Meta  `path:"/donation/serverIds" method:"GET"`
-	UserID  uint32 `json:"user_id" v:"required|min:1#用户ID不能为空|用户ID非法"`
-	Session string `json:"session" v:"required#session不能为空"`
+	g.Meta `path:"/donation/serverIds" method:"GET"`
 }

 type DonationRedeemReq struct {
 	g.Meta     `path:"/donation/redeem" method:"POST"`
-	UserID     uint32 `json:"user_id" v:"required|min:1#用户ID不能为空|用户ID非法"`
-	Session    string `json:"session" v:"required#session不能为空"`
 	CDKCode    string `json:"cdk_code" v:"required#CDK不能为空"`
 	ServerID   uint32 `json:"server_id" v:"required|min:1#服务器ID不能为空|服务器ID非法"`
 	ServerName string `json:"server_name" v:"required#服务器名称不能为空"`
 }

 type DonationRedeemRes struct {
-	ServerID   uint32    `json:"server_id"`
-	ServerName string    `json:"server_name"`
-	OwnerID    uint32    `json:"owner_id"`
-	ExpireTime time.Time `json:"expire_time"`
+	ServerID         uint32    `json:"server_id"`
+	ServerName       string    `json:"server_name"`
+	OwnerID          uint32    `json:"owner_id"`
+	ServerExpireTime time.Time `json:"server_expire_time"`
+	CDKExpireTime    time.Time `json:"cdk_expire_time"`
 }

+// DonationServerIDs 查询当前可用于服务器绑定CDK的服务器ID列表。
 func (c *CdkController) DonationServerIDs(ctx context.Context, req *DonationServerListReq) (res *cool.BaseRes, err error) {
-	if err = g.Validator().Data(req).Run(ctx); err != nil {
-		return cool.Fail(err.Error()), nil
-	}
-	if err = validateGameSession(req.UserID, req.Session); err != nil {
-		return cool.Fail(err.Error()), nil
+	admin := cool.GetAdmin(ctx)
+	if admin == nil || admin.UserId == 0 {
+		return cool.Fail("未登录或登录已失效"), nil
 	}

 	return cool.Ok(g.Map{
@@ -61,13 +58,17 @@ func (c *CdkController) DonationServerIDs(ctx context.Context, req *DonationServ
 	}), nil
 }

+// DonationRedeem 兑换服务器绑定类型CDK并完成冠名写入。
 func (c *CdkController) DonationRedeem(ctx context.Context, req *DonationRedeemReq) (res *cool.BaseRes, err error) {
 	if err = g.Validator().Data(req).Run(ctx); err != nil {
 		return cool.Fail(err.Error()), nil
 	}
-	if err = validateGameSession(req.UserID, req.Session); err != nil {
-		return cool.Fail(err.Error()), nil
+
+	admin := cool.GetAdmin(ctx)
+	if admin == nil || admin.UserId == 0 {
+		return cool.Fail("未登录或登录已失效"), nil
 	}
+	ownerID := uint32(admin.UserId)

 	cdkCode := strings.TrimSpace(req.CDKCode)
 	if cdkCode == "" {
@@ -88,28 +89,29 @@ func (c *CdkController) DonationRedeem(ctx context.Context, req *DonationRedeemR
 	if cdkInfo.CDKType != configservice.CDKTypeServerNaming {
 		return cool.Fail("CDK类型不匹配"), nil
 	}
-	if cdkInfo.BindUserId != 0 && cdkInfo.BindUserId != req.UserID {
+	if cdkInfo.BindUserId != 0 && cdkInfo.BindUserId != ownerID {
 		return cool.Fail("CDK已绑定其他用户"), nil
 	}
 	if !cdkInfo.ValidEndTime.IsZero() && cdkInfo.ValidEndTime.Before(time.Now()) {
 		return cool.Fail("CDK已过期"), nil
 	}

-	playerCdkService := playerservice.NewCdkService(req.UserID)
+	playerCdkService := playerservice.NewCdkService(ownerID)
 	if !playerCdkService.CanGet(uint32(cdkInfo.ID)) {
 		return cool.Fail("CDK已领取"), nil
 	}

-	serverInfo, useErr := cdkService.UseServerNamingCDK(ctx, cdkCode, req.UserID, req.ServerID, serverName)
+	serverInfo, useErr := cdkService.UseServerNamingCDK(ctx, cdkCode, ownerID, req.ServerID, serverName)
 	if useErr != nil {
 		return cool.Fail(useErr.Error()), nil
 	}

 	playerCdkService.Log(uint32(cdkInfo.ID))
 	return cool.Ok(&DonationRedeemRes{
-		ServerID:   serverInfo.OnlineID,
-		ServerName: serverInfo.Name,
-		OwnerID:    serverInfo.Owner,
-		ExpireTime: serverInfo.ExpireTime,
+		ServerID:         serverInfo.ServerID,
+		ServerName:       serverInfo.ServerName,
+		OwnerID:          serverInfo.OwnerID,
+		ServerExpireTime: serverInfo.ServerExpireTime,
+		CDKExpireTime:    serverInfo.CDKExpireTime,
 	}), nil
 }