synthesis.go 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479
  1. package service
  2. import (
  3. "context"
  4. "math"
  5. "sort"
  6. "time"
  7. mhayaTime "github.com/mhaya/extend/time"
  8. cutils "github.com/mhaya/extend/utils"
  9. "github.com/mhaya/game/game_cluster/internal/code"
  10. "github.com/mhaya/game/game_cluster/internal/constant"
  11. "github.com/mhaya/game/game_cluster/internal/mdb"
  12. "github.com/mhaya/game/game_cluster/internal/mdb/eventmodels"
  13. "github.com/mhaya/game/game_cluster/internal/mdb/models"
  14. "github.com/mhaya/game/game_cluster/nodes/webadmin/common"
  15. "github.com/mhaya/game/game_cluster/nodes/webadmin/entity"
  16. "github.com/mhaya/game/game_cluster/nodes/webadmin/model"
  17. mhayaLogger "github.com/mhaya/logger"
  18. "github.com/spf13/cast"
  19. "go.mongodb.org/mongo-driver/bson"
  20. "go.mongodb.org/mongo-driver/bson/primitive"
  21. "go.mongodb.org/mongo-driver/mongo"
  22. "go.mongodb.org/mongo-driver/mongo/options"
  23. "gorm.io/gorm"
  24. )
  25. type Synthesis struct {
  26. nodeId string
  27. db *mongo.Database
  28. }
  29. func NewSynthesis(nodeId string) *Synthesis {
  30. return &Synthesis{
  31. nodeId: nodeId,
  32. db: mdb.MDB,
  33. }
  34. }
  35. // 综合统计
  36. func (s *Synthesis) Overview(req entity.OverviewReq) (*entity.OverviewResp, *code.Result) {
  37. page, pageSize := checkPageParam(req.Page, req.Size)
  38. playerMgr := NewPlayerManage()
  39. // 根据条件查询
  40. if req.UserName != "" {
  41. detail, codeResult := s.getOverviewDetail(req.UserName)
  42. if codeResult != nil {
  43. mhayaLogger.Warnf("Overview getOverviewDetail error:%v", codeResult)
  44. return nil, codeResult
  45. }
  46. if detail == nil {
  47. return nil, nil
  48. }
  49. var results []*entity.OverviewDetail
  50. results = append(results, detail)
  51. return &entity.OverviewResp{
  52. Details: results,
  53. Total: 1,
  54. }, nil
  55. }
  56. // 查询列表
  57. listResp, codeResult := playerMgr.List(context.Background(), entity.PlayerListReq{
  58. Page: page,
  59. Size: pageSize,
  60. })
  61. if codeResult != nil {
  62. mhayaLogger.Warnf("Overview List error:%v", codeResult)
  63. return nil, codeResult
  64. }
  65. var results []*entity.OverviewDetail
  66. for _, detail := range listResp.Details {
  67. detail, codeResult := s.getOverviewDetail(detail.UserName)
  68. if codeResult != nil {
  69. mhayaLogger.Warnf("Overview getOverviewDetail error:%v", codeResult)
  70. return nil, codeResult
  71. }
  72. if detail != nil {
  73. results = append(results, detail)
  74. }
  75. }
  76. count, codeResult := playerMgr.GetPlayerTotalCount()
  77. if codeResult != nil {
  78. mhayaLogger.Warnf("Overview GetPlayerTotalCount codeResult:%v", codeResult)
  79. return nil, codeResult
  80. }
  81. return &entity.OverviewResp{
  82. Details: results,
  83. Total: count,
  84. }, nil
  85. }
  86. func (s *Synthesis) getOverviewDetail(userName string) (*entity.OverviewDetail, *code.Result) {
  87. userDetail, codeResult := s.getUserDetail(userName)
  88. if codeResult != nil {
  89. mhayaLogger.Warnf("getOverviewDetail getUserDetail error:%v", codeResult)
  90. return nil, codeResult
  91. }
  92. activeDetail, codeResult := s.getActive(userName)
  93. if codeResult != nil {
  94. mhayaLogger.Warnf("getOverviewDetail getActive error:%v", codeResult)
  95. return nil, codeResult
  96. }
  97. assetDetail, codeResult := s.getAsset(userName)
  98. if codeResult != nil {
  99. mhayaLogger.Warnf("getOverviewDetail getAsset error:%v", codeResult)
  100. return nil, codeResult
  101. }
  102. playerMgr := NewPlayerManage()
  103. // 获取最新的用户骰子投掷记录
  104. diceRecord, codeResult := playerMgr.GetPlayerLastestDiceRecord(userName)
  105. if codeResult != nil {
  106. mhayaLogger.Warnf("getOverviewDetail GetPlayerLastestDiceRecord error:%v", codeResult)
  107. return nil, codeResult
  108. }
  109. // 获取最新的用户登录记录
  110. loginRecord, codeResult := playerMgr.GetPlayerLastestLoginRecord(userName)
  111. if codeResult != nil {
  112. mhayaLogger.Warnf("getOverviewDetail GetPlayerLastestLoginRecord error:%v", codeResult)
  113. return nil, codeResult
  114. }
  115. return &entity.OverviewDetail{
  116. UserName: userName,
  117. OpenId: userDetail.OpenId,
  118. FirstLoginDate: userDetail.FirstLoginDate,
  119. CreateTime: userDetail.CreateTime,
  120. LastLoginTime: userDetail.LastLoginTime,
  121. MaxSuccessions: activeDetail.MaxSuccessions,
  122. DieRoll: int64(diceRecord.Count),
  123. CurrentIntegral: assetDetail.PointsValue,
  124. SuccessfulInvitations: activeDetail.SuccessfulInvitations,
  125. Withdrawals: activeDetail.Withdrawals,
  126. CumulativeWithdrawalAmount: activeDetail.CumulativeWithdrawalAmount,
  127. TurntableRuns: activeDetail.TurntableRuns,
  128. PointsRanking: func() int64 {
  129. if loginRecord != nil {
  130. return loginRecord.PointsRankSeq
  131. }
  132. return 0
  133. }(),
  134. RankingOfDiceRolls: func() int64 {
  135. if loginRecord != nil {
  136. return loginRecord.ThrowDiceRankSeq
  137. }
  138. return 0
  139. }(),
  140. InvitesRanking: func() int64 {
  141. if loginRecord != nil {
  142. return loginRecord.InviteUserRankSeq
  143. }
  144. return 0
  145. }(),
  146. TonValue: assetDetail.TonValue,
  147. UsdtValue: assetDetail.UsdtValue,
  148. }, nil
  149. }
  150. // 统计用户相关信息
  151. func (s *Synthesis) UserList(req entity.UserListReq) (*entity.UserListResp, *code.Result) {
  152. page, pageSize := checkPageParam(req.Page, req.Size)
  153. playerMgr := NewPlayerManage()
  154. // 根据条件查询
  155. if req.UserName != "" {
  156. detail, codeResult := s.getUserDetail(req.UserName)
  157. if codeResult != nil {
  158. mhayaLogger.Warnf("UserList getUserDetail error:%v", codeResult)
  159. return nil, codeResult
  160. }
  161. if detail == nil {
  162. return nil, nil
  163. }
  164. var results []*entity.UserListDetail
  165. results = append(results, detail)
  166. return &entity.UserListResp{
  167. Details: results,
  168. Total: 1,
  169. }, nil
  170. }
  171. // 查询列表
  172. listResp, codeResult := playerMgr.List(context.Background(), entity.PlayerListReq{
  173. Page: page,
  174. Size: pageSize,
  175. })
  176. if codeResult != nil {
  177. mhayaLogger.Warnf("UserList List error:%v", codeResult)
  178. return nil, codeResult
  179. }
  180. var results []*entity.UserListDetail
  181. for _, detail := range listResp.Details {
  182. ret, codeResult := s.getUserDetail(detail.UserName)
  183. if codeResult != nil {
  184. mhayaLogger.Warnf("UserList getUserDetail error:%v", codeResult)
  185. return nil, codeResult
  186. }
  187. if detail != nil {
  188. results = append(results, ret)
  189. }
  190. }
  191. count, codeResult := playerMgr.GetPlayerTotalCount()
  192. if codeResult != nil {
  193. mhayaLogger.Warnf("UserList GetPlayerTotalCount codeResult:%v", codeResult)
  194. return nil, codeResult
  195. }
  196. return &entity.UserListResp{
  197. Details: results,
  198. Total: int64(count),
  199. }, nil
  200. }
  201. func (s *Synthesis) getUserDetail(userName string) (*entity.UserListDetail, *code.Result) {
  202. playerMgr := NewPlayerManage()
  203. // 获取注册记录
  204. registerRecord, codeResult := playerMgr.GetRegisterRecord(userName)
  205. if codeResult != nil {
  206. mhayaLogger.Warnf("getUserDetail GetRegisterRecord error:%v", codeResult)
  207. return nil, codeResult
  208. }
  209. if registerRecord == nil {
  210. mhayaLogger.Warnf("getUserDetail registerRecord:%v", registerRecord)
  211. return nil, nil
  212. }
  213. // 获取用户的首次登录记录
  214. firstLoginRecord, codeResult := playerMgr.GetPlayerFirstLoginRecord(userName)
  215. if codeResult != nil {
  216. mhayaLogger.Warnf("getUserDetail GetPlayerFirstLoginRecord error:%v", codeResult)
  217. return nil, codeResult
  218. }
  219. // 获取最新的用户登录记录
  220. loginRecord, codeResult := playerMgr.GetPlayerLastestLoginRecord(userName)
  221. if codeResult != nil {
  222. mhayaLogger.Warnf("getUserDetail GetPlayerLastestLoginRecord error:%v", codeResult)
  223. return nil, codeResult
  224. }
  225. return &entity.UserListDetail{
  226. UserName: userName,
  227. OpenId: func() string {
  228. if loginRecord != nil {
  229. return loginRecord.TgId
  230. }
  231. return ""
  232. }(),
  233. CreateTime: registerRecord.CreateAt,
  234. FirstLoginDate: func() int64 {
  235. if firstLoginRecord != nil {
  236. return firstLoginRecord.CreateAt
  237. }
  238. return 0
  239. }(),
  240. LoginIP: func() string {
  241. if loginRecord != nil {
  242. return loginRecord.Ip
  243. }
  244. return ""
  245. }(),
  246. LastLoginTime: func() int64 {
  247. if loginRecord != nil {
  248. return loginRecord.CreateAt
  249. }
  250. return 0
  251. }(),
  252. TonWall: func() string {
  253. if loginRecord != nil {
  254. return loginRecord.TonWall
  255. }
  256. return ""
  257. }(),
  258. Fingerprint: func() string {
  259. if loginRecord != nil {
  260. return loginRecord.Fingerprint
  261. }
  262. return ""
  263. }(),
  264. }, nil
  265. }
  266. func (s *Synthesis) FindMDBUserLogDaily(req entity.UserLogDailyReq) (*entity.UserLogDailyResp, *code.Result) {
  267. page, pageSize := checkPageParam(req.Page, req.Size)
  268. // 定义上下文
  269. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  270. defer cancel()
  271. // 指定集合
  272. collection := mdb.MDB.Collection(constant.CNamePlayerDailyRecord)
  273. // 构建查询条件 - 如果查询值为空那就不添加查询条件
  274. filter := bson.M{}
  275. if req.StartTime != 0 {
  276. filter["daily"] = bson.M{
  277. "$gte": req.StartTime,
  278. "$lte": req.EndTime,
  279. }
  280. }
  281. if req.Platform != "" && req.Platform != "all" {
  282. filter["platform"] = req.Platform
  283. }
  284. if req.Channel != "" {
  285. filter["channel"] = req.Channel
  286. }
  287. // 计算总数
  288. count, err := collection.CountDocuments(ctx, filter)
  289. if err != nil {
  290. mhayaLogger.Warnf("FindMDBUserLogDaily CountDocuments error:%v", err)
  291. return nil, common.NewResult(code.InternalError)
  292. }
  293. // 执行查询
  294. opts := options.Find()
  295. opts.SetSkip(int64((page - 1) * pageSize))
  296. opts.SetLimit(int64(pageSize))
  297. opts.SetSort(bson.D{{"daily", -1}})
  298. cursor, err := collection.Find(ctx, filter, opts)
  299. if err != nil {
  300. mhayaLogger.Warnf("FindMDBUserLogDaily Find error:%v", err)
  301. return nil, common.NewResult(code.InternalError)
  302. }
  303. defer cursor.Close(ctx)
  304. // 解析查询结果
  305. var results []*entity.UserLogDailyDetail
  306. for cursor.Next(ctx) {
  307. var result *entity.UserLogDailyDetail
  308. err := cursor.Decode(&result)
  309. if err != nil {
  310. mhayaLogger.Warnf("FindMDBUserLogDaily Decode error:%v", err)
  311. return nil, common.NewResult(code.InternalError)
  312. }
  313. results = append(results, result)
  314. }
  315. // 同一天的数据全部放到platform= ALl 且数据累加
  316. // 如果没有platform=all 的那就新增一个 同一个时间段只能有一个 platform=all
  317. // 将同一天的数据累加到platform=all的记录中
  318. allPlatformRecordMap := make(map[int64]*entity.UserLogDailyDetail)
  319. for _, result := range results {
  320. allPlatformRecordMap[result.Timestamp] = &entity.UserLogDailyDetail{
  321. Timestamp: result.Timestamp,
  322. Platform: "all",
  323. }
  324. }
  325. for _, v := range results {
  326. if v.Timestamp == allPlatformRecordMap[v.Timestamp].Timestamp {
  327. allPlatformRecordMap[v.Timestamp].Registered += v.Registered
  328. allPlatformRecordMap[v.Timestamp].LoggedIn += v.LoggedIn
  329. allPlatformRecordMap[v.Timestamp].NewActive += v.NewActive
  330. allPlatformRecordMap[v.Timestamp].OldActive += v.OldActive
  331. allPlatformRecordMap[v.Timestamp].TotalPoints += v.TotalPoints
  332. allPlatformRecordMap[v.Timestamp].UProduced += v.UProduced
  333. allPlatformRecordMap[v.Timestamp].UCashout += v.UCashout
  334. allPlatformRecordMap[v.Timestamp].NewLogin += v.NewLogin
  335. allPlatformRecordMap[v.Timestamp].OldLogin += v.OldLogin
  336. }
  337. }
  338. // 替换原有结果
  339. for _, record := range allPlatformRecordMap {
  340. results = append(results, record)
  341. }
  342. return &entity.UserLogDailyResp{
  343. Details: results,
  344. Total: count,
  345. }, nil
  346. }
  347. // FindWithdrawal 根据请求查询提现记录
  348. func (s *Synthesis) FindWithdrawal(req entity.UserWithdrawalReq) (*entity.UserWithdrawalResp, *code.Result) {
  349. page, pageSize := checkPageParam(req.Page, req.Size)
  350. if req.Currency != "" && !constant.CurrencyValid(req.Currency) {
  351. mhayaLogger.Warnf("FindWithdrawal unknow currency:%v", req.Currency)
  352. return nil, common.NewResult(code.ParamError)
  353. }
  354. var records []*eventmodels.UserWithdrawEventContent
  355. where := &eventmodels.UserWithdrawEventContent{
  356. UserBasic: eventmodels.UserBasic{
  357. UserId: req.UserName,
  358. UserName: req.NickName,
  359. },
  360. EventBasic: eventmodels.EventBasic{
  361. ServerId: s.nodeId,
  362. Status: req.Withdrawal,
  363. },
  364. WithdrawId: req.ID,
  365. Currency: req.Currency,
  366. Address: req.Address,
  367. State: req.State,
  368. }
  369. db := mdb.LogstashDB.Model(&eventmodels.UserWithdrawEventContent{}).Where(where).Order("create_at")
  370. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  371. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  372. }
  373. if req.AmountMin > 0 && req.AmountMax > 0 && req.AmountMin <= req.AmountMax {
  374. db = db.Where("amount >= ? and amount <= ?", req.AmountMin, req.AmountMax)
  375. }
  376. if req.AfterAmountMin > 0 && req.AfterAmountMax > 0 && req.AfterAmountMin <= req.AfterAmountMax {
  377. db = db.Where("after_balance >= ? and after_balance <= ?", req.AfterAmountMin, req.AfterAmountMax)
  378. }
  379. pages := Paginate(db, page, pageSize)
  380. err := db.Scopes(pages.Limit).Find(&records).Error
  381. if err != nil && err != gorm.ErrRecordNotFound {
  382. mhayaLogger.Warnf("FindWithdrawal Find error:%v", err)
  383. return nil, common.NewResult(code.InternalError)
  384. }
  385. var results []*entity.UserWithdrawalDetail
  386. for _, v := range records {
  387. results = append(results, &entity.UserWithdrawalDetail{
  388. Id: v.WithdrawId,
  389. UserName: v.UserId,
  390. NickName: v.UserName,
  391. OpenId: v.TgId,
  392. Status: v.State,
  393. Reason: v.Reason,
  394. Withdrawal: v.Status,
  395. Amount: int(v.Amount),
  396. AfterAmount: int(v.AfterBalance),
  397. Type: v.Currency,
  398. Address: v.Address,
  399. CreateAt: v.CreateAt,
  400. UpdateAt: v.CreateAt,
  401. })
  402. }
  403. return &entity.UserWithdrawalResp{
  404. Details: results,
  405. Total: pages.Count,
  406. }, nil
  407. }
  408. // 导出提现记录
  409. func (s *Synthesis) WithdrawalExport(req entity.UserWithdrawalExportReq) (*entity.UserWithdrawalResp, *code.Result) {
  410. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  411. defer cancel()
  412. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  413. // 构建过滤器
  414. filter := bson.M{}
  415. if req.UserName != "" {
  416. filter["userName"] = req.UserName
  417. }
  418. if req.NickName != "" {
  419. filter["nickName"] = req.NickName
  420. }
  421. if req.ID != "" {
  422. filter["_id"], _ = primitive.ObjectIDFromHex(req.ID)
  423. }
  424. if req.Address != "" {
  425. filter["address"] = req.Address
  426. }
  427. if req.State > 0 {
  428. filter["state"] = req.State
  429. }
  430. if req.Withdrawal > 0 {
  431. filter["withdrawal"] = req.Withdrawal
  432. }
  433. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  434. filter["createAt"] = bson.M{
  435. "$gte": req.StartTime,
  436. "$lte": req.EndTime,
  437. }
  438. }
  439. if req.AmountMin > 0 && req.AmountMax > 0 && req.AmountMin <= req.AmountMax {
  440. filter["amount"] = bson.M{
  441. "$gte": req.AmountMin,
  442. "$lte": req.AmountMax,
  443. }
  444. }
  445. if req.AfterAmountMin > 0 && req.AfterAmountMax > 0 && req.AfterAmountMin <= req.AfterAmountMax {
  446. filter["after_amount"] = bson.M{
  447. "$gte": req.AfterAmountMin,
  448. "$lte": req.AfterAmountMax,
  449. }
  450. }
  451. findOptions := options.Find()
  452. findOptions.SetSort(bson.D{{"createAt", -1}})
  453. // 获取总数total
  454. count, err := collection.CountDocuments(ctx, filter)
  455. if err != nil {
  456. mhayaLogger.Warnf("WithdrawalExportData CountDocuments error:%v", err)
  457. return nil, common.NewResult(code.InternalError)
  458. }
  459. // 查询数据
  460. var results []*entity.UserWithdrawalDetail
  461. cursor, err := collection.Find(ctx, filter, findOptions)
  462. if err != nil {
  463. mhayaLogger.Warnf("WithdrawalExportData Find error:%v", err)
  464. return nil, common.NewResult(code.InternalError)
  465. }
  466. defer cursor.Close(ctx)
  467. // 解析结果
  468. for cursor.Next(ctx) {
  469. var result entity.UserWithdrawalDetail
  470. if err := cursor.Decode(&result); err != nil {
  471. mhayaLogger.Warnf("WithdrawalExportData Decode error:%v", err)
  472. return nil, common.NewResult(code.InternalError)
  473. }
  474. results = append(results, &result)
  475. }
  476. if err := cursor.Err(); err != nil {
  477. mhayaLogger.Warnf("WithdrawalExportData cursor error:%v", err)
  478. return nil, common.NewResult(code.InternalError)
  479. }
  480. return &entity.UserWithdrawalResp{
  481. Details: results,
  482. Total: count,
  483. }, nil
  484. }
  485. // WithdrawalStatus 更新提现状态
  486. func (s *Synthesis) WithdrawalStatus(req entity.UserWithdrawalStatus) *code.Result {
  487. // ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  488. // defer cancel()
  489. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  490. // 更新条件
  491. updateCondition := bson.M{"userName": req.UserName}
  492. // 更新内容
  493. updateContent := bson.M{"$set": bson.M{"status": req.Status}}
  494. // 设置更新选项
  495. updateOptions := options.Update() // 设置 upsert 选项
  496. // 执行更新操作
  497. _, err := collection.UpdateOne(context.TODO(), updateCondition, updateContent, updateOptions)
  498. if err != nil {
  499. mhayaLogger.Warnf("WithdrawalStatus UpdateOne error:%v", err)
  500. return common.NewResult(code.InternalError)
  501. }
  502. return nil
  503. }
  504. // WithdrawalStatusBatch 更新提现状态
  505. func (s *Synthesis) WithdrawalStatusBatch(req entity.UserWithdrawalStatusBatch) *code.Result {
  506. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  507. if len(req.ID) == 0 {
  508. return common.NewResult(code.ParamError)
  509. }
  510. for _, id := range req.ID {
  511. objID := primitive.ObjectID{}
  512. objID, _ = primitive.ObjectIDFromHex(id)
  513. updateCondition := bson.M{"_id": objID}
  514. updateContent := bson.M{}
  515. withdrawal := models.CashOutRecord{}
  516. err := collection.FindOne(context.TODO(), updateCondition).Decode(&withdrawal)
  517. if err != nil {
  518. mhayaLogger.Warnf("WithdrawalStatusBatch FindOne error:%v", err)
  519. continue
  520. }
  521. if req.Withdrawal != 0 {
  522. if withdrawal.Status == 1 {
  523. updateContent = bson.M{"$set": bson.M{"withdrawal": req.Withdrawal}}
  524. } else {
  525. continue
  526. }
  527. }
  528. if req.Status > 0 {
  529. if withdrawal.Status != 0 {
  530. continue
  531. }
  532. updateContent = bson.M{"$set": bson.M{"status": req.Status}}
  533. }
  534. updateOptions := options.Update().SetUpsert(true)
  535. _, err = collection.UpdateOne(context.TODO(), updateCondition, updateContent, updateOptions)
  536. if err != nil {
  537. mhayaLogger.Warnf("WithdrawalStatusBatch UpdateOne error:%v", err)
  538. continue
  539. }
  540. }
  541. return nil
  542. }
  543. // FindUserCountryCount 查询用户国家分布
  544. //
  545. // 返回值为 UserCountryResp 的切片和错误。
  546. func (s *Synthesis) FindUserCountryCount() (*entity.UserCountryResp, *code.Result) {
  547. // 选择数据库和集合
  548. collection := mdb.MDB.Collection(constant.CNamePlayerCountryByIPStat)
  549. // 定义聚合管道
  550. // 定义聚合管道
  551. pipeline := []bson.D{
  552. {
  553. {Key: "$project", Value: bson.D{
  554. {Key: "playerRegisterCountry", Value: bson.D{{Key: "$objectToArray", Value: "$playerRegisterCountry"}}},
  555. }},
  556. },
  557. {
  558. {Key: "$unwind", Value: "$playerRegisterCountry"},
  559. },
  560. {
  561. {Key: "$group", Value: bson.D{
  562. {Key: "_id", Value: "$playerRegisterCountry.k"},
  563. {Key: "totalValue", Value: bson.D{{Key: "$sum", Value: "$playerRegisterCountry.v"}}},
  564. }},
  565. },
  566. {
  567. {Key: "$project", Value: bson.D{
  568. {Key: "_id", Value: 0},
  569. {Key: "countryKey", Value: "$_id"},
  570. {Key: "totalValue", Value: 1},
  571. }},
  572. },
  573. }
  574. // 执行聚合查询
  575. cursor, err := collection.Aggregate(context.TODO(), pipeline)
  576. if err != nil {
  577. mhayaLogger.Warnf("FindUserCountryCount Aggregate error:%v", err)
  578. return nil, common.NewResult(code.InternalError)
  579. }
  580. defer cursor.Close(context.TODO())
  581. // 遍历查询结果
  582. var results []bson.M
  583. if err := cursor.All(context.TODO(), &results); err != nil {
  584. mhayaLogger.Warnf("FindUserCountryCount All error:%v", err)
  585. return nil, common.NewResult(code.InternalError)
  586. }
  587. var totalIPCount int64
  588. var data []*entity.UserCountryDetail
  589. // 将结果转换为 UserCountryDetail
  590. for _, r := range results {
  591. var resp entity.UserCountryDetail
  592. resp.Country = r["countryKey"].(string)
  593. resp.IPCount = int(r["totalValue"].(int32))
  594. totalIPCount += int64(resp.IPCount)
  595. data = append(data, &resp)
  596. }
  597. for _, v := range data {
  598. // 保留小数点后两位
  599. v.Percentage = math.Round(float64(v.IPCount)/float64(totalIPCount)*10000) / 100
  600. }
  601. // 根据阈值过滤结果
  602. otherCount := 0
  603. otherPercentage := 0.00
  604. filteredResults := make([]*entity.UserCountryDetail, 0)
  605. threshold := 1.00
  606. for _, r := range data {
  607. if r.Percentage >= threshold {
  608. filteredResults = append(filteredResults, r)
  609. // 保留小数点后两位
  610. r.Percentage = math.Round(r.Percentage*100) / 100
  611. otherPercentage += r.Percentage
  612. } else {
  613. otherCount += r.IPCount
  614. }
  615. }
  616. // 将其他国家添加到过滤后的结果中
  617. if otherCount > 0 {
  618. p := 100.00 - math.Round(otherPercentage*100)/100
  619. filteredResults = append(filteredResults, &entity.UserCountryDetail{
  620. Country: "other",
  621. IPCount: otherCount,
  622. Percentage: math.Round(p*100) / 100,
  623. })
  624. }
  625. return &entity.UserCountryResp{
  626. Details: filteredResults,
  627. }, nil
  628. }
  629. // FindUserRetention UserRetentionResp 用户留存率
  630. // 1. 获取指定日期范围内的注册用户数量
  631. // 2. 获取指定日期范围内的活跃用户数量
  632. // 3. 计算留存率
  633. // 4. 返回结果
  634. func (s *Synthesis) FindUserRetention(req entity.UserRetentionReq) (*entity.UserRetentionResp, *code.Result) {
  635. playerPreserve := models.GetPlayerPreserve(req.StartTime, req.EndTime)
  636. // 查询数据
  637. var results []*entity.UserRetentionDetail
  638. for key, v := range playerPreserve {
  639. var retention entity.Retention
  640. for _, vv := range v {
  641. if vv.ID == 1 {
  642. retention.Day1 = entity.DayRetention{
  643. LoggedIn: vv.Ratio,
  644. LoginDate: key,
  645. }
  646. }
  647. if vv.ID == 3 {
  648. retention.Day3 = entity.DayRetention{
  649. LoggedIn: vv.Ratio,
  650. LoginDate: key,
  651. }
  652. }
  653. if vv.ID == 7 {
  654. retention.Day7 = entity.DayRetention{
  655. LoggedIn: vv.Ratio,
  656. LoginDate: key,
  657. }
  658. }
  659. if vv.ID == 14 {
  660. retention.Day14 = entity.DayRetention{
  661. LoggedIn: vv.Ratio,
  662. LoginDate: key,
  663. }
  664. }
  665. if vv.ID == 30 {
  666. retention.Day30 = entity.DayRetention{
  667. LoggedIn: vv.Ratio,
  668. LoginDate: key,
  669. }
  670. }
  671. }
  672. results = append(results, &entity.UserRetentionDetail{
  673. RegistrationDate: key,
  674. RetentionData: retention,
  675. })
  676. }
  677. return &entity.UserRetentionResp{
  678. Details: results,
  679. }, nil
  680. }
  681. // FindUserLevel 用户等级统计
  682. func (s *Synthesis) FindUserLevel() (*entity.UserLevelCountResp, *code.Result) {
  683. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  684. defer cancel()
  685. collection := mdb.MDB.Collection(constant.CNamePlayerLevelStat)
  686. // 查询所有文档
  687. cursor, err := collection.Find(ctx, bson.M{})
  688. if err != nil {
  689. mhayaLogger.Warnf("FindUserLevel Find error:%v", err)
  690. return nil, common.NewResult(code.InternalError)
  691. }
  692. defer cursor.Close(ctx)
  693. var results []*entity.UserLevelCountDetail
  694. var reData []*models.PlayerLevelStat
  695. for cursor.Next(ctx) {
  696. var result models.PlayerLevelStat
  697. if err := cursor.Decode(&result); err != nil {
  698. mhayaLogger.Warnf("FindUserLevel Decode error:%v", err)
  699. return nil, common.NewResult(code.InternalError)
  700. }
  701. reData = append(reData, &result)
  702. }
  703. if err := cursor.Err(); err != nil {
  704. mhayaLogger.Warnf("FindUserLevel cursor error:%v", err)
  705. return nil, common.NewResult(code.InternalError)
  706. }
  707. dataMap := make(map[string]int)
  708. for _, v := range reData {
  709. for key, v1 := range v.ServerLevel {
  710. dataMap[key] += v1
  711. }
  712. }
  713. keys := make([]string, 0, len(dataMap))
  714. for k := range dataMap {
  715. keys = append(keys, k)
  716. }
  717. sort.Strings(keys)
  718. for _, v := range keys {
  719. results = append(results, &entity.UserLevelCountDetail{
  720. Level: cast.ToInt(v),
  721. UserCount: dataMap[v],
  722. })
  723. }
  724. return &entity.UserLevelCountResp{
  725. Details: results,
  726. }, nil
  727. }
  728. func (s *Synthesis) InsertRecord(param model.UserOperationLog) {
  729. mhayaLogger.Warnf("InsertRecord param:%#v", param)
  730. record := new(model.UserOperationLog)
  731. collection := mdb.MDB.Collection(record.TableName())
  732. insertData := bson.M{}
  733. insertData["user_name"] = param.Username
  734. insertData["role_id"] = param.RoleId
  735. insertData["url"] = param.Path
  736. insertData["method"] = param.Method
  737. insertData["status_code"] = param.StatusCode
  738. insertData["dur"] = param.Dur
  739. insertData["client_ip"] = param.ClientIP
  740. insertData["error_message"] = param.ErrorMessage
  741. insertData["created_at"] = mhayaTime.Now().Unix()
  742. _, err := collection.InsertOne(context.Background(), insertData)
  743. if err != nil {
  744. mhayaLogger.Warnf("InsertRecord InsertOne error:%v", err)
  745. return
  746. }
  747. }
  748. func (s *Synthesis) RecordList(req entity.RecordListReq) (*entity.RecordListResp, *code.Result) {
  749. page, pageSize := checkPageParam(req.Page, req.Size)
  750. var records []*eventmodels.BackendOperationEventContent
  751. where := &eventmodels.BackendOperationEventContent{
  752. UserBasic: eventmodels.UserBasic{
  753. UserName: req.UserName,
  754. },
  755. EventBasic: eventmodels.EventBasic{
  756. ServerId: s.nodeId,
  757. },
  758. RoleId: req.RoleId,
  759. Path: req.Path,
  760. StatusCode: req.StatusCode,
  761. ClientIP: req.ClientIP,
  762. }
  763. db := mdb.LogstashDB.Model(&eventmodels.BackendOperationEventContent{}).Where(where).Order("create_at")
  764. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  765. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  766. }
  767. if req.DurMin > 0 && req.DurMax > 0 && req.DurMin <= req.DurMax {
  768. db = db.Where("dur >= ? and dur <= ?", req.DurMin, req.DurMax)
  769. }
  770. pages := Paginate(db, page, pageSize)
  771. err := db.Scopes(pages.Limit).Find(&records).Error
  772. if err != nil && err != gorm.ErrRecordNotFound {
  773. mhayaLogger.Warnf("RecordList Find error:%v", err)
  774. return nil, common.NewResult(code.InternalError)
  775. }
  776. var results []*entity.RecordListDetail
  777. for _, v := range records {
  778. results = append(results, &entity.RecordListDetail{
  779. Username: v.UserName,
  780. RoleId: v.RoleId,
  781. Path: v.Path,
  782. Method: v.Method,
  783. StatusCode: v.StatusCode,
  784. Dur: v.Dur,
  785. ClientIP: v.ClientIP,
  786. ErrorMessage: v.ErrorMessage,
  787. CreatedAt: v.CreateAt,
  788. })
  789. }
  790. return &entity.RecordListResp{
  791. Details: results,
  792. Total: pages.Count,
  793. }, nil
  794. }
  795. // 转盘统计
  796. func (s *Synthesis) Turntable(req entity.TurntableReq) (*entity.TurntableResp, *code.Result) {
  797. page, pageSize := checkPageParam(req.Page, req.Size)
  798. var records []*eventmodels.TurntableEventContent
  799. where := &eventmodels.TurntableEventContent{
  800. UserBasic: eventmodels.UserBasic{
  801. UserId: req.UserName,
  802. },
  803. }
  804. db := mdb.LogstashDB.Model(&eventmodels.TurntableEventContent{}).Where(where).Order("create_at")
  805. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  806. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  807. }
  808. pages := Paginate(db, page, pageSize)
  809. err := db.Scopes(pages.Limit).Find(&records).Error
  810. if err != nil && err != gorm.ErrRecordNotFound {
  811. mhayaLogger.Warnf("Turntable Find error:%v", err)
  812. return nil, common.NewResult(code.InternalError)
  813. }
  814. var results []*entity.TurntableDetail
  815. for _, v := range records {
  816. results = append(results, &entity.TurntableDetail{
  817. UserName: v.UserName,
  818. OpenId: v.TgId,
  819. TurntableExecutionTime: v.CreateAt,
  820. PrizeName: v.PrizeName,
  821. PrizeNumber: v.PrizeNumber,
  822. })
  823. }
  824. return &entity.TurntableResp{
  825. Details: results,
  826. Total: pages.Count,
  827. }, nil
  828. }
  829. // 资产统计
  830. func (s *Synthesis) Assets(req entity.AssetsReq) (*entity.AssetsResp, *code.Result) {
  831. page, pageSize := checkPageParam(req.Page, req.Size)
  832. playerMgr := NewPlayerManage()
  833. // 根据条件查询
  834. if req.UserName != "" {
  835. registerRecord, codeResult := playerMgr.GetRegisterRecord(req.UserName)
  836. if codeResult != nil {
  837. mhayaLogger.Warnf("Assets GetRegisterRecord error:%v", codeResult)
  838. return nil, codeResult
  839. }
  840. if registerRecord == nil {
  841. mhayaLogger.Warnf("Assets registerRecord:%v", registerRecord)
  842. return nil, nil
  843. }
  844. var results []*entity.AssetsDetail
  845. detail, codeResult := s.getAsset(req.UserName)
  846. if codeResult != nil {
  847. mhayaLogger.Warnf("Assets getAsset error:%v", codeResult)
  848. return nil, codeResult
  849. }
  850. results = append(results, detail)
  851. return &entity.AssetsResp{
  852. Details: results,
  853. Total: 1,
  854. }, nil
  855. }
  856. // 查询列表
  857. listResp, codeResult := playerMgr.List(context.Background(), entity.PlayerListReq{
  858. Page: page,
  859. Size: pageSize,
  860. })
  861. if codeResult != nil {
  862. mhayaLogger.Warnf("Assets List error:%v", codeResult)
  863. return nil, codeResult
  864. }
  865. var results []*entity.AssetsDetail
  866. for _, detail := range listResp.Details {
  867. ret, codeResult := s.getAsset(detail.UserName)
  868. if codeResult != nil {
  869. mhayaLogger.Warnf("Assets getAsset error:%v", codeResult)
  870. return nil, codeResult
  871. }
  872. results = append(results, ret)
  873. }
  874. count, codeResult := playerMgr.GetPlayerTotalCount()
  875. if codeResult != nil {
  876. mhayaLogger.Warnf("Assets GetPlayerTotalCount codeResult:%v", codeResult)
  877. return nil, codeResult
  878. }
  879. return &entity.AssetsResp{
  880. Details: results,
  881. Total: int64(count),
  882. }, nil
  883. }
  884. func (s *Synthesis) getAsset(userName string) (*entity.AssetsDetail, *code.Result) {
  885. var totalUsdtAmount int64
  886. where := &eventmodels.AssetsEventContent{
  887. UserBasic: eventmodels.UserBasic{
  888. UserId: userName,
  889. },
  890. EventBasic: eventmodels.EventBasic{
  891. ServerId: s.nodeId,
  892. },
  893. }
  894. where.Currency = string(constant.UsdtCurrency)
  895. err := mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &totalUsdtAmount).Error
  896. if err != nil {
  897. mhayaLogger.Warnf("getAsset Pluck UsdtCurrency error:%v", err)
  898. return nil, common.NewResult(code.InternalError)
  899. }
  900. var totalTonAmount int64
  901. where.Currency = string(constant.TonCurrency)
  902. err = mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &totalTonAmount).Error
  903. if err != nil {
  904. mhayaLogger.Warnf("getAsset Pluck TonCurrency error:%v", err)
  905. return nil, common.NewResult(code.InternalError)
  906. }
  907. var totalPointsAmount int64
  908. where.Currency = string(constant.PointsCurrency)
  909. err = mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &totalPointsAmount).Error
  910. if err != nil {
  911. mhayaLogger.Warnf("getAsset Pluck PointsCurrency error:%v", err)
  912. return nil, common.NewResult(code.InternalError)
  913. }
  914. return &entity.AssetsDetail{
  915. UserName: userName,
  916. TonValue: cutils.QuoInt64ByRatioToFloat64(totalTonAmount, constant.MoneyRatio),
  917. UsdtValue: cutils.QuoInt64ByRatioToFloat64(totalUsdtAmount, constant.MoneyRatio),
  918. PointsValue: totalPointsAmount,
  919. StatisticalDate: mhayaTime.Now().Unix(),
  920. }, nil
  921. }
  922. // 资产变动记录
  923. func (s *Synthesis) AssetsRecord(req entity.AssetsRecordReq) (*entity.AssetsRecordResp, *code.Result) {
  924. page, pageSize := checkPageParam(req.Page, req.Size)
  925. if req.Currency != "" && !constant.CurrencyValid(req.Currency) {
  926. mhayaLogger.Warnf("AssetsRecord unknow currency:%v", req.Currency)
  927. return nil, common.NewResult(code.ParamError)
  928. }
  929. if req.OperationType != "" && req.OperationType != string(constant.IncreaseOp) && req.OperationType != string(constant.DecreaseOp) {
  930. mhayaLogger.Warnf("AssetsRecord unknow OperationType:%v", req.OperationType)
  931. return nil, common.NewResult(code.ParamError)
  932. }
  933. var records []*eventmodels.AssetsEventContent
  934. where := &eventmodels.AssetsEventContent{
  935. UserBasic: eventmodels.UserBasic{
  936. UserId: req.UserName,
  937. },
  938. EventBasic: eventmodels.EventBasic{
  939. ServerId: s.nodeId,
  940. },
  941. OperationType: req.OperationType,
  942. Currency: req.Currency,
  943. Amount: req.Amount,
  944. }
  945. db := mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Order("create_at")
  946. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  947. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  948. }
  949. pages := Paginate(db, page, pageSize)
  950. err := db.Scopes(pages.Limit).Find(&records).Error
  951. if err != nil && err != gorm.ErrRecordNotFound {
  952. mhayaLogger.Warnf("Invite Find error:%v", err)
  953. return nil, common.NewResult(code.InternalError)
  954. }
  955. var results []*entity.AssetsRecordDetail
  956. for _, v := range records {
  957. results = append(results, &entity.AssetsRecordDetail{
  958. UserName: v.UserName,
  959. AssetType: string(v.Currency),
  960. AssetsGrowth: v.Amount,
  961. AcquisitionTime: v.CreateAt,
  962. SourceOfAssetAcquisition: v.Reason,
  963. })
  964. }
  965. return &entity.AssetsRecordResp{
  966. Details: results,
  967. Total: pages.Count,
  968. }, nil
  969. }
  970. // 邀请统计
  971. func (s *Synthesis) Invite(req entity.InviteReq) (*entity.InviteResp, *code.Result) {
  972. page, pageSize := checkPageParam(req.Page, req.Size)
  973. var records []*eventmodels.InviteEventContent
  974. where := &eventmodels.InviteEventContent{
  975. UserBasic: eventmodels.UserBasic{
  976. UserId: req.UserName,
  977. },
  978. }
  979. db := mdb.LogstashDB.Model(&eventmodels.InviteEventContent{}).Where(where).Order("create_at")
  980. pages := Paginate(db, page, pageSize)
  981. err := db.Scopes(pages.Limit).Find(&records).Error
  982. if err != nil && err != gorm.ErrRecordNotFound {
  983. mhayaLogger.Warnf("Invite Find error:%v", err)
  984. return nil, common.NewResult(code.InternalError)
  985. }
  986. var results []*entity.InviteDetail
  987. for _, v := range records {
  988. results = append(results, &entity.InviteDetail{
  989. UserName: v.UserName,
  990. OpenId: v.TgId,
  991. InviteID: v.UserId,
  992. IfInviteVIP: v.IfInviteUserTgVip,
  993. InviteTime: v.CreateAt,
  994. SuccessfulInvitationsLV2: int64(v.InviteUserLevel2),
  995. CumulativesuccessfulInvitations: int64(v.ContinuousInviteUserCount),
  996. InvitationChannel: v.Channel,
  997. })
  998. }
  999. return &entity.InviteResp{
  1000. Details: results,
  1001. Total: pages.Count,
  1002. }, nil
  1003. }
  1004. // 活跃统计
  1005. func (s *Synthesis) Active(req entity.ActiveReq) (*entity.ActiveResp, *code.Result) {
  1006. page, pageSize := checkPageParam(req.Page, req.Size)
  1007. playerMgr := NewPlayerManage()
  1008. // 根据条件查询
  1009. if req.UserName != "" {
  1010. detail, codeResult := s.getActive(req.UserName)
  1011. if codeResult != nil {
  1012. mhayaLogger.Warnf("Active getActive error:%v", codeResult)
  1013. return nil, codeResult
  1014. }
  1015. if detail == nil {
  1016. return nil, nil
  1017. }
  1018. var results []*entity.ActiveDetail
  1019. results = append(results, detail)
  1020. return &entity.ActiveResp{
  1021. Details: results,
  1022. Total: 1,
  1023. }, nil
  1024. }
  1025. // 查询列表
  1026. listResp, codeResult := playerMgr.List(context.Background(), entity.PlayerListReq{
  1027. Page: page,
  1028. Size: pageSize,
  1029. })
  1030. if codeResult != nil {
  1031. mhayaLogger.Warnf("Active List error:%v", codeResult)
  1032. return nil, codeResult
  1033. }
  1034. var results []*entity.ActiveDetail
  1035. for _, detail := range listResp.Details {
  1036. detail, codeResult := s.getActive(detail.UserName)
  1037. if codeResult != nil {
  1038. mhayaLogger.Warnf("Active getActive error:%v", codeResult)
  1039. return nil, codeResult
  1040. }
  1041. if detail != nil {
  1042. results = append(results, detail)
  1043. }
  1044. }
  1045. count, codeResult := playerMgr.GetPlayerTotalCount()
  1046. if codeResult != nil {
  1047. mhayaLogger.Warnf("Active GetPlayerTotalCount codeResult:%v", codeResult)
  1048. return nil, codeResult
  1049. }
  1050. return &entity.ActiveResp{
  1051. Details: results,
  1052. Total: count,
  1053. }, nil
  1054. }
  1055. func (s *Synthesis) getActive(userName string) (*entity.ActiveDetail, *code.Result) {
  1056. playerMgr := NewPlayerManage()
  1057. // 获取注册记录
  1058. registerRecord, codeResult := playerMgr.GetRegisterRecord(userName)
  1059. if codeResult != nil {
  1060. mhayaLogger.Warnf("getActive GetRegisterRecord error:%v", codeResult)
  1061. return nil, codeResult
  1062. }
  1063. if registerRecord == nil {
  1064. mhayaLogger.Warnf("getActive registerRecord:%v", registerRecord)
  1065. return nil, nil
  1066. }
  1067. // 获取最新的用户登录记录
  1068. loginRecord, codeResult := playerMgr.GetPlayerLastestLoginRecord(userName)
  1069. if codeResult != nil {
  1070. mhayaLogger.Warnf("getActive GetPlayerLastestLoginRecord error:%v", codeResult)
  1071. return nil, codeResult
  1072. }
  1073. // 邀请成功人数
  1074. inviteCount, codeResult := playerMgr.GetSuccessfulInvitations(userName, 0)
  1075. if codeResult != nil {
  1076. mhayaLogger.Warnf("getActive GetSuccessfulInvitations error:%v", codeResult)
  1077. return nil, codeResult
  1078. }
  1079. // 提现次数
  1080. withdrawCount, codeResult := playerMgr.GetWithdrawalCount(userName, -1)
  1081. if codeResult != nil {
  1082. mhayaLogger.Warnf("getActive GetWithdrawalCount error:%v", codeResult)
  1083. return nil, codeResult
  1084. }
  1085. // 累计提现金额
  1086. // TODO 暂定USDT
  1087. cumulativeWithdrawalAmount, codeResult := playerMgr.GetCumulativeWithdrawalAmount(userName, string(constant.UsdtCurrency), -1)
  1088. if codeResult != nil {
  1089. mhayaLogger.Warnf("getActive GetCumulativeWithdrawalAmount error:%v", codeResult)
  1090. return nil, codeResult
  1091. }
  1092. // 转盘实际抽奖次数
  1093. turntableRuns, codeResult := playerMgr.GetTurntableRuns(userName)
  1094. if codeResult != nil {
  1095. mhayaLogger.Warnf("getActive GetTurntableRuns error:%v", codeResult)
  1096. return nil, codeResult
  1097. }
  1098. return &entity.ActiveDetail{
  1099. UserName: userName,
  1100. IfUserVip: func() bool {
  1101. if loginRecord != nil {
  1102. return loginRecord.IfTgVip
  1103. }
  1104. return false
  1105. }(),
  1106. CreateTime: registerRecord.CreateAt,
  1107. LastLoginTime: func() int64 {
  1108. if loginRecord != nil {
  1109. return loginRecord.CreateAt
  1110. }
  1111. return 0
  1112. }(),
  1113. MaxSuccessions: func() int {
  1114. if loginRecord != nil {
  1115. return loginRecord.ContinuousDaysMax
  1116. }
  1117. return 0
  1118. }(),
  1119. SuccessfulInvitations: inviteCount,
  1120. Withdrawals: withdrawCount,
  1121. CumulativeWithdrawalAmount: cumulativeWithdrawalAmount,
  1122. TurntableRuns: turntableRuns,
  1123. }, nil
  1124. }
  1125. // 任务完成度统计
  1126. func (s *Synthesis) TaskCompletion(req entity.TaskCompletionReq) (*entity.TaskCompletionResp, *code.Result) {
  1127. page, pageSize := checkPageParam(req.Page, req.Size)
  1128. playerMgr := NewPlayerManage()
  1129. // 根据条件查询
  1130. if req.UserName != "" {
  1131. detail, codeResult := s.getTaskCompletionDetail(req.UserName)
  1132. if codeResult != nil {
  1133. mhayaLogger.Warnf("TaskCompletion getTaskCompletionDetail error:%v", codeResult)
  1134. return nil, codeResult
  1135. }
  1136. if detail == nil {
  1137. return nil, nil
  1138. }
  1139. var results []*entity.TaskCompletionDetail
  1140. results = append(results, detail)
  1141. return &entity.TaskCompletionResp{
  1142. Details: results,
  1143. Total: 1,
  1144. }, nil
  1145. }
  1146. // 查询列表
  1147. listResp, codeResult := playerMgr.List(context.Background(), entity.PlayerListReq{
  1148. Page: page,
  1149. Size: pageSize,
  1150. })
  1151. if codeResult != nil {
  1152. mhayaLogger.Warnf("TaskCompletion List error:%v", codeResult)
  1153. return nil, codeResult
  1154. }
  1155. var results []*entity.TaskCompletionDetail
  1156. for _, detail := range listResp.Details {
  1157. ret, codeResult := s.getTaskCompletionDetail(detail.UserName)
  1158. if codeResult != nil {
  1159. mhayaLogger.Warnf("TaskCompletion getTaskCompletionDetail error:%v", codeResult)
  1160. return nil, codeResult
  1161. }
  1162. if detail != nil {
  1163. results = append(results, ret)
  1164. }
  1165. }
  1166. count, codeResult := playerMgr.GetPlayerTotalCount()
  1167. if codeResult != nil {
  1168. mhayaLogger.Warnf("TaskCompletion GetPlayerTotalCount codeResult:%v", codeResult)
  1169. return nil, codeResult
  1170. }
  1171. return &entity.TaskCompletionResp{
  1172. Details: results,
  1173. Total: int64(count),
  1174. }, nil
  1175. }
  1176. func (s *Synthesis) getTaskCompletionDetail(userName string) (*entity.TaskCompletionDetail, *code.Result) {
  1177. playerMgr := NewPlayerManage()
  1178. // 获取注册记录
  1179. registerRecord, codeResult := playerMgr.GetRegisterRecord(userName)
  1180. if codeResult != nil {
  1181. mhayaLogger.Warnf("getTaskCompletionDetail GetRegisterRecord error:%v", codeResult)
  1182. return nil, codeResult
  1183. }
  1184. if registerRecord == nil {
  1185. mhayaLogger.Warnf("getTaskCompletionDetail registerRecord:%v", registerRecord)
  1186. return nil, nil
  1187. }
  1188. // 获取用户加入频道的记录
  1189. joinTgRecord, codeResult := playerMgr.GetJoinRecord(userName, string(constant.TGJoinName))
  1190. if codeResult != nil {
  1191. mhayaLogger.Warnf("getTaskCompletionDetail GetJoinRecord error:%v", codeResult)
  1192. return nil, codeResult
  1193. }
  1194. joinDcRecord, codeResult := playerMgr.GetJoinRecord(userName, string(constant.DCJoinName))
  1195. if codeResult != nil {
  1196. mhayaLogger.Warnf("getTaskCompletionDetail GetJoinRecord error:%v", codeResult)
  1197. return nil, codeResult
  1198. }
  1199. // 获取用户关注的记录
  1200. followTwitterRecord, codeResult := playerMgr.GetFollowRecord(userName, string(constant.TwitterFollowName))
  1201. if codeResult != nil {
  1202. mhayaLogger.Warnf("getTaskCompletionDetail GetFollowRecord error:%v", codeResult)
  1203. return nil, codeResult
  1204. }
  1205. followYoutubeRecord, codeResult := playerMgr.GetFollowRecord(userName, string(constant.YoutubeFollowName))
  1206. if codeResult != nil {
  1207. mhayaLogger.Warnf("getTaskCompletionDetail GetFollowRecord error:%v", codeResult)
  1208. return nil, codeResult
  1209. }
  1210. // 获取最新的用户信息更新记录
  1211. updateRecord, codeResult := playerMgr.GetPlayerLastestUpdateRecord(userName)
  1212. if codeResult != nil {
  1213. mhayaLogger.Warnf("getTaskCompletionDetail GetPlayerLastestUpdateRecord error:%v", codeResult)
  1214. return nil, codeResult
  1215. }
  1216. // 邀请成功人数
  1217. inviteCount, codeResult := playerMgr.GetSuccessfulInvitations(userName, 0)
  1218. if codeResult != nil {
  1219. mhayaLogger.Warnf("getTaskCompletionDetail GetSuccessfulInvitations error:%v", codeResult)
  1220. return nil, codeResult
  1221. }
  1222. return &entity.TaskCompletionDetail{
  1223. UserName: userName,
  1224. DicePerDay: 0, // TODO 每日骰子数量
  1225. IsJoinTgChannel: func() bool {
  1226. return joinTgRecord != nil
  1227. }(),
  1228. IsFollowTwitter: func() bool {
  1229. return followTwitterRecord != nil
  1230. }(),
  1231. IsJoinDcChannel: func() bool {
  1232. return joinDcRecord != nil
  1233. }(),
  1234. IsBindTwitter: func() bool {
  1235. if updateRecord != nil {
  1236. return updateRecord.XId == ""
  1237. }
  1238. return false
  1239. }(),
  1240. IsFollowYoutube: func() bool {
  1241. return followYoutubeRecord != nil
  1242. }(),
  1243. InviteNumber: inviteCount,
  1244. }, nil
  1245. }
  1246. // 用户行为检测
  1247. func (s *Synthesis) BehaviorMonitoring(req entity.BehaviorMonitoringReq) (*entity.BehaviorMonitoringResp, *code.Result) {
  1248. page, pageSize := checkPageParam(req.Page, req.Size)
  1249. var records []*eventmodels.DiceEventContent
  1250. where := &eventmodels.DiceEventContent{
  1251. UserBasic: eventmodels.UserBasic{
  1252. UserId: req.UserName,
  1253. },
  1254. }
  1255. db := mdb.LogstashDB.Model(&eventmodels.DiceEventContent{}).Where(where).Order("create_at")
  1256. pages := Paginate(db, page, pageSize)
  1257. err := db.Scopes(pages.Limit).Find(&records).Error
  1258. if err != nil && err != gorm.ErrRecordNotFound {
  1259. mhayaLogger.Warnf("BehaviorMonitoring Find error:%v", err)
  1260. return nil, common.NewResult(code.InternalError)
  1261. }
  1262. var results []*entity.BehaviorMonitoringDetail
  1263. for _, v := range records {
  1264. results = append(results, &entity.BehaviorMonitoringDetail{
  1265. UserName: v.UserName,
  1266. DieExecutionTime: v.CreateAt,
  1267. })
  1268. }
  1269. return &entity.BehaviorMonitoringResp{
  1270. Details: results,
  1271. Total: pages.Count,
  1272. }, nil
  1273. }