synthesis.go 42 KB

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