player.go 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308
  1. package player
  2. import (
  3. "errors"
  4. "gadmin/config"
  5. "gadmin/internal/admin/consts"
  6. "gadmin/internal/gorm/model"
  7. "gadmin/internal/gorm/query"
  8. "strconv"
  9. "sync"
  10. "github.com/sirupsen/logrus"
  11. "github.com/spf13/cast"
  12. "gorm.io/gorm"
  13. )
  14. const (
  15. maxMapStoreLimit = 50000
  16. )
  17. var (
  18. AllRoleData = make(map[int]*RoleData, 0)
  19. lock sync.RWMutex
  20. userStampMap = make(map[int64]int32)
  21. userChannelLock sync.RWMutex
  22. userChannelMap = make(map[int64]string, 0)
  23. )
  24. type RoleData struct {
  25. ID int `json:"id"`
  26. Type int `json:"type"`
  27. Evolution int `json:"evolution"`
  28. Level int `json:"level"`
  29. Break int `json:"break"`
  30. }
  31. // GetRoleData 根据角色ID获取角色信息
  32. func GetRoleData(id int) *RoleData {
  33. idx := strconv.Itoa(id)
  34. if len(idx) == 9 {
  35. level, _ := strconv.Atoi(idx[5:8])
  36. role := &RoleData{
  37. ID: id,
  38. Type: cast.ToInt(idx[:1]),
  39. Evolution: cast.ToInt(idx[4:5]),
  40. Level: level,
  41. Break: id % 10,
  42. }
  43. return role
  44. }
  45. if len(idx) == 10 {
  46. level, _ := strconv.Atoi(idx[6:9])
  47. role := &RoleData{
  48. ID: id,
  49. Type: cast.ToInt(idx[:2]),
  50. Evolution: cast.ToInt(idx[5:6]),
  51. Level: level,
  52. Break: id % 10,
  53. }
  54. return role
  55. }
  56. logrus.Warnf("GetRoleData role = nil, id:%+v", id)
  57. return nil
  58. }
  59. func GetUserChannel(userId int64) string {
  60. if userId == 0 {
  61. return "0"
  62. }
  63. //if val, ok := userChannelMap[userId]; ok {
  64. // return val
  65. //}
  66. userChannelLock.RLock()
  67. defer userChannelLock.RUnlock()
  68. u := query.Use(config.DB).PlayerChannel
  69. user, err := u.Where(u.Playerid.Eq(userId)).Select(u.ChannelID).First()
  70. if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
  71. logrus.Errorf("GetUserChannel query user failed:%v", err)
  72. return "0"
  73. }
  74. if user == nil {
  75. logrus.Errorf("GetUserChannel 未查到玩家%v渠道ID..\n", userId)
  76. return "0"
  77. } else {
  78. return user.ChannelID
  79. }
  80. //if user == nil {
  81. // userChannelMap[userId] = "0"
  82. //} else {
  83. // userChannelMap[userId] = user.ChannelID
  84. //}
  85. //
  86. //if len(userChannelMap) >= maxMapStoreLimit {
  87. // userChannelMap = make(map[int64]string, 0)
  88. //}
  89. //return userChannelMap[userId]
  90. }
  91. func GetUserStamp(userId int64) int32 {
  92. if val, ok := userStampMap[userId]; ok {
  93. return val
  94. }
  95. if userId == 0 {
  96. return 0
  97. }
  98. var (
  99. user *model.PlayerAttr
  100. err error
  101. )
  102. lock.RLock()
  103. defer func() {
  104. if len(userStampMap) > maxMapStoreLimit {
  105. userStampMap = make(map[int64]int32)
  106. }
  107. lock.RUnlock()
  108. }()
  109. for _, DB := range config.GDBGroup {
  110. u := query.Use(DB).PlayerAttr
  111. user, err = u.Where(u.Playerid.Eq(userId)).Select(u.CreateTime).First()
  112. if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
  113. logrus.Errorf("GetUserStamp query user failed:%v", err)
  114. return 0
  115. }
  116. if user == nil {
  117. continue
  118. } else {
  119. userStampMap[userId] = int32(user.CreateTime.Unix())
  120. break
  121. }
  122. }
  123. if user == nil {
  124. //userStampMap[userId] = 0
  125. logrus.Errorf("GetUserStamp 未查到玩家注册时间..")
  126. return 0
  127. }
  128. return userStampMap[userId]
  129. }
  130. // GetServerIDByUserId 通过玩家ID获取所在的数据库
  131. func GetServerIDByUserId(playerId int64) (serverID int32, err error) {
  132. var models *model.PlayerAttr
  133. for k, DB := range config.GDBGroup {
  134. logrus.Warnf("GetDBByUserId DB:%+v", k)
  135. var (
  136. q = query.Use(DB).PlayerAttr
  137. m = q.Where(q.Playerid.Eq(playerId))
  138. )
  139. if err = m.Scan(&models); err != nil {
  140. return
  141. }
  142. logrus.Warnf("GetDBByUserId models:%+v", models)
  143. if models != nil {
  144. break
  145. }
  146. }
  147. if models != nil {
  148. return models.Line, nil
  149. }
  150. return 0, errors.New("用户不存在")
  151. }
  152. // GetAccIDByUserId 通过玩家ID获取accId
  153. func GetAccIDByUserId(playerId int64) (accID int64, err error) {
  154. var models *model.PlayerAttr
  155. for _, DB := range config.GDBGroup {
  156. var (
  157. q = query.Use(DB).PlayerAttr
  158. m = q.Where(q.Playerid.Eq(playerId))
  159. )
  160. if err = m.Scan(&models); err != nil {
  161. return
  162. }
  163. if models != nil {
  164. break
  165. }
  166. }
  167. if models != nil {
  168. return models.AccID, nil
  169. }
  170. return 0, errors.New("用户不存在")
  171. }
  172. // GetAccIDByOpenId 通过openID获取accId
  173. func GetAccIDByOpenId(openID string) (accID int64, err error) {
  174. var models *model.PlayerAttr
  175. for _, DB := range config.GDBGroup {
  176. var (
  177. q = query.Use(DB).PlayerAttr
  178. m = q.Where(q.OpenID.Eq(openID))
  179. )
  180. if err = m.Scan(&models); err != nil {
  181. return
  182. }
  183. if models != nil {
  184. break
  185. }
  186. }
  187. if models != nil {
  188. return models.AccID, nil
  189. }
  190. return 0, errors.New("用户不存在")
  191. }
  192. // GetAttrByUserId 通过User获取user attr
  193. func GetAttrByUserId(playerId int64) (attr *model.PlayerAttr, err error) {
  194. var models *model.PlayerAttr
  195. for _, DB := range config.GDBGroup {
  196. var (
  197. q = query.Use(DB).PlayerAttr
  198. m = q.Where(q.Playerid.Eq(playerId))
  199. )
  200. if err = m.Scan(&models); err != nil {
  201. return
  202. }
  203. if models != nil {
  204. break
  205. }
  206. }
  207. if models != nil {
  208. return models, nil
  209. }
  210. return nil, errors.New("用户不存在")
  211. }
  212. func GetAccountByAccId(accId int64) (account *model.UserAccount, err error) {
  213. var models *model.UserAccount
  214. q := config.LDB.Scopes(model.UserAccountTable(models, accId)).Where("accId = ?", accId)
  215. if err := q.Scan(&models).Error; err != nil {
  216. return nil, err
  217. }
  218. if models != nil {
  219. return models, nil
  220. }
  221. return nil, errors.New("账号不存在")
  222. }
  223. // GetDBByUserId 通过玩家ID获取所在的数据库
  224. func GetDBByUserId(playerId int64) (Db int, err error) {
  225. serverID, err := GetServerIDByUserId(playerId)
  226. if err != nil {
  227. return 0, err
  228. }
  229. if serverID > 0 {
  230. return GetDBByServerID(int(serverID))
  231. }
  232. return 0, errors.New("用户不存在")
  233. }
  234. // GetDBByServerID 通过serverID获取数据库
  235. func GetDBByServerID(serverID int) (int, error) {
  236. // 目前serverId 和db是对应关系,后面如果不是可以在这里做处理
  237. return serverID, nil
  238. }
  239. // GetDBByPlayerAttr 通过用户信息获取数据库(兼容合服)
  240. func GetDBByPlayerAttr(playerAttr *model.PlayerAttr) (int, error) {
  241. // 目前serverId 和db是对应关系,后面如果不是可以在这里做处理
  242. return int(playerAttr.Line), nil
  243. }
  244. func GetUserFlag(serverID int, userId int64) int32 {
  245. u := query.Use(config.GDBGroup[serverID]).PlayerAttr
  246. user, err := u.Where(u.Playerid.Eq(userId)).Select(u.Flag).First()
  247. if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
  248. logrus.Errorf("GetUserFlag query user failed:%v", err)
  249. return 0
  250. }
  251. if user == nil {
  252. return consts.UserFlagOld
  253. } else {
  254. return user.Flag
  255. }
  256. }