synthesis.go 41 KB

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