feat: 优化CDK服务器冠名逻辑与鉴权
Some checks failed
ci/woodpecker/push/my-first-workflow Pipeline failed

This commit is contained in:
xinian
2026-04-08 19:31:44 +08:00
parent 28b6386963
commit 3b35789b47
5 changed files with 248 additions and 184 deletions

View File

@@ -45,6 +45,8 @@ type ServerList struct {
OldScreen string `gorm:"comment:'服务器screen参数'" json:"old_screen"`
//到期时间ServerList
ExpireTime time.Time `gorm:"default:0;comment:'到期时间'" json:"expire_time"`
//CDK冠名到期时间ServerList
CDKExpireTime time.Time `gorm:"column:cdk_expire_time;default:0;comment:'CDK冠名到期时间'" json:"cdk_expire_time"`
}
func (s *ServerList) GetID() string {

View File

@@ -1,56 +1,92 @@
package controller
import "blazing/logic/service/common"
import (
"blazing/cool"
"blazing/logic/service/common"
"blazing/modules/player/model"
"context"
"encoding/hex"
"fmt"
"hash/crc32"
)
type Login struct {
// MAIN_LOGIN_IN 定义请求或响应数据结构。
type MAIN_LOGIN_IN struct {
Head common.TomeeHeader `cmd:"1001" struc:"skip"`
Sid []byte `struc:"[16]byte"`
}
type HeartBeat struct {
// CheakSession 校验登录session。
func (l *MAIN_LOGIN_IN) CheakSession() (bool, uint32) {
t1 := hex.EncodeToString(l.Sid)
r, err := cool.CacheManager.Get(context.Background(), fmt.Sprintf("session:%d", l.Head.UserID))
if err != nil {
return false, 0
}
if r.String() != t1 {
return false, 0
}
crc32Table := crc32.MakeTable(crc32.IEEE)
crcValue := crc32.Checksum([]byte(l.Sid), crc32Table)
cool.CacheManager.Remove(context.Background(), fmt.Sprintf("session:%d", l.Head.UserID))
return true, crcValue
}
// SimUserInfoInboundInfo 定义请求或响应数据结构。
type SimUserInfoInboundInfo struct {
Head common.TomeeHeader `cmd:"2051" struc:"skip"`
Time uint32
UserId uint32 `fieldDescription:"米米号" uint:"true" codec:"true"`
}
type KeepAlive struct {
// MoreUserInfoInboundInfo 定义请求或响应数据结构。
type MoreUserInfoInboundInfo struct {
Head common.TomeeHeader `cmd:"2052" struc:"skip"`
Time uint32
UserId uint32 `fieldDescription:"米米号" uint:"true" codec:"true"`
}
type UserInfo struct {
// AimatInboundInfo 定义请求或响应数据结构。
type AimatInboundInfo struct {
Head common.TomeeHeader `cmd:"2104" struc:"skip"`
UserID uint32
ShowNono uint8
RoomStyle uint32
ItemId uint32 `description:"物品id 射击激光 物品id为0" codec:"auto" uint:"true"`
ShootType uint32 `description:"射击类型 未知 给0" codec:"auto" uint:"true"`
Point model.Pos `description:"射击的坐标 x y" codec:"auto"`
}
type UserTalk struct {
// ChatInboundInfo 定义请求或响应数据结构。
type ChatInboundInfo struct {
Head common.TomeeHeader `cmd:"2102" struc:"skip"`
ReceiverID uint32
Reserve uint32 `json:"reserve" fieldDescription:"填充 默认值为0" uint:"true"`
MessageLen uint32 `struc:"sizeof=Message"`
Message []byte
Message string `json:"message" fieldDescription:"消息内容, 结束符为utf-8的数字0"`
}
type ChangeTitle struct {
// ChangeColorInboundInfo 定义请求或响应数据结构。
type ChangeColorInboundInfo struct {
Head common.TomeeHeader `cmd:"2063" struc:"skip"`
Title uint32
Color uint32 `codec:"color"`
}
type ChangePlayerIcon struct {
// ChangeDoodleInboundInfo 定义请求或响应数据结构。
type ChangeDoodleInboundInfo struct {
Head common.TomeeHeader `cmd:"2062" struc:"skip"`
Icon uint32
Id uint32 `codec:"id"`
Color uint32 `codec:"color"`
}
type GetMotto struct {
// ChangeNONOColorInboundInfo 定义请求或响应数据结构。
type ChangeNONOColorInboundInfo struct {
Head common.TomeeHeader `cmd:"9012" struc:"skip"`
Color uint32 `codec:"color"`
}
type GetTask struct {
// C2SDanceAction 定义请求或响应数据结构。
type C2SDanceAction struct {
Head common.TomeeHeader `cmd:"2103" struc:"skip"`
TaskLen uint32 `struc:"sizeof=TaskList"`
TaskList []uint32
Reserve uint32 `struc:"uint32,big"`
Type uint32 `struc:"uint32,big"`
}
// C2SPEOPLE_TRANSFROM 定义请求或响应数据结构。
type C2SPEOPLE_TRANSFROM struct {
Head common.TomeeHeader `cmd:"2111" struc:"skip"`
SuitID uint32 `struc:"uint32,big"`

View File

@@ -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,
"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) {

View File

@@ -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)
}
// 保存版本号

View File

@@ -15,6 +15,7 @@ type CdkController struct {
*cool.Controller
}
// init 注册用户态CDK相关Web接口。
func init() {
controller := &CdkController{
&cool.Controller{
@@ -28,14 +29,10 @@ 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不能为空"`
}
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#服务器名称不能为空"`
@@ -45,15 +42,15 @@ type DonationRedeemRes struct {
ServerID uint32 `json:"server_id"`
ServerName string `json:"server_name"`
OwnerID uint32 `json:"owner_id"`
ExpireTime time.Time `json:"expire_time"`
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
}