synthesis.go 42 KB

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