```
All checks were successful
ci/woodpecker/push/my-first-workflow Pipeline was successful

feat(game): 实现扭蛋系统批量物品添加功能并优化地图逻辑

- 新增ItemAddBatch方法用于批量添加物品,支持普通道具和特殊道具的分别处理
- 优化扭蛋游戏玩法中的物品添加逻辑,使用新的批量接口提升性能
- 在扭蛋机器人命令中实现完整的物品检查和批量添加流程

refactor(map): 重构地图控制器代码结构并添加注释

- 为EnterMap、LeaveMap、GetMapPlayerList等方法添加中文注释
- 统一地图相关的命名规范,如enter
This commit is contained in:
昔念
2026-04-01 20:10:29 +08:00
parent 1b6586aedc
commit 5995f0670c
14 changed files with 597 additions and 214 deletions

View File

@@ -138,30 +138,46 @@ func NewBaseSysRoleService() *BaseSysRoleService {
Model: model.NewBaseSysRole(),
ListQueryOp: &cool.QueryOp{
Where: func(ctx context.Context) [][]interface{} {
return [][]interface{}{
{"label != ?", g.Slice{"admin"}, true},
}
},
Extend: func(ctx g.Ctx, m *gdb.Model) *gdb.Model {
var (
admin = cool.GetAdmin(ctx)
userId = admin.UserId
roleIds = garray.NewIntArrayFromCopy(gconv.Ints(admin.RoleIds))
)
return [][]interface{}{
{"label != ?", g.Slice{"admin"}, true},
{"(userId=? or id in (?))", g.Slice{userId, admin.RoleIds}, !roleIds.Contains(1)},
if roleIds.Contains(1) {
return m
}
if roleIds.Len() == 0 {
return m.Where("userId", userId)
}
return m.Wheref("(userId = ? or id in (?))", userId, admin.RoleIds)
},
},
PageQueryOp: &cool.QueryOp{
KeyWordField: []string{"name", "label"},
AddOrderby: map[string]string{},
Where: func(ctx context.Context) [][]interface{} {
return [][]interface{}{
{"label != ?", g.Slice{"admin"}, true},
}
},
Extend: func(ctx g.Ctx, m *gdb.Model) *gdb.Model {
var (
admin = cool.GetAdmin(ctx)
userId = admin.UserId
roleIds = garray.NewIntArrayFromCopy(gconv.Ints(admin.RoleIds))
)
return [][]interface{}{
{"label != ?", g.Slice{"admin"}, true},
{"(userid=? or id in (?))", g.Slice{gconv.String(userId), admin.RoleIds}, !roleIds.Contains(1)},
if roleIds.Contains(1) {
return m
}
if roleIds.Len() == 0 {
return m.Where("userid", gconv.String(userId))
}
return m.Wheref("(userid = ? or id in (?))", gconv.String(userId), admin.RoleIds)
},
},
InsertParam: func(ctx context.Context) map[string]interface{} {

View File

@@ -9,10 +9,21 @@ const (
TableNameMapModel = "map_model"
)
// 模型类型常量 (1:NPC, 2:精灵)
// 模型类型常量 (0:精灵, 1:NPC)
const (
MapModelTypePet = 0
MapModelTypeNPC = 1
MapModelTypePet = 2
)
const (
MapModelDirectionRight = iota
MapModelDirectionRightDown
MapModelDirectionDown
MapModelDirectionLeftDown
MapModelDirectionLeft
MapModelDirectionLeftUp
MapModelDirectionUp
MapModelDirectionRightUp
)
// MapModelBroadcastNode 地图模型广播节点配置
@@ -20,7 +31,7 @@ type MapModelBroadcastNode struct {
TriggerID uint32 `gorm:"comment:'触发器ID'" json:"trigger_id" description:"触发器ID"`
Pos string `gorm:"type:varchar(255);default:'';comment:'位置'" json:"pos" description:"位置"`
HP int32 `gorm:"type:int;default:0;comment:'血量'" json:"hp" description:"血量"`
Direction int32 `gorm:"type:int;default:0;comment:'方向'" json:"direction" description:"方向"`
Direction int32 `gorm:"type:int;default:2;comment:'BOSS方向(0-7)'" json:"direction" description:"BOSS方向"`
}
// MapModel 地图模型配置表(NPC/宠物)
@@ -34,7 +45,7 @@ type MapModel struct {
ModelID uint32 `gorm:"not null;default:0;comment:'模型ID(NPCID或精灵ID)'" json:"model_id" description:"模型ID"`
ModelType int32 `gorm:"type:int;default:1;comment:'模型类型(1:NPC,2:精灵)'" json:"model_type" description:"模型类型"`
ModelType int32 `gorm:"type:int;default:0;comment:'模型类型(0:精灵,1:NPC)'" json:"model_type" description:"模型类型"`
ModelName string `gorm:"type:varchar(100);default:'';comment:'模型名称'" json:"model_name" description:"模型名称"`

View File

@@ -29,7 +29,7 @@ type MapNode struct {
WinBonusID int `gorm:"type:int;default:0;comment:'胜利奖励ID'" json:"win_bonus_id"`
FailBonusID int `gorm:"type:int;default:0;comment:'失败奖励ID'" json:"fail_bonus_id"`
IsBroadcast uint32 `gorm:"type:int;default:0;comment:'是否需要广播'" json:"is_broadcast"`
IsBroadcast uint32 `gorm:"type:int;default:0;comment:'广播模型ID(0表示不广播)'" json:"is_broadcast"`
TriggerPlotID uint32 `gorm:"default:0;comment:'触发剧情ID0表示无剧情'" json:"trigger_plot_id" description:"触发剧情ID"`

View File

@@ -3,6 +3,10 @@ package service
import (
"blazing/cool"
"blazing/modules/config/model"
"context"
"github.com/gogf/gf/v2/database/gdb"
"github.com/gogf/gf/v2/util/gconv"
)
type MapmodelService struct {
@@ -21,8 +25,57 @@ func NewMapmodelService() *MapmodelService {
}
}
func (s *MapmodelService) GetDataByModelId(modelid uint32) (ret *model.MapModel) {
func (s *MapmodelService) appendShinyFilter(item map[string]interface{}) map[string]interface{} {
if item == nil {
return item
}
shinyID := gconv.Int(item["shiny"])
if shinyID <= 0 {
shinyID = gconv.Int(item["shinyid"])
}
if shinyID <= 0 {
item["shiny_filter"] = nil
return item
}
item["shiny_filter"] = NewShinyService().GetShiny(shinyID)
return item
}
func (s *MapmodelService) ServiceInfo(ctx context.Context, req *cool.InfoReq) (data interface{}, err error) {
result, err := s.Service.ServiceInfo(ctx, req)
if err != nil || result == nil {
return result, err
}
record, ok := result.(gdb.Record)
if !ok || record.IsEmpty() {
return result, err
}
return s.appendShinyFilter(record.Map()), nil
}
func (s *MapmodelService) ServiceList(ctx context.Context, req *cool.ListReq) (data interface{}, err error) {
result, err := s.Service.ServiceList(ctx, req)
if err != nil || result == nil {
return result, err
}
rows, ok := result.(gdb.Result)
if !ok {
return result, err
}
list := make([]map[string]interface{}, 0, len(rows))
for _, row := range rows {
list = append(list, s.appendShinyFilter(row.Map()))
}
return list, nil
}
func (s *MapmodelService) GetDataByModelId(modelid uint32) (ret *model.MapModel) {
dbm_notenable(s.Model).Where("model_id", modelid).Scan(&ret)
return
}

View File

@@ -144,6 +144,24 @@ func (s *DictInfoService) GetMax(value int64) (max uint32) {
}
// 获取稀有精灵的光环显示
// GetMaxMap 批量获取物品最大持有上限。
func (s *DictInfoService) GetMaxMap(values ...uint32) map[uint32]uint32 {
if len(values) == 0 {
return nil
}
var ress []model.DictInfo
cool.DBM(s.Model).WhereIn("value", values).Cache(gdb.CacheOption{
Force: false,
}).Scan(&ress)
result := make(map[uint32]uint32, len(ress))
for _, item := range ress {
result[gconv.Uint32(item.Value)] = uint32(item.Ordernum)
}
return result
}
func (s *DictInfoService) GetShiny() []int {
//获取精灵的排序作为精灵的数组
m := cool.DBM(s.Model)

View File

@@ -1,9 +1,11 @@
package robot
import (
"blazing/common/data"
"blazing/common/data/xmlres"
base "blazing/modules/base/service"
config "blazing/modules/config/service"
dictservice "blazing/modules/dict/service"
"blazing/modules/player/model"
"blazing/modules/player/service"
"strings"
@@ -17,51 +19,129 @@ import (
func init() {
zero.OnCommand("扭蛋").
Handle(func(ctx *zero.Ctx) {
msgs := strings.Fields(ctx.Event.Message.String())
if len(msgs) > 1 {
count := gconv.Int(msgs[1])
if count > 10 {
count = 10
}
user := base.NewBaseSysUserService().GetQQ(ctx.Event.Sender.ID)
if user == nil {
ctx.Send("未绑定请个人中心复制token发给机器人")
return
}
itemservice := service.NewItemService(uint32(user.ID))
havs := itemservice.CheakItem(400501)
if havs < int64(count) {
ctx.Send("扭蛋币不足,当前扭蛋币数量:" + gconv.String(havs))
return
}
var buf strings.Builder
buf.WriteString("当前扭蛋币数量:" + gconv.String(havs) + "\n")
if grand.Meet(int(count), 100) {
r := config.NewPetRewardService().GetEgg()
newPet := model.GenPetInfo(int(r.MonID), int(r.DV), int(r.Nature), int(r.Effect), int(r.Lv), nil, 0)
if grand.Meet(1, 500) {
newPet.RandomByWeightShiny()
}
service.NewPetService(uint32(user.ID)).PetAdd(newPet, 0)
buf.WriteString("恭喜你获得" + xmlres.PetMAP[int(newPet.ID)].DefName + "\n")
}
items := config.NewItemService().GetEgg(int(count))
for _, item := range items {
itemservice.UPDATE(uint32(item.ItemId), int(item.ItemCnt))
buf.WriteString("恭喜你获得" + xmlres.ItemsMAP[int(item.ItemId)].Name + ":" + gconv.String(item.ItemCnt) + "\n")
}
itemservice.UPDATE(400501, int(-count))
ctx.SendChain(message.At(ctx.Event.Sender.ID), message.Reply(ctx.Event.MessageID), message.Text(buf.String()))
if len(msgs) <= 1 {
return
}
count := gconv.Int(msgs[1])
if count > 10 {
count = 10
}
if count <= 0 {
return
}
user := base.NewBaseSysUserService().GetQQ(ctx.Event.Sender.ID)
if user == nil {
ctx.Send("未绑定,请先在个人中心绑定 token")
return
}
userID := uint32(user.ID)
itemService := service.NewItemService(userID)
infoService := service.NewInfoService(userID)
playerInfo := infoService.GetLogin()
if playerInfo == nil {
ctx.Send("未创建角色,请先登录游戏")
return
}
havs := itemService.CheakItem(400501)
if havs < int64(count) {
ctx.Send("扭蛋币不足,当前扭蛋币数量:" + gconv.String(havs))
return
}
var buf strings.Builder
buf.WriteString("当前扭蛋币数量:" + gconv.String(havs) + "\n")
if grand.Meet(count, 100) {
r := config.NewPetRewardService().GetEgg()
newPet := model.GenPetInfo(int(r.MonID), int(r.DV), int(r.Nature), int(r.Effect), int(r.Lv), nil, 0)
if grand.Meet(1, 500) {
newPet.RandomByWeightShiny()
}
service.NewPetService(userID).PetAdd(newPet, 0)
buf.WriteString("恭喜你获得 " + xmlres.PetMAP[int(newPet.ID)].DefName + "\n")
}
items := config.NewItemService().GetEgg(count)
regularItems := make([]data.ItemInfo, 0, len(items))
itemIDs := make([]uint32, 0, len(items))
seenIDs := make(map[uint32]struct{}, len(items))
infoDirty := false
for _, item := range items {
switch item.ItemId {
case 1:
playerInfo.Coins += item.ItemCnt
infoDirty = true
buf.WriteString("恭喜你获得 " + xmlres.ItemsMAP[int(item.ItemId)].Name + ":" + gconv.String(item.ItemCnt) + "\n")
case 3:
playerInfo.ExpPool += item.ItemCnt
infoDirty = true
buf.WriteString("恭喜你获得 " + xmlres.ItemsMAP[int(item.ItemId)].Name + ":" + gconv.String(item.ItemCnt) + "\n")
case 5:
base.NewBaseSysUserService().UpdateGold(userID, item.ItemCnt*100)
buf.WriteString("恭喜你获得 " + xmlres.ItemsMAP[int(item.ItemId)].Name + ":" + gconv.String(item.ItemCnt) + "\n")
case 9:
playerInfo.EVPool += item.ItemCnt
infoDirty = true
buf.WriteString("恭喜你获得 " + xmlres.ItemsMAP[int(item.ItemId)].Name + ":" + gconv.String(item.ItemCnt) + "\n")
default:
regularItems = append(regularItems, item)
itemID := uint32(item.ItemId)
if _, ok := seenIDs[itemID]; ok {
continue
}
seenIDs[itemID] = struct{}{}
itemIDs = append(itemIDs, itemID)
}
}
if infoDirty {
infoService.Save(*playerInfo)
}
currentItems := itemService.CheakItemM(itemIDs...)
currentMap := make(map[uint32]int64, len(currentItems))
for _, item := range currentItems {
currentMap[item.ItemId] = item.ItemCnt
}
maxMap := dictservice.NewDictInfoService().GetMaxMap(itemIDs...)
addableItems := make([]data.ItemInfo, 0, len(regularItems))
pendingMap := make(map[uint32]int64, len(itemIDs))
for _, item := range regularItems {
itemID := uint32(item.ItemId)
itemMax := maxMap[itemID]
if itemMax == 0 {
buf.WriteString("未发放奖励:物品不存在 " + gconv.String(item.ItemId) + "\n")
continue
}
if currentMap[itemID]+pendingMap[itemID]+item.ItemCnt > int64(itemMax) {
buf.WriteString("未发放奖励:" + xmlres.ItemsMAP[int(item.ItemId)].Name + " 超出最大持有数量\n")
continue
}
pendingMap[itemID] += item.ItemCnt
addableItems = append(addableItems, item)
}
addedItems, err := itemService.AddItemsChecked(addableItems, currentMap)
if err != nil {
ctx.Send("扭蛋失败,请稍后再试")
return
}
for _, item := range addedItems {
buf.WriteString("恭喜你获得 " + xmlres.ItemsMAP[int(item.ItemId)].Name + ":" + gconv.String(item.ItemCnt) + "\n")
}
itemService.UPDATE(400501, -count)
ctx.SendChain(message.At(ctx.Event.Sender.ID), message.Reply(ctx.Event.MessageID), message.Text(buf.String()))
})
}

View File

@@ -120,7 +120,7 @@ func (s *InfoService) GetLogin() *model.PlayerInfo {
// //defer t.Service.Talk_Reset()
_, err := s.dbm_fix(s.Model).Data("last_reset_time", gtime.Now()).Update()
if err != nil {
panic(err)
cool.Logger.Error(context.TODO(), "update last_reset_time failed", s.userid, err)
}
}
if !utils.IsWEEK(tt.WeekLastResetTime) {
@@ -136,7 +136,7 @@ func (s *InfoService) GetLogin() *model.PlayerInfo {
}
_, err := s.dbm_fix(s.Model).Data("week_last_reset_time", gtime.Now()).Update()
if err != nil {
panic(err)
cool.Logger.Error(context.TODO(), "update week_last_reset_time failed", s.userid, err)
}
}
}
@@ -246,15 +246,14 @@ func (s *InfoService) Save(data model.PlayerInfo) {
for i := 0; i < 3; i++ {
_, err := s.dbm_fix(s.Model).Data("data", data).Update()
if err != nil {
if i == 2 {
//todo 待实现兜底保存,现在有可能出错
s.saveToLocalFile(&data, err)
panic(err)
}
if err == nil {
return
}
} else {
break
if i == 2 {
cool.Logger.Error(context.TODO(), "player save failed after retries, fallback to local file", data.UserID, err)
s.saveToLocalFile(&data, err)
return
}
}

View File

@@ -1,12 +1,15 @@
package service
import (
"blazing/common/data"
"blazing/cool"
"blazing/modules/player/model"
"context"
"strings"
dictservice "blazing/modules/dict/service"
"github.com/gogf/gf/v2/database/gdb"
"github.com/gogf/gf/v2/frame/g"
)
@@ -75,8 +78,151 @@ func (s *ItemService) UPDATE(id uint32, count int) error {
return nil
}
// AddUniqueItems 为一组互不重复的物品各增加 1 个
// 返回值只包含本次实际成功增加的物品 id
// AddItems 批量添加道具,返回本次实际成功添加的奖励明细(保留原始顺序)
// AddItemsChecked 写入已完成上限校验的批量道具
func (s *ItemService) AddItemsChecked(items []data.ItemInfo, currentMap map[uint32]int64) ([]data.ItemInfo, error) {
if len(items) == 0 {
return nil, nil
}
updateCounts := make(map[uint32]int64, len(items))
insertData := g.List{}
successItems := make([]data.ItemInfo, 0, len(items))
for _, item := range items {
if item.ItemId <= 0 || item.ItemCnt <= 0 {
continue
}
itemID := uint32(item.ItemId)
successItems = append(successItems, item)
if _, ok := currentMap[itemID]; ok {
updateCounts[itemID] += item.ItemCnt
continue
}
currentMap[itemID] = item.ItemCnt
insertData = append(insertData, g.Map{
"player_id": s.userid,
"item_id": itemID,
"item_cnt": item.ItemCnt,
"is_vip": cool.Config.ServerInfo.IsVip,
})
}
err := g.DB().Transaction(context.TODO(), func(ctx context.Context, tx gdb.TX) error {
if len(updateCounts) > 0 {
if err := s.batchIncrementItems(tx, updateCounts); err != nil {
return err
}
}
if len(insertData) > 0 {
if _, err := tx.Model(s.Model).Data(insertData).Insert(); err != nil {
return err
}
}
return nil
})
if err != nil {
return nil, err
}
return successItems, nil
}
func (s *ItemService) AddItems(items []data.ItemInfo) ([]data.ItemInfo, error) {
if len(items) == 0 {
return nil, nil
}
var (
itemIDs = make([]uint32, 0, len(items))
seenIDs = make(map[uint32]struct{}, len(items))
)
for _, item := range items {
if item.ItemId <= 0 || item.ItemCnt <= 0 {
continue
}
itemID := uint32(item.ItemId)
if _, ok := seenIDs[itemID]; ok {
continue
}
seenIDs[itemID] = struct{}{}
itemIDs = append(itemIDs, itemID)
}
if len(itemIDs) == 0 {
return nil, nil
}
currentItems := s.CheakItemM(itemIDs...)
currentMap := make(map[uint32]int64, len(currentItems))
for _, item := range currentItems {
currentMap[item.ItemId] = item.ItemCnt
}
maxMap := dictservice.NewDictInfoService().GetMaxMap(itemIDs...)
pendingMap := make(map[uint32]int64, len(itemIDs))
checkedItems := make([]data.ItemInfo, 0, len(items))
for _, item := range items {
if item.ItemId <= 0 || item.ItemCnt <= 0 {
continue
}
itemID := uint32(item.ItemId)
itemMax := maxMap[itemID]
if itemMax == 0 {
continue
}
if currentMap[itemID]+pendingMap[itemID]+item.ItemCnt > int64(itemMax) {
continue
}
pendingMap[itemID] += item.ItemCnt
checkedItems = append(checkedItems, item)
}
return s.AddItemsChecked(checkedItems, currentMap)
}
func (s *ItemService) batchIncrementItems(tx gdb.TX, itemCounts map[uint32]int64) error {
var (
builder strings.Builder
args = make([]any, 0, len(itemCounts)*3+2)
itemIDs = make([]any, 0, len(itemCounts))
)
builder.WriteString("UPDATE ")
builder.WriteString(s.Model.TableName())
builder.WriteString(" SET item_cnt = CASE item_id ")
for itemID, itemCnt := range itemCounts {
builder.WriteString("WHEN ? THEN item_cnt + ? ")
args = append(args, itemID, itemCnt)
itemIDs = append(itemIDs, itemID)
}
builder.WriteString("ELSE item_cnt END WHERE player_id = ? AND is_vip = ? AND item_id IN (")
for i := range itemIDs {
if i > 0 {
builder.WriteString(",")
}
builder.WriteString("?")
}
builder.WriteString(")")
args = append(args, s.userid, cool.Config.ServerInfo.IsVip)
args = append(args, itemIDs...)
_, err := tx.Exec(builder.String(), args...)
return err
}
func (s *ItemService) AddUniqueItems(ids []uint32) ([]uint32, error) {
if len(ids) == 0 {
return nil, nil

View File

@@ -3,6 +3,7 @@ package service
import (
"blazing/cool"
"blazing/modules/player/model"
"context"
"github.com/pointernil/bitset32"
)
@@ -58,7 +59,7 @@ func (s *TaskService) Exec(id uint32, t func(*model.Task) bool) {
gg.TaskID = id
_, err := m1.Save(gg)
if err != nil {
panic(err)
cool.Logger.Error(context.TODO(), "task save failed", s.userid, id, err)
}
}