game.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971
  1. package redis
  2. import (
  3. "errors"
  4. "f1-game/internal/enum"
  5. nameRedis "f1-game/internal/name/redis"
  6. "f1-game/internal/pb"
  7. "fmt"
  8. "slices"
  9. cmapStructure "github.com/cherry-game/cherry/extend/mapstructure"
  10. cstring "github.com/cherry-game/cherry/extend/string"
  11. clog "github.com/cherry-game/cherry/logger"
  12. cprofile "github.com/cherry-game/cherry/profile"
  13. "github.com/go-redis/redis/v8"
  14. jsoniter "github.com/json-iterator/go"
  15. )
  16. type (
  17. game struct {
  18. *client
  19. playerDataKey string // 玩家数据key
  20. playerTeamKey string // 玩家部队key
  21. playerHeroKey string // 玩家英雄key
  22. battleReportGroupKey string // 战报组key
  23. battleReportKey string // 战报key
  24. settingKey string // 玩家设置
  25. mapExchangeLogKey string // 地图交易所日志key
  26. }
  27. )
  28. var notNilFunc = func(v any) bool { return v != nil }
  29. func newGame() *game {
  30. return &game{}
  31. }
  32. func (p *game) setKey() {
  33. prefix := fmt.Sprintf("%s-game:", cprofile.Env())
  34. p.playerDataKey = prefix + "player_data:%d" // type = hash, redisKey = player_data:<playerID> value = playerData
  35. p.playerTeamKey = prefix + "player_team:%d" // type = hash, redisKey = player_team:<playerID> value = pb.BattleTeam
  36. p.playerHeroKey = prefix + "player_hero:%d" // type = hash, redisKey = player_hero:<playerID> value = pb.PlayerHero
  37. p.battleReportGroupKey = prefix + "battle_report_group:%s" // type = list, redisKey = battle_report_group:<groupID> value = reportID 列表
  38. p.battleReportKey = prefix + "battle_report:%s" // Key = battle_report:<reportID> value = pb.Report
  39. p.settingKey = prefix + "setting:%d" // type = hash, redisKey = setting:<playerID> value = setting
  40. p.mapExchangeLogKey = prefix + "map_exchange_log:%s:%d:%d" // type = list, redisKey = map_exchange_log:<mapNodeID>:<playerID>:<logType> value = pb.MapExchangeLog
  41. }
  42. func (p *game) ConfigID() string {
  43. return "game_redis_id"
  44. }
  45. func (p *game) load(client *client) {
  46. p.client = client
  47. p.setKey()
  48. }
  49. func (p *game) GetPlayerDataHash(playerID int64, fields ...string) map[string]any {
  50. if len(fields) == 0 {
  51. return nil
  52. }
  53. key := p.buildPlayerDataKey(playerID)
  54. result, err := p.HMGet(p.Context(), key, fields...).Result()
  55. if err != nil {
  56. clog.Warnf("[GetPlayerData] error: %v", err)
  57. return nil
  58. }
  59. maps := make(map[string]any, len(fields))
  60. for i := range len(fields) {
  61. maps[fields[i]] = result[i]
  62. }
  63. return maps
  64. }
  65. // GetPlayerData 获取玩家数据
  66. func (p *game) GetPlayerData(playerID int64, fields ...string) *pb.PlayerData {
  67. dataMaps := p.GetPlayerDataHash(playerID, fields...)
  68. if dataMaps == nil {
  69. return nil
  70. }
  71. var playerData pb.PlayerData
  72. if err := cmapStructure.WeakDecode(dataMaps, &playerData); err != nil {
  73. clog.Errorf("[GetPlayerData] decode error: %v", err)
  74. return nil
  75. }
  76. return &playerData
  77. }
  78. // GetChatPlayer 获取玩家聊天数据信息
  79. func (p *game) GetChatPlayer(playerID int64) *pb.ChatPlayer {
  80. dataMaps := p.GetPlayerDataHash(playerID, nameRedis.PlayerChatFields...)
  81. var chatPlayer pb.ChatPlayer
  82. if err := cmapStructure.WeakDecode(dataMaps, &chatPlayer); err != nil {
  83. clog.Errorf("[GetChatPlayer] decode error: %v", err)
  84. return nil
  85. }
  86. return &chatPlayer
  87. }
  88. // GetPlayerDataList 批量获取玩家数据列表
  89. func (p *game) GetPlayerDataList(playerIDS []int64, fields ...string) []*pb.PlayerData {
  90. if len(fields) == 0 {
  91. return nil
  92. }
  93. pipeline := p.Pipeline()
  94. defer pipeline.Close()
  95. cmds := make([]*redis.SliceCmd, 0, len(playerIDS))
  96. for _, playerID := range playerIDS {
  97. key := p.buildPlayerDataKey(playerID)
  98. cmds = append(cmds, pipeline.HMGet(p.Context(), key, fields...))
  99. }
  100. if _, err := pipeline.Exec(p.Context()); err != nil {
  101. clog.Errorf("[GetPlayerDataMap] error: %v", err)
  102. return nil
  103. }
  104. players := make([]*pb.PlayerData, 0, len(playerIDS))
  105. for i, playerID := range playerIDS {
  106. result, err := cmds[i].Result()
  107. if err != nil {
  108. continue
  109. }
  110. maps := make(map[string]any, len(fields))
  111. for i := range len(fields) {
  112. maps[fields[i]] = result[i]
  113. }
  114. // 添加玩家 ID 字段
  115. maps[nameRedis.Player_PlayerID] = playerID
  116. var playerData pb.PlayerData
  117. if err := cmapStructure.WeakDecode(maps, &playerData); err != nil {
  118. continue
  119. }
  120. players = append(players, &playerData)
  121. }
  122. return players
  123. }
  124. // 获取玩家数据(聊天玩家基础信息)
  125. func (p *game) GetChatPlayerMap(playerIDs ...int64) map[int64]*pb.ChatPlayer {
  126. if len(playerIDs) == 0 {
  127. return nil
  128. }
  129. fields := nameRedis.PlayerChatFields
  130. pipeline := p.Pipeline()
  131. defer pipeline.Close()
  132. cmds := make([]*redis.SliceCmd, 0, len(playerIDs))
  133. for _, playerID := range playerIDs {
  134. key := p.buildPlayerDataKey(playerID)
  135. cmds = append(cmds, pipeline.HMGet(p.Context(), key, fields...))
  136. }
  137. if _, err := pipeline.Exec(p.Context()); err != nil {
  138. clog.Errorf("[GetPlayerDataMap] error: %v", err)
  139. return nil
  140. }
  141. players := make(map[int64]*pb.ChatPlayer, len(playerIDs))
  142. for i, playerID := range playerIDs {
  143. result, err := cmds[i].Result()
  144. if err != nil {
  145. continue
  146. }
  147. // hmget 的机制:当 key 不存在时,返回一个全为 nil 的 slice
  148. if !slices.ContainsFunc(result, notNilFunc) {
  149. continue
  150. }
  151. maps := make(map[string]any, len(fields))
  152. for i := range len(fields) {
  153. maps[fields[i]] = result[i]
  154. }
  155. // 添加玩家 ID 字段
  156. maps[nameRedis.Player_PlayerID] = playerID
  157. var chatPlayer pb.ChatPlayer
  158. if err := cmapStructure.WeakDecode(maps, &chatPlayer); err != nil {
  159. continue
  160. }
  161. players[playerID] = &chatPlayer
  162. }
  163. return players
  164. }
  165. func (p *game) getPlayerDataMap(pipeline redis.Pipeliner, playerIDs []int64, fields ...string) map[int64]*pb.PlayerData {
  166. if len(fields) == 0 {
  167. return nil
  168. }
  169. cmds := make([]*redis.SliceCmd, 0, len(playerIDs))
  170. for _, playerID := range playerIDs {
  171. key := p.buildPlayerDataKey(playerID)
  172. cmds = append(cmds, pipeline.HMGet(p.Context(), key, fields...))
  173. }
  174. if _, err := pipeline.Exec(p.Context()); err != nil {
  175. clog.Errorf("[GetPlayerDataMap] error: %v", err)
  176. return nil
  177. }
  178. players := make(map[int64]*pb.PlayerData, len(playerIDs))
  179. for i, playerID := range playerIDs {
  180. result, err := cmds[i].Result()
  181. if err != nil {
  182. continue
  183. }
  184. // hmget 的机制:当 key 不存在时,返回一个全为 nil 的 slice
  185. if !slices.ContainsFunc(result, notNilFunc) {
  186. continue
  187. }
  188. maps := make(map[string]any, len(fields))
  189. for i := range len(fields) {
  190. maps[fields[i]] = result[i]
  191. }
  192. // 添加玩家 ID 字段
  193. maps[nameRedis.Player_PlayerID] = playerID
  194. var playerData pb.PlayerData
  195. if err := cmapStructure.WeakDecode(maps, &playerData); err != nil {
  196. continue
  197. }
  198. players[playerID] = &playerData
  199. }
  200. return players
  201. }
  202. func (p *game) GetPlayerDataMap(playerIDs []int64, fields ...string) map[int64]*pb.PlayerData {
  203. pipeline := p.Pipeline()
  204. defer pipeline.Close()
  205. return p.getPlayerDataMap(pipeline, playerIDs, fields...)
  206. }
  207. // SavePlayerDataFields 保存玩家数据
  208. func (p *game) SavePlayerDataFields(playerID int64, fields map[string]any) bool {
  209. if len(fields) == 0 {
  210. return false
  211. }
  212. key := p.buildPlayerDataKey(playerID)
  213. _, err := p.HSet(p.Context(), key, fields).Result()
  214. if err != nil {
  215. clog.Errorf("[SavePlayerDataFields] error: %v", err)
  216. return false
  217. }
  218. return true
  219. }
  220. // 更新玩家的联盟信息,支持多个
  221. func (p *game) UpdatePlayerDataLeagueInfo(playerLeagueInfoList ...*pb.PlayerLeagueInfo) {
  222. if len(playerLeagueInfoList) == 0 {
  223. return
  224. }
  225. // 批量修改玩家联盟ID和名称字段
  226. pipeline := p.Pipeline()
  227. defer pipeline.Close()
  228. cmds := make([]*redis.BoolCmd, 0, len(playerLeagueInfoList))
  229. for _, playerLeagueInfo := range playerLeagueInfoList {
  230. key := p.buildPlayerDataKey(playerLeagueInfo.PlayerID)
  231. cmds = append(cmds, pipeline.HMSet(p.Context(), key, map[string]any{
  232. nameRedis.Player_LeagueID: playerLeagueInfo.LeagueID,
  233. nameRedis.Player_LeagueName: playerLeagueInfo.LeagueName,
  234. nameRedis.Player_LeagueAbbName: playerLeagueInfo.LeagueAbbName,
  235. nameRedis.Player_LeagueJob: playerLeagueInfo.LeagueJob,
  236. }))
  237. }
  238. if _, err := pipeline.Exec(p.Context()); err != nil {
  239. clog.Errorf("[UpdatePlayerDataLeagueInfo] error: %v", err)
  240. return
  241. }
  242. }
  243. // UpdatePlayerDataField 更新玩家数据
  244. func (p *game) UpdatePlayerDataField(playerID int64, field string, value any) bool {
  245. key := p.buildPlayerDataKey(playerID)
  246. _, err := p.HSet(p.Context(), key, field, value).Result()
  247. if err != nil {
  248. clog.Errorf("[UpdatePlayerDataField] error: %v", err)
  249. return false
  250. }
  251. return true
  252. }
  253. // DeletePlayerDataFields 删除玩家数据字段
  254. func (p *game) DeletePlayerDataFields(playerID int64, fields ...string) bool {
  255. if len(fields) == 0 {
  256. return false
  257. }
  258. key := p.buildPlayerDataKey(playerID)
  259. _, err := p.HDel(p.Context(), key, fields...).Result()
  260. if err != nil {
  261. clog.Errorf("[DeletePlayerDataFields] error: %v", err)
  262. return false
  263. }
  264. return true
  265. }
  266. // 获取部队信息
  267. func (p *game) GetPlayerTeamsByID(playerID int64, teamIDList ...int32) map[int32]*pb.BattleTeam {
  268. if len(teamIDList) == 0 {
  269. return nil
  270. }
  271. key := p.buildPlayerTeamKey(playerID)
  272. fields := make([]string, 0, len(teamIDList))
  273. for _, teamID := range teamIDList {
  274. fields = append(fields, cstring.ToString(teamID))
  275. }
  276. result, err := p.HMGet(p.Context(), key, fields...).Result()
  277. if err != nil {
  278. if !errors.Is(err, redis.Nil) {
  279. clog.Errorf("[GetPlayerTeamsByID] playerID = %d, error: %v", playerID, err)
  280. }
  281. return nil
  282. }
  283. ret := map[int32]*pb.BattleTeam{}
  284. for _, teamStr := range result {
  285. if teamStr == nil {
  286. continue
  287. }
  288. team := &pb.BattleTeam{}
  289. if err := jsoniter.UnmarshalFromString(teamStr.(string), team); err != nil {
  290. clog.Errorf("[GetPlayerTeamsByID] playerID = %d, unmarshal error: %v", playerID, err)
  291. return nil
  292. }
  293. ret[team.TeamID] = team
  294. }
  295. return ret
  296. }
  297. // 获取部队信息
  298. func (p *game) GetPlayerTeamByID(playerID int64, teamID int32) (*pb.BattleTeam, bool) {
  299. m := p.GetPlayerTeamsByID(playerID, teamID)
  300. if m == nil {
  301. return nil, false
  302. }
  303. team, ok := m[teamID]
  304. return team, ok
  305. }
  306. // 更新玩家队伍
  307. func (p *game) UpdatePlayerTeams(playerID int64, battleTeams ...*pb.BattleTeam) {
  308. if len(battleTeams) == 0 {
  309. return
  310. }
  311. key := p.buildPlayerTeamKey(playerID)
  312. values := make([]any, 0, len(battleTeams)*2)
  313. for _, team := range battleTeams {
  314. values = append(values, cstring.ToString(team.TeamID))
  315. jsonStr, err := jsoniter.MarshalToString(team)
  316. if err != nil {
  317. clog.Errorf("[UpdatePlayerTeams] player = %d, marshal error: %v", playerID, err)
  318. return
  319. }
  320. values = append(values, jsonStr)
  321. }
  322. err := p.HSet(p.Context(), key, values).Err()
  323. if err != nil {
  324. clog.Errorf("[UpdatePlayerTeams] player = %d, error: %v", playerID, err)
  325. }
  326. }
  327. // 获取玩家英雄
  328. func (p *game) GetPlayerHeros(playerID int64, heroIDList ...int64) map[int64]*pb.PlayerHero {
  329. if len(heroIDList) == 0 {
  330. return nil
  331. }
  332. key := p.buildPlayerHeroKey(playerID)
  333. fields := make([]string, 0, len(heroIDList))
  334. for _, heroID := range heroIDList {
  335. fields = append(fields, cstring.ToString(heroID))
  336. }
  337. result, err := p.HMGet(p.Context(), key, fields...).Result()
  338. if err != nil {
  339. if !errors.Is(err, redis.Nil) {
  340. clog.Errorf("[GetPlayerHeros] playerID = %d, error: %v", playerID, err)
  341. }
  342. return nil
  343. }
  344. ret := map[int64]*pb.PlayerHero{}
  345. for _, playerHeroStr := range result {
  346. if playerHeroStr == nil {
  347. continue
  348. }
  349. playerHero := &pb.PlayerHero{}
  350. if err := jsoniter.UnmarshalFromString(playerHeroStr.(string), playerHero); err != nil {
  351. clog.Errorf("[GetPlayerHeros] playerID = %d, unmarshal error: %v", playerID, err)
  352. return nil
  353. }
  354. ret[playerHero.Id] = playerHero
  355. }
  356. return ret
  357. }
  358. // 更新玩家英雄信息
  359. func (p *game) UpdatePlayerHeros(playerID int64, heroList ...*pb.PlayerHero) {
  360. if len(heroList) == 0 {
  361. return
  362. }
  363. key := p.buildPlayerHeroKey(playerID)
  364. values := make([]any, 0, len(heroList)*2)
  365. for _, hero := range heroList {
  366. values = append(values, cstring.ToString(hero.Id))
  367. jsonStr, err := jsoniter.MarshalToString(hero)
  368. if err != nil {
  369. clog.Errorf("[UpdatePlayerHeros] player = %d, marshal error: %v", playerID, err)
  370. return
  371. }
  372. values = append(values, jsonStr)
  373. }
  374. err := p.HSet(p.Context(), key, values).Err()
  375. if err != nil {
  376. clog.Errorf("[UpdatePlayerHeros] player = %d, error: %v", playerID, err)
  377. }
  378. }
  379. // 删除玩家英雄
  380. func (p *game) DeletePlayerHeros(playerID int64, heroIDList ...int64) {
  381. if len(heroIDList) == 0 {
  382. return
  383. }
  384. key := p.buildPlayerHeroKey(playerID)
  385. fields := make([]string, 0, len(heroIDList))
  386. for _, heroID := range heroIDList {
  387. fields = append(fields, cstring.ToString(heroID))
  388. }
  389. err := p.HDel(p.Context(), key, fields...).Err()
  390. if err != nil {
  391. clog.Errorf("[DeletePlayerHeros] player = %d, error: %v", playerID, err)
  392. }
  393. }
  394. // 根据玩家战斗部队数据获取部队和英雄数据
  395. func (p *game) GetPlayerTeamHeroData(playerID int64, teamID int32, heroGUID int64) (*pb.PlayerData, *pb.BattleTeam, *pb.PlayerHero) {
  396. pipeline := p.Pipeline()
  397. defer pipeline.Close()
  398. // 获取玩家数据
  399. playerKey := p.buildPlayerDataKey(playerID)
  400. playerCmd := pipeline.HMGet(p.Context(), playerKey, nameRedis.PlayerBaseFields...)
  401. // 获取队伍数据
  402. teamKey := p.buildPlayerTeamKey(playerID)
  403. teamCmd := pipeline.HGet(p.Context(), teamKey, cstring.ToString(teamID))
  404. // 获取英雄数据
  405. heroKey := p.buildPlayerHeroKey(playerID)
  406. heroCmd := pipeline.HGet(p.Context(), heroKey, cstring.ToString(heroGUID))
  407. if _, err := pipeline.Exec(p.Context()); err != nil && !errors.Is(err, redis.Nil) {
  408. clog.Warnf("[GetPlayerTeamHeroData] err = %v", err)
  409. return nil, nil, nil
  410. }
  411. playerData, _ := p.decodePlayerData(playerID, nameRedis.PlayerBaseFields, playerCmd)
  412. teamData, _ := p.decodeBattleTeam(teamID, teamCmd)
  413. heroData, _ := p.decodePlayerHero(heroGUID, heroCmd)
  414. return playerData, teamData, heroData
  415. }
  416. // 根据玩家战斗部队数据获取部队和英雄数据
  417. func (p *game) GetPlayerBattleData(battleTeamDataList ...*pb.BattleTeamData) (map[int64]*pb.BattleTeam, map[int64]*pb.PlayerHero) {
  418. if len(battleTeamDataList) == 0 {
  419. return nil, nil
  420. }
  421. // 收集每个玩家对应的 teamID(每个玩家只有一个部队)和 heroID 列表
  422. playerTeamIDs := map[int64]int32{}
  423. playerHeroIDs := map[int64][]int64{}
  424. for _, teamData := range battleTeamDataList {
  425. if teamData == nil {
  426. continue
  427. }
  428. playerTeamIDs[teamData.PlayerID] = teamData.TeamID
  429. playerHeroIDs[teamData.PlayerID] = teamData.TeamHeros
  430. }
  431. // 去重玩家列表并保持顺序(用于 pipeline 对应)
  432. playerIDs := make([]int64, 0, len(playerTeamIDs))
  433. for pid := range playerTeamIDs {
  434. playerIDs = append(playerIDs, pid)
  435. }
  436. pipeline := p.Pipeline()
  437. defer pipeline.Close()
  438. // 为每个玩家创建 HGET 请求(因为每个玩家只有一个部队字段)
  439. teamCmds := make([]*redis.StringCmd, 0, len(playerIDs))
  440. heroCmds := make([]*redis.SliceCmd, 0, len(playerIDs))
  441. for _, pid := range playerIDs {
  442. teamKey := p.buildPlayerTeamKey(pid)
  443. field := cstring.ToString(playerTeamIDs[pid])
  444. teamCmds = append(teamCmds, pipeline.HGet(p.Context(), teamKey, field))
  445. // 为每个玩家创建 HMGET 请求(因为每个玩家有多个英雄字段)
  446. heroKey := p.buildPlayerHeroKey(pid)
  447. var fields []string
  448. for _, hid := range playerHeroIDs[pid] {
  449. fields = append(fields, cstring.ToString(hid))
  450. }
  451. heroCmds = append(heroCmds, pipeline.HMGet(p.Context(), heroKey, fields...))
  452. }
  453. if _, err := pipeline.Exec(p.Context()); err != nil {
  454. clog.Errorf("[GetPlayerBattleData] pipeline exec error: %v", err)
  455. return nil, nil
  456. }
  457. teamsMap := map[int64]*pb.BattleTeam{}
  458. heroesMap := map[int64]*pb.PlayerHero{}
  459. // 解析部队结果
  460. for i, cmd := range teamCmds {
  461. playerID := playerIDs[i]
  462. str, err := cmd.Result()
  463. if err != nil {
  464. if !errors.Is(err, redis.Nil) {
  465. clog.Errorf("[GetPlayerBattleData] get team error player = %d, err = %v", playerID, err)
  466. }
  467. continue
  468. }
  469. if str == "" {
  470. continue
  471. }
  472. var team pb.BattleTeam
  473. if err := jsoniter.UnmarshalFromString(str, &team); err != nil {
  474. clog.Errorf("[GetPlayerBattleData] unmarshal team error player = %d, err = %v", playerID, err)
  475. continue
  476. }
  477. teamsMap[playerID] = &team
  478. }
  479. // 解析英雄结果
  480. for _, cmd := range heroCmds {
  481. result, err := cmd.Result()
  482. if err != nil {
  483. if !errors.Is(err, redis.Nil) {
  484. clog.Errorf("[GetPlayerBattleData] get heroes error: %v", err)
  485. }
  486. } else {
  487. for _, item := range result {
  488. if item == nil {
  489. continue
  490. }
  491. str, ok := item.(string)
  492. if !ok || str == "" {
  493. continue
  494. }
  495. var hero pb.PlayerHero
  496. if err := jsoniter.UnmarshalFromString(str, &hero); err != nil {
  497. clog.Errorf("[GetPlayerBattleData] unmarshal hero error, err = %v", err)
  498. continue
  499. }
  500. heroesMap[hero.Id] = &hero
  501. }
  502. }
  503. }
  504. return teamsMap, heroesMap
  505. }
  506. // 保存战报数据,并追加到战报组数据中
  507. func (p *game) AddReports(groupID string, reports ...*pb.Report) bool {
  508. if len(reports) == 0 {
  509. return false
  510. }
  511. groupKey := p.buildBattleReportGroupKey(groupID)
  512. pipeline := p.Pipeline()
  513. defer pipeline.Close()
  514. var idValues []any
  515. for _, report := range reports {
  516. if report == nil {
  517. continue
  518. }
  519. jsonStr, err := jsoniter.MarshalToString(report)
  520. if err != nil {
  521. clog.Errorf("[AddReports] marshal error: %v", err)
  522. continue
  523. }
  524. // 先写入战报 key
  525. reportKey := p.buildBattleReportKey(report.ReportID)
  526. pipeline.Set(p.Context(), reportKey, jsonStr, NoExpire)
  527. // 只有序列化成功的战报ID才追加到组列表
  528. idValues = append(idValues, report.ReportID)
  529. }
  530. // 如果没有任何要写入的 id,则仍可能成功(但无实际写入),根据需要可视为失败,这里视为失败
  531. if len(idValues) == 0 {
  532. clog.Warnf("[AddReports] no valid reports to save for group = %s", groupID)
  533. return false
  534. }
  535. // 将所有战报ID 追加到组列表
  536. pipeline.LPush(p.Context(), groupKey, idValues...)
  537. if _, err := pipeline.Exec(p.Context()); err != nil {
  538. clog.Errorf("[AddReports] pipeline error: %v", err)
  539. return false
  540. }
  541. return true
  542. }
  543. // 获取战报组的所有战报ID集合
  544. func (p *game) GetReportIDListOfGroup(groupIDList ...string) map[string][]string {
  545. resultMap := map[string][]string{}
  546. if len(groupIDList) == 0 {
  547. return resultMap
  548. }
  549. pipeline := p.Pipeline()
  550. defer pipeline.Close()
  551. cmds := make([]*redis.StringSliceCmd, 0, len(groupIDList))
  552. for _, groupID := range groupIDList {
  553. key := p.buildBattleReportGroupKey(groupID)
  554. cmds = append(cmds, pipeline.LRange(p.Context(), key, 0, -1))
  555. }
  556. if _, err := pipeline.Exec(p.Context()); err != nil {
  557. clog.Errorf("[GetReportIDListOfGroup] pipeline error: %v", err)
  558. return resultMap
  559. }
  560. for i, groupID := range groupIDList {
  561. result, err := cmds[i].Result()
  562. if err != nil {
  563. if !errors.Is(err, redis.Nil) {
  564. clog.Errorf("[GetReportIDListOfGroup] get error: %v", err)
  565. }
  566. continue
  567. }
  568. var idList []string
  569. for _, id := range result {
  570. if id != "" {
  571. idList = append(idList, id)
  572. }
  573. }
  574. resultMap[groupID] = idList
  575. }
  576. return resultMap
  577. }
  578. // 获取战报
  579. func (p *game) GetBattleReports(reportIDList ...string) map[string]*pb.Report {
  580. reportMap := map[string]*pb.Report{}
  581. if len(reportIDList) == 0 {
  582. return reportMap
  583. }
  584. pipeline := p.Pipeline()
  585. defer pipeline.Close()
  586. cmds := make([]*redis.StringCmd, 0, len(reportIDList))
  587. for _, reportID := range reportIDList {
  588. key := p.buildBattleReportKey(reportID)
  589. cmds = append(cmds, pipeline.Get(p.Context(), key))
  590. }
  591. if _, err := pipeline.Exec(p.Context()); err != nil {
  592. clog.Errorf("[GetBattleReports] pipeline error: %v", err)
  593. return reportMap
  594. }
  595. for i, reportID := range reportIDList {
  596. result, err := cmds[i].Result()
  597. if err != nil {
  598. if !errors.Is(err, redis.Nil) {
  599. clog.Errorf("[GetBattleReports] get error: %v", err)
  600. }
  601. continue
  602. }
  603. var report pb.Report
  604. if err := jsoniter.UnmarshalFromString(result, &report); err != nil {
  605. clog.Errorf("[GetBattleReports] unmarshal error: %v", err)
  606. continue
  607. }
  608. reportMap[reportID] = &report
  609. }
  610. return reportMap
  611. }
  612. // 更新玩家设置信息
  613. func (p *game) UpdatePlayerSetting(playerID int64, setting *pb.LordShowDataList) {
  614. key := p.buildPlayerSettingKey(playerID)
  615. value, err := jsoniter.MarshalToString(setting)
  616. if err != nil {
  617. clog.Errorf("[UpdatePlayerSetting] player = %d, marshal error: %v", playerID, err)
  618. return
  619. }
  620. err = p.Set(p.Context(), key, value, 0).Err()
  621. if err != nil {
  622. clog.Errorf("[UpdatePlayerSetting] player = %d, error: %v", playerID, err)
  623. }
  624. }
  625. // 获取玩家设置信息
  626. func (p *game) GetPlayerSetting(playerID int64) *pb.LordShowDataList {
  627. key := p.buildPlayerSettingKey(playerID)
  628. value, err := p.Get(p.Context(), key).Result()
  629. if err != nil {
  630. if !errors.Is(err, redis.Nil) {
  631. clog.Errorf("[GetPlayerSetting] player = %d, error: %v", playerID, err)
  632. }
  633. return nil
  634. }
  635. list := &pb.LordShowDataList{}
  636. if err := jsoniter.UnmarshalFromString(value, list); err != nil {
  637. clog.Errorf("[GetBattleReports] unmarshal error: %v", err)
  638. return nil
  639. }
  640. return list
  641. }
  642. func (p *game) buildPlayerDataKey(playerID int64) string {
  643. return fmt.Sprintf(p.playerDataKey, playerID)
  644. }
  645. func (p *game) buildPlayerTeamKey(playerID int64) string {
  646. return fmt.Sprintf(p.playerTeamKey, playerID)
  647. }
  648. func (p *game) buildPlayerHeroKey(playerID int64) string {
  649. return fmt.Sprintf(p.playerHeroKey, playerID)
  650. }
  651. func (p *game) buildBattleReportGroupKey(groupID string) string {
  652. return fmt.Sprintf(p.battleReportGroupKey, groupID)
  653. }
  654. func (p *game) buildBattleReportKey(reportID string) string {
  655. return fmt.Sprintf(p.battleReportKey, reportID)
  656. }
  657. func (p *game) buildPlayerSettingKey(playerID int64) string {
  658. return fmt.Sprintf(p.settingKey, playerID)
  659. }
  660. func (p *game) decodePlayerData(playerID int64, fields []string, value *redis.SliceCmd) (*pb.PlayerData, bool) {
  661. result, err := value.Result()
  662. if err != nil {
  663. if !errors.Is(err, redis.Nil) {
  664. clog.Errorf("[GetPlayerData] player = %d, error: %v", playerID, err)
  665. }
  666. return nil, false
  667. }
  668. maps := make(map[string]any, len(fields))
  669. for i := range len(fields) {
  670. maps[fields[i]] = result[i]
  671. }
  672. // 添加玩家 ID 字段
  673. maps[nameRedis.Player_PlayerID] = playerID
  674. var playerData pb.PlayerData
  675. if err := cmapStructure.WeakDecode(maps, &playerData); err != nil {
  676. clog.Errorf("[GetPlayerData] player = %d, error: %v", playerID, err)
  677. return nil, false
  678. }
  679. return &playerData, true
  680. }
  681. func (p *game) decodeBattleTeam(teamID int32, value *redis.StringCmd) (*pb.BattleTeam, bool) {
  682. result, err := value.Result()
  683. if err != nil {
  684. if !errors.Is(err, redis.Nil) {
  685. clog.Errorf("[decodeBattleTeam] team = %d, error: %v", teamID, err)
  686. }
  687. return nil, false
  688. }
  689. var teamData pb.BattleTeam
  690. if err := jsoniter.UnmarshalFromString(result, &teamData); err != nil {
  691. clog.Errorf("[decodeBattleTeam] team = %d, error: %v", teamID, err)
  692. return nil, false
  693. }
  694. return &teamData, true
  695. }
  696. func (p *game) decodePlayerHero(playerID int64, value *redis.StringCmd) (*pb.PlayerHero, bool) {
  697. result, err := value.Result()
  698. if err != nil {
  699. if !errors.Is(err, redis.Nil) {
  700. clog.Errorf("[decodePlayerHero] player = %d, error: %v", playerID, err)
  701. }
  702. return nil, false
  703. }
  704. var heroData pb.PlayerHero
  705. if err := jsoniter.UnmarshalFromString(result, &heroData); err != nil {
  706. clog.Errorf("[decodePlayerHero] player = %d, error: %v", playerID, err)
  707. return nil, false
  708. }
  709. return &heroData, true
  710. }
  711. func (p *game) buildMapExchangeLogKey(mapNodeID string, playerID int64, logType enum.ExchangeLogType) string {
  712. return fmt.Sprintf(p.mapExchangeLogKey, mapNodeID, playerID, logType)
  713. }
  714. // 获取地图交易所日志列表
  715. // lastLogID > 0 获取最近pageSize条消息,不包括lastLogID
  716. // lastLogID = 0 获取最新的pageSize条消息
  717. func (p *game) GetMapExchangeLogList(mapNodeID string, playerID int64, logType enum.ExchangeLogType, lastLogID int64) *pb.ExchangeLogList {
  718. key := p.buildMapExchangeLogKey(mapNodeID, playerID, logType)
  719. return p.getMapExchangeLogList(key, lastLogID)
  720. }
  721. // 获取地图交易所日志列表
  722. // lastLogID > 0 获取最近pageSize条消息,不包括lastLogID
  723. // lastLogID = 0 获取最新的pageSize条消息
  724. func (p *game) getMapExchangeLogList(key string, lastLogID int64) *pb.ExchangeLogList {
  725. zSliceToChatMsgs := func(zs []redis.Z, isReverse bool) *pb.ExchangeLogList {
  726. logList := &pb.ExchangeLogList{}
  727. for _, v := range zs {
  728. if v.Member == nil {
  729. continue
  730. }
  731. var log pb.ExchangeLog
  732. if err := jsoniter.UnmarshalFromString(v.Member.(string), &log); err == nil {
  733. logList.List = append(logList.List, &log)
  734. }
  735. }
  736. // 反转
  737. if isReverse {
  738. for i, j := 0, len(logList.List)-1; i < j; i, j = i+1, j-1 {
  739. logList.List[i], logList.List[j] = logList.List[j], logList.List[i]
  740. }
  741. }
  742. return logList
  743. }
  744. if lastLogID > 0 {
  745. // 获取 lastLogID 之前的 20 条
  746. values, err := p.ZRevRangeByScoreWithScores(p.Context(), key, &redis.ZRangeBy{
  747. Max: fmt.Sprintf("(%d", lastLogID), // 开区间,排除自身
  748. Min: "-inf",
  749. Offset: 0,
  750. Count: pageSize,
  751. }).Result()
  752. if err != nil && err != redis.Nil {
  753. clog.Warnf("[getMapExchangeLogList] ZRevRangeByScoreWithScores error. [key = %s, lastLogID = %d, error: %v]", key, lastLogID, err)
  754. return nil
  755. }
  756. return zSliceToChatMsgs(values, false)
  757. } else {
  758. // 最新 20 条,降序
  759. values, err := p.ZRevRangeWithScores(p.Context(), key, 0, pageSize-1).Result()
  760. if err != nil && err != redis.Nil {
  761. clog.Warnf("[getMapExchangeLogList] ZRevRangeWithScores error. [key = %s, lastLogID = %d, error: %v]", key, lastLogID, err)
  762. return nil
  763. }
  764. // 反转成升序
  765. return zSliceToChatMsgs(values, false)
  766. }
  767. }
  768. // 添加一条地图交易所日志
  769. func (p *game) AddMapExchangeLog(mapNodeID string, playerID int64, logType enum.ExchangeLogType, maxLimit int64, log *pb.ExchangeLog) {
  770. key := p.buildMapExchangeLogKey(mapNodeID, playerID, logType)
  771. p.addMapExchangeLog(key, maxLimit, log)
  772. }
  773. // 保存一条日志消息
  774. func (p *game) addMapExchangeLog(key string, maxLimit int64, log *pb.ExchangeLog) {
  775. data, err := jsoniter.MarshalToString(log)
  776. if err != nil {
  777. clog.Warnf("[addMapExchangeLog] marshal error. [key = %s, maxLimit = %d, error: %v]", key, maxLimit, err)
  778. return
  779. }
  780. // 使用 LogID 作为 score,保证顺序
  781. err = p.ZAdd(p.Context(), key, &redis.Z{
  782. Score: float64(log.LogID),
  783. Member: data,
  784. }).Err()
  785. if err != nil {
  786. clog.Warnf("[addMapExchangeLog] zadd error. [key = %s, maxLimit = %d, error: %v]", key, maxLimit, err)
  787. return
  788. }
  789. // 裁剪到最多 maxLimit 条
  790. err = p.ZRemRangeByRank(p.Context(), key, 0, -maxLimit-1).Err()
  791. if err != nil {
  792. clog.Warnf("[addMapExchangeLog] zremrangebyrank error. [key = %s, maxLimit = %d, error: %v]", key, maxLimit, err)
  793. }
  794. }