synthesis.go 32 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141
  1. package service
  2. import (
  3. "context"
  4. "math"
  5. "time"
  6. mhayaTime "github.com/mhaya/extend/time"
  7. "github.com/mhaya/game/game_cluster/internal/code"
  8. "github.com/mhaya/game/game_cluster/internal/constant"
  9. "github.com/mhaya/game/game_cluster/internal/mdb"
  10. "github.com/mhaya/game/game_cluster/internal/mdb/eventmodels"
  11. "github.com/mhaya/game/game_cluster/internal/mdb/models"
  12. "github.com/mhaya/game/game_cluster/nodes/webadmin/common"
  13. "github.com/mhaya/game/game_cluster/nodes/webadmin/entity"
  14. "github.com/mhaya/game/game_cluster/nodes/webadmin/model"
  15. mhayaLogger "github.com/mhaya/logger"
  16. "github.com/spf13/cast"
  17. "go.mongodb.org/mongo-driver/bson"
  18. "go.mongodb.org/mongo-driver/bson/primitive"
  19. "go.mongodb.org/mongo-driver/mongo"
  20. "go.mongodb.org/mongo-driver/mongo/options"
  21. "gorm.io/gorm"
  22. )
  23. type Synthesis struct {
  24. nodeId string
  25. db *mongo.Database
  26. }
  27. func NewSynthesis(nodeId string) *Synthesis {
  28. return &Synthesis{
  29. nodeId: nodeId,
  30. db: mdb.MDB,
  31. }
  32. }
  33. // 综合统计
  34. func (s *Synthesis) Overview(req entity.OverviewReq) (*entity.OverviewResp, *code.Result) {
  35. page, pageSize := checkPageParam(req.Page, req.Size)
  36. // 构建查询条件
  37. filter := bson.M{}
  38. if req.UserName != "" {
  39. filter["userName"] = req.UserName
  40. }
  41. if req.OpenId != "" {
  42. filter["openId"] = req.OpenId
  43. }
  44. // 设置分页选项
  45. findOptions := options.Find()
  46. findOptions.SetSkip(int64((page - 1) * pageSize))
  47. findOptions.SetLimit(int64(pageSize))
  48. findOptions.SetSort(bson.D{{"createTime", -1}})
  49. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  50. defer cancel()
  51. collection := mdb.MDB.Collection(constant.CNamePlayer)
  52. // 查询数据
  53. var results []*entity.OverviewDetail
  54. cursor, err := collection.Find(ctx, filter, findOptions)
  55. if err != nil {
  56. mhayaLogger.Warnf("Overview Find error:%v", err)
  57. return nil, common.NewResult(code.InternalError)
  58. }
  59. defer cursor.Close(ctx)
  60. // 解析结果
  61. for cursor.Next(ctx) {
  62. var result entity.OverviewDetail
  63. if err := cursor.Decode(&result); err != nil {
  64. mhayaLogger.Warnf("Overview Decode error:%v", err)
  65. return nil, common.NewResult(code.InternalError)
  66. }
  67. results = append(results, &result)
  68. }
  69. if err := cursor.Err(); err != nil {
  70. mhayaLogger.Warnf("Overview cursor error:%v", err)
  71. return nil, common.NewResult(code.InternalError)
  72. }
  73. // TODO 综合统计相关信息
  74. // TODO 获取总数total
  75. var count int64
  76. return &entity.OverviewResp{
  77. Details: results,
  78. Total: count,
  79. }, nil
  80. }
  81. // 统计用户相关信息
  82. func (s *Synthesis) UserList(req entity.UserListReq) (*entity.UserListResp, *code.Result) {
  83. page, pageSize := checkPageParam(req.Page, req.Size)
  84. // 构建查询条件
  85. filter := bson.M{}
  86. if req.UserName != "" {
  87. filter["userName"] = req.UserName
  88. }
  89. if req.OpenId != "" {
  90. filter["openId"] = req.OpenId
  91. }
  92. // 设置分页选项
  93. findOptions := options.Find()
  94. findOptions.SetSkip(int64((page - 1) * pageSize))
  95. findOptions.SetLimit(int64(pageSize))
  96. findOptions.SetSort(bson.D{{"createTime", -1}})
  97. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  98. defer cancel()
  99. collection := mdb.MDB.Collection(constant.CNamePlayer)
  100. // 查询数据
  101. var results []*entity.UserListDetail
  102. cursor, err := collection.Find(ctx, filter, findOptions)
  103. if err != nil {
  104. mhayaLogger.Warnf("UserList Find error:%v", err)
  105. return nil, common.NewResult(code.InternalError)
  106. }
  107. defer cursor.Close(ctx)
  108. // 解析结果
  109. for cursor.Next(ctx) {
  110. var result entity.UserListDetail
  111. if err := cursor.Decode(&result); err != nil {
  112. mhayaLogger.Warnf("UserList Decode error:%v", err)
  113. return nil, common.NewResult(code.InternalError)
  114. }
  115. results = append(results, &result)
  116. }
  117. if err := cursor.Err(); err != nil {
  118. mhayaLogger.Warnf("UserList cursor error:%v", err)
  119. return nil, common.NewResult(code.InternalError)
  120. }
  121. // TODO UserList 统计用户相关信息
  122. // TODO 获取总数total
  123. var count int64
  124. return &entity.UserListResp{
  125. Details: results,
  126. Total: count,
  127. }, nil
  128. }
  129. func (s *Synthesis) FindMDBUserLogDaily(req entity.UserLogDailyReq) (*entity.UserLogDailyResp, *code.Result) {
  130. page, pageSize := checkPageParam(req.Page, req.Size)
  131. // 定义上下文
  132. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  133. defer cancel()
  134. // 指定集合
  135. collection := mdb.MDB.Collection(constant.CNamePlayerDailyRecord)
  136. // 构建查询条件 - 如果查询值为空那就不添加查询条件
  137. filter := bson.M{}
  138. if req.StartTime != 0 {
  139. filter["daily"] = bson.M{
  140. "$gte": req.StartTime,
  141. "$lte": req.EndTime,
  142. }
  143. }
  144. if req.Platform != "" && req.Platform != "all" {
  145. filter["platform"] = req.Platform
  146. }
  147. if req.Channel != "" {
  148. filter["channel"] = req.Channel
  149. }
  150. // 计算总数
  151. count, err := collection.CountDocuments(ctx, filter)
  152. if err != nil {
  153. mhayaLogger.Warnf("FindMDBUserLogDaily CountDocuments error:%v", err)
  154. return nil, common.NewResult(code.InternalError)
  155. }
  156. // 执行查询
  157. opts := options.Find()
  158. opts.SetSkip(int64((page - 1) * pageSize))
  159. opts.SetLimit(int64(pageSize))
  160. opts.SetSort(bson.D{{"daily", -1}})
  161. cursor, err := collection.Find(ctx, filter, opts)
  162. if err != nil {
  163. mhayaLogger.Warnf("FindMDBUserLogDaily Find error:%v", err)
  164. return nil, common.NewResult(code.InternalError)
  165. }
  166. defer cursor.Close(ctx)
  167. // 解析查询结果
  168. var results []*entity.UserLogDailyDetail
  169. for cursor.Next(ctx) {
  170. var result *entity.UserLogDailyDetail
  171. err := cursor.Decode(&result)
  172. if err != nil {
  173. mhayaLogger.Warnf("FindMDBUserLogDaily Decode error:%v", err)
  174. return nil, common.NewResult(code.InternalError)
  175. }
  176. results = append(results, result)
  177. }
  178. // 同一天的数据全部放到platform= ALl 且数据累加
  179. // 如果没有platform=all 的那就新增一个 同一个时间段只能有一个 platform=all
  180. // 将同一天的数据累加到platform=all的记录中
  181. allPlatformRecordMap := make(map[int64]*entity.UserLogDailyDetail)
  182. for _, result := range results {
  183. allPlatformRecordMap[result.Timestamp] = &entity.UserLogDailyDetail{
  184. Timestamp: result.Timestamp,
  185. Platform: "all",
  186. }
  187. }
  188. for _, v := range results {
  189. if v.Timestamp == allPlatformRecordMap[v.Timestamp].Timestamp {
  190. allPlatformRecordMap[v.Timestamp].Registered += v.Registered
  191. allPlatformRecordMap[v.Timestamp].LoggedIn += v.LoggedIn
  192. allPlatformRecordMap[v.Timestamp].NewActive += v.NewActive
  193. allPlatformRecordMap[v.Timestamp].OldActive += v.OldActive
  194. allPlatformRecordMap[v.Timestamp].TotalPoints += v.TotalPoints
  195. allPlatformRecordMap[v.Timestamp].UProduced += v.UProduced
  196. allPlatformRecordMap[v.Timestamp].UCashout += v.UCashout
  197. allPlatformRecordMap[v.Timestamp].NewLogin += v.NewLogin
  198. allPlatformRecordMap[v.Timestamp].OldLogin += v.OldLogin
  199. }
  200. }
  201. // 替换原有结果
  202. for _, record := range allPlatformRecordMap {
  203. results = append(results, record)
  204. }
  205. return &entity.UserLogDailyResp{
  206. Details: results,
  207. Total: count,
  208. }, nil
  209. }
  210. // FindWithdrawal 根据请求查询提现记录
  211. func (s *Synthesis) FindWithdrawal(req entity.UserWithdrawalReq) (*entity.UserWithdrawalResp, *code.Result) {
  212. page, pageSize := checkPageParam(req.Page, req.Size)
  213. if !constant.CurrencyValid(req.Currency) {
  214. mhayaLogger.Warnf("FindWithdrawal unknow currency:%v", req.Currency)
  215. return nil, common.NewResult(code.ParamError)
  216. }
  217. var records []*eventmodels.UserWithdrawEventContent
  218. where := &eventmodels.UserWithdrawEventContent{
  219. UserBasic: eventmodels.UserBasic{
  220. UserId: req.UserName,
  221. UserName: req.NickName,
  222. },
  223. EventBasic: eventmodels.EventBasic{
  224. ServerId: s.nodeId,
  225. },
  226. WithdrawId: req.ID,
  227. Currency: constant.CurrencyTypeStr(req.Currency),
  228. Address: req.Address,
  229. State: req.State,
  230. WithdrawalState: req.Withdrawal,
  231. }
  232. db := mdb.LogstashDB.Model(&eventmodels.UserWithdrawEventContent{}).Where(where).Order("create_at")
  233. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  234. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  235. }
  236. if req.AmountMin > 0 && req.AmountMax > 0 && req.AmountMin <= req.AmountMax {
  237. db = db.Where("amount >= ? and amount <= ?", req.AmountMin, req.AmountMax)
  238. }
  239. if req.AfterAmountMin > 0 && req.AfterAmountMax > 0 && req.AfterAmountMin <= req.AfterAmountMax {
  240. db = db.Where("after_balance >= ? and after_balance <= ?", req.AfterAmountMin, req.AfterAmountMax)
  241. }
  242. pages := Paginate(db, page, pageSize)
  243. err := db.Scopes(pages.Limit).Find(&records).Error
  244. if err != nil && err != gorm.ErrRecordNotFound {
  245. mhayaLogger.Warnf("FindWithdrawal Find error:%v", err)
  246. return nil, common.NewResult(code.InternalError)
  247. }
  248. var results []*entity.UserWithdrawalDetail
  249. for _, v := range records {
  250. results = append(results, &entity.UserWithdrawalDetail{
  251. Id: v.WithdrawId,
  252. UserName: v.UserId,
  253. NickName: v.UserName,
  254. OpenId: v.TgId,
  255. Status: v.State,
  256. Reason: v.Reason,
  257. Withdrawal: v.WithdrawalState,
  258. Amount: int(v.Amount),
  259. AfterAmount: int(v.AfterBalance),
  260. Type: v.Currency,
  261. Address: v.Address,
  262. CreateAt: v.CreateAt,
  263. UpdateAt: v.CreateAt,
  264. })
  265. }
  266. return &entity.UserWithdrawalResp{
  267. Details: results,
  268. Total: pages.Count,
  269. }, nil
  270. }
  271. // 导出提现记录
  272. func (s *Synthesis) WithdrawalExport(req entity.UserWithdrawalExportReq) (*entity.UserWithdrawalResp, *code.Result) {
  273. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  274. defer cancel()
  275. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  276. // 构建过滤器
  277. filter := bson.M{}
  278. if req.UserName != "" {
  279. filter["userName"] = req.UserName
  280. }
  281. if req.NickName != "" {
  282. filter["nickName"] = req.NickName
  283. }
  284. if req.ID != "" {
  285. filter["_id"], _ = primitive.ObjectIDFromHex(req.ID)
  286. }
  287. if req.Address != "" {
  288. filter["address"] = req.Address
  289. }
  290. if req.State > 0 {
  291. filter["state"] = req.State
  292. }
  293. if req.Withdrawal > 0 {
  294. filter["withdrawal"] = req.Withdrawal
  295. }
  296. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  297. filter["createAt"] = bson.M{
  298. "$gte": req.StartTime,
  299. "$lte": req.EndTime,
  300. }
  301. }
  302. if req.AmountMin > 0 && req.AmountMax > 0 && req.AmountMin <= req.AmountMax {
  303. filter["amount"] = bson.M{
  304. "$gte": req.AmountMin,
  305. "$lte": req.AmountMax,
  306. }
  307. }
  308. if req.AfterAmountMin > 0 && req.AfterAmountMax > 0 && req.AfterAmountMin <= req.AfterAmountMax {
  309. filter["after_amount"] = bson.M{
  310. "$gte": req.AfterAmountMin,
  311. "$lte": req.AfterAmountMax,
  312. }
  313. }
  314. findOptions := options.Find()
  315. findOptions.SetSort(bson.D{{"createAt", -1}})
  316. // 获取总数total
  317. count, err := collection.CountDocuments(ctx, filter)
  318. if err != nil {
  319. mhayaLogger.Warnf("WithdrawalExportData CountDocuments error:%v", err)
  320. return nil, common.NewResult(code.InternalError)
  321. }
  322. // 查询数据
  323. var results []*entity.UserWithdrawalDetail
  324. cursor, err := collection.Find(ctx, filter, findOptions)
  325. if err != nil {
  326. mhayaLogger.Warnf("WithdrawalExportData Find error:%v", err)
  327. return nil, common.NewResult(code.InternalError)
  328. }
  329. defer cursor.Close(ctx)
  330. // 解析结果
  331. for cursor.Next(ctx) {
  332. var result entity.UserWithdrawalDetail
  333. if err := cursor.Decode(&result); err != nil {
  334. mhayaLogger.Warnf("WithdrawalExportData Decode error:%v", err)
  335. return nil, common.NewResult(code.InternalError)
  336. }
  337. results = append(results, &result)
  338. }
  339. if err := cursor.Err(); err != nil {
  340. mhayaLogger.Warnf("WithdrawalExportData cursor error:%v", err)
  341. return nil, common.NewResult(code.InternalError)
  342. }
  343. return &entity.UserWithdrawalResp{
  344. Details: results,
  345. Total: count,
  346. }, nil
  347. }
  348. // WithdrawalStatus 更新提现状态
  349. func (s *Synthesis) WithdrawalStatus(req entity.UserWithdrawalStatus) *code.Result {
  350. // ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  351. // defer cancel()
  352. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  353. // 更新条件
  354. updateCondition := bson.M{"userName": req.UserName}
  355. // 更新内容
  356. updateContent := bson.M{"$set": bson.M{"status": req.Status}}
  357. // 设置更新选项
  358. updateOptions := options.Update() // 设置 upsert 选项
  359. // 执行更新操作
  360. _, err := collection.UpdateOne(context.TODO(), updateCondition, updateContent, updateOptions)
  361. if err != nil {
  362. mhayaLogger.Warnf("WithdrawalStatus UpdateOne error:%v", err)
  363. return common.NewResult(code.InternalError)
  364. }
  365. return nil
  366. }
  367. // WithdrawalStatusBatch 更新提现状态
  368. func (s *Synthesis) WithdrawalStatusBatch(req entity.UserWithdrawalStatusBatch) *code.Result {
  369. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  370. if len(req.ID) == 0 {
  371. return common.NewResult(code.ParamError)
  372. }
  373. for _, id := range req.ID {
  374. objID := primitive.ObjectID{}
  375. objID, _ = primitive.ObjectIDFromHex(id)
  376. updateCondition := bson.M{"_id": objID}
  377. updateContent := bson.M{}
  378. withdrawal := models.CashOutRecord{}
  379. err := collection.FindOne(context.TODO(), updateCondition).Decode(&withdrawal)
  380. if err != nil {
  381. mhayaLogger.Warnf("WithdrawalStatusBatch FindOne error:%v", err)
  382. continue
  383. }
  384. if req.Withdrawal != 0 {
  385. if withdrawal.Status == 1 {
  386. updateContent = bson.M{"$set": bson.M{"withdrawal": req.Withdrawal}}
  387. } else {
  388. continue
  389. }
  390. }
  391. if req.Status > 0 {
  392. if withdrawal.Status != 0 {
  393. continue
  394. }
  395. updateContent = bson.M{"$set": bson.M{"status": req.Status}}
  396. }
  397. updateOptions := options.Update().SetUpsert(true)
  398. _, err = collection.UpdateOne(context.TODO(), updateCondition, updateContent, updateOptions)
  399. if err != nil {
  400. mhayaLogger.Warnf("WithdrawalStatusBatch UpdateOne error:%v", err)
  401. continue
  402. }
  403. }
  404. return nil
  405. }
  406. // FindUserCountryCount 查询用户国家分布
  407. //
  408. // 返回值为 UserCountryResp 的切片和错误。
  409. func (s *Synthesis) FindUserCountryCount() (*entity.UserCountryResp, *code.Result) {
  410. // 选择数据库和集合
  411. collection := mdb.MDB.Collection(constant.CNamePlayerCountryByIPStat)
  412. // 定义聚合管道
  413. // 定义聚合管道
  414. pipeline := []bson.D{
  415. {
  416. {Key: "$project", Value: bson.D{
  417. {Key: "playerRegisterCountry", Value: bson.D{{Key: "$objectToArray", Value: "$playerRegisterCountry"}}},
  418. }},
  419. },
  420. {
  421. {Key: "$unwind", Value: "$playerRegisterCountry"},
  422. },
  423. {
  424. {Key: "$group", Value: bson.D{
  425. {Key: "_id", Value: "$playerRegisterCountry.k"},
  426. {Key: "totalValue", Value: bson.D{{Key: "$sum", Value: "$playerRegisterCountry.v"}}},
  427. }},
  428. },
  429. {
  430. {Key: "$project", Value: bson.D{
  431. {Key: "_id", Value: 0},
  432. {Key: "countryKey", Value: "$_id"},
  433. {Key: "totalValue", Value: 1},
  434. }},
  435. },
  436. }
  437. // 执行聚合查询
  438. cursor, err := collection.Aggregate(context.TODO(), pipeline)
  439. if err != nil {
  440. mhayaLogger.Warnf("FindUserCountryCount Aggregate error:%v", err)
  441. return nil, common.NewResult(code.InternalError)
  442. }
  443. defer cursor.Close(context.TODO())
  444. // 遍历查询结果
  445. var results []bson.M
  446. if err := cursor.All(context.TODO(), &results); err != nil {
  447. mhayaLogger.Warnf("FindUserCountryCount All error:%v", err)
  448. return nil, common.NewResult(code.InternalError)
  449. }
  450. var totalIPCount int64
  451. var data []*entity.UserCountryDetail
  452. // 将结果转换为 UserCountryDetail
  453. for _, r := range results {
  454. var resp entity.UserCountryDetail
  455. resp.Country = r["countryKey"].(string)
  456. resp.IPCount = int(r["totalValue"].(int32))
  457. totalIPCount += int64(resp.IPCount)
  458. data = append(data, &resp)
  459. }
  460. for _, v := range data {
  461. // 保留小数点后两位
  462. v.Percentage = math.Round(float64(v.IPCount)/float64(totalIPCount)*10000) / 100
  463. }
  464. // 根据阈值过滤结果
  465. otherCount := 0
  466. otherPercentage := 0.00
  467. filteredResults := make([]*entity.UserCountryDetail, 0)
  468. threshold := 1.00
  469. for _, r := range data {
  470. if r.Percentage >= threshold {
  471. filteredResults = append(filteredResults, r)
  472. // 保留小数点后两位
  473. r.Percentage = math.Round(r.Percentage*100) / 100
  474. otherPercentage += r.Percentage
  475. } else {
  476. otherCount += r.IPCount
  477. }
  478. }
  479. // 将其他国家添加到过滤后的结果中
  480. if otherCount > 0 {
  481. p := 100.00 - math.Round(otherPercentage*100)/100
  482. filteredResults = append(filteredResults, &entity.UserCountryDetail{
  483. Country: "other",
  484. IPCount: otherCount,
  485. Percentage: math.Round(p*100) / 100,
  486. })
  487. }
  488. return &entity.UserCountryResp{
  489. Details: filteredResults,
  490. }, nil
  491. }
  492. // FindUserRetention UserRetentionResp 用户留存率
  493. // 1. 获取指定日期范围内的注册用户数量
  494. // 2. 获取指定日期范围内的活跃用户数量
  495. // 3. 计算留存率
  496. // 4. 返回结果
  497. func (s *Synthesis) FindUserRetention(req entity.UserRetentionReq) (*entity.UserRetentionResp, *code.Result) {
  498. playerPreserve := models.GetPlayerPreserve(req.StartTime, req.EndTime)
  499. // 查询数据
  500. var results []*entity.UserRetentionDetail
  501. for key, v := range playerPreserve {
  502. var retention entity.Retention
  503. for _, vv := range v {
  504. if vv.ID == 1 {
  505. retention.Day1 = entity.DayRetention{
  506. LoggedIn: vv.Ratio,
  507. LoginDate: key,
  508. }
  509. }
  510. if vv.ID == 3 {
  511. retention.Day3 = entity.DayRetention{
  512. LoggedIn: vv.Ratio,
  513. LoginDate: key,
  514. }
  515. }
  516. if vv.ID == 7 {
  517. retention.Day7 = entity.DayRetention{
  518. LoggedIn: vv.Ratio,
  519. LoginDate: key,
  520. }
  521. }
  522. if vv.ID == 14 {
  523. retention.Day14 = entity.DayRetention{
  524. LoggedIn: vv.Ratio,
  525. LoginDate: key,
  526. }
  527. }
  528. if vv.ID == 30 {
  529. retention.Day30 = entity.DayRetention{
  530. LoggedIn: vv.Ratio,
  531. LoginDate: key,
  532. }
  533. }
  534. }
  535. results = append(results, &entity.UserRetentionDetail{
  536. RegistrationDate: key,
  537. RetentionData: retention,
  538. })
  539. }
  540. return &entity.UserRetentionResp{
  541. Details: results,
  542. }, nil
  543. }
  544. // FindUserLevel 用户等级统计
  545. func (s *Synthesis) FindUserLevel() (*entity.UserLevelCountResp, *code.Result) {
  546. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  547. defer cancel()
  548. collection := mdb.MDB.Collection(constant.CNamePlayerLevelStat)
  549. // 查询所有文档
  550. cursor, err := collection.Find(ctx, bson.M{})
  551. if err != nil {
  552. mhayaLogger.Warnf("FindUserLevel Find error:%v", err)
  553. return nil, common.NewResult(code.InternalError)
  554. }
  555. defer cursor.Close(ctx)
  556. var results []*entity.UserLevelCountDetail
  557. var reData []*models.PlayerLevelStat
  558. for cursor.Next(ctx) {
  559. var result models.PlayerLevelStat
  560. if err := cursor.Decode(&result); err != nil {
  561. mhayaLogger.Warnf("FindUserLevel Decode error:%v", err)
  562. return nil, common.NewResult(code.InternalError)
  563. }
  564. reData = append(reData, &result)
  565. }
  566. if err := cursor.Err(); err != nil {
  567. mhayaLogger.Warnf("FindUserLevel cursor error:%v", err)
  568. return nil, common.NewResult(code.InternalError)
  569. }
  570. dataMap := make(map[string]int)
  571. for _, v := range reData {
  572. for key, v1 := range v.ServerLevel {
  573. dataMap[key] += v1
  574. }
  575. }
  576. for k, v := range dataMap {
  577. results = append(results, &entity.UserLevelCountDetail{
  578. Level: cast.ToInt(k),
  579. UserCount: v,
  580. })
  581. }
  582. return &entity.UserLevelCountResp{
  583. Details: results,
  584. }, nil
  585. }
  586. func (s *Synthesis) InsertRecord(param model.UserOperationLog) {
  587. mhayaLogger.Warnf("InsertRecord param:%#v", param)
  588. record := new(model.UserOperationLog)
  589. collection := mdb.MDB.Collection(record.TableName())
  590. insertData := bson.M{}
  591. insertData["user_name"] = param.Username
  592. insertData["role_id"] = param.RoleId
  593. insertData["url"] = param.Path
  594. insertData["method"] = param.Method
  595. insertData["status_code"] = param.StatusCode
  596. insertData["dur"] = param.Dur
  597. insertData["client_ip"] = param.ClientIP
  598. insertData["error_message"] = param.ErrorMessage
  599. insertData["created_at"] = mhayaTime.Now().Unix()
  600. _, err := collection.InsertOne(context.Background(), insertData)
  601. if err != nil {
  602. mhayaLogger.Warnf("InsertRecord InsertOne error:%v", err)
  603. return
  604. }
  605. }
  606. func (s *Synthesis) RecordList(req entity.RecordListReq) (*entity.RecordListResp, *code.Result) {
  607. page, pageSize := checkPageParam(req.Page, req.Size)
  608. var records []*eventmodels.BackendOperationEventContent
  609. where := &eventmodels.BackendOperationEventContent{
  610. UserBasic: eventmodels.UserBasic{
  611. UserName: req.UserName,
  612. },
  613. EventBasic: eventmodels.EventBasic{
  614. ServerId: s.nodeId,
  615. },
  616. RoleId: req.RoleId,
  617. Path: req.Path,
  618. StatusCode: req.StatusCode,
  619. ClientIP: req.ClientIP,
  620. }
  621. db := mdb.LogstashDB.Model(&eventmodels.BackendOperationEventContent{}).Where(where).Order("create_at")
  622. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  623. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  624. }
  625. if req.DurMin > 0 && req.DurMax > 0 && req.DurMin <= req.DurMax {
  626. db = db.Where("dur >= ? and dur <= ?", req.DurMin, req.DurMax)
  627. }
  628. pages := Paginate(db, page, pageSize)
  629. err := db.Scopes(pages.Limit).Find(&records).Error
  630. if err != nil && err != gorm.ErrRecordNotFound {
  631. mhayaLogger.Warnf("RecordList Find error:%v", err)
  632. return nil, common.NewResult(code.InternalError)
  633. }
  634. var results []*entity.RecordListDetail
  635. for _, v := range records {
  636. results = append(results, &entity.RecordListDetail{
  637. Username: v.UserName,
  638. RoleId: v.RoleId,
  639. Path: v.Path,
  640. Method: v.Method,
  641. StatusCode: v.StatusCode,
  642. Dur: v.Dur,
  643. ClientIP: v.ClientIP,
  644. ErrorMessage: v.ErrorMessage,
  645. CreatedAt: v.CreateAt,
  646. })
  647. }
  648. return &entity.RecordListResp{
  649. Details: results,
  650. Total: pages.Count,
  651. }, nil
  652. }
  653. // 转盘统计
  654. func (s *Synthesis) Turntable(req entity.TurntableReq) (*entity.TurntableResp, *code.Result) {
  655. page, pageSize := checkPageParam(req.Page, req.Size)
  656. var records []*eventmodels.TurntableEventContent
  657. where := &eventmodels.TurntableEventContent{
  658. UserBasic: eventmodels.UserBasic{
  659. UserId: req.UserName,
  660. TgId: req.OpenId,
  661. },
  662. EventBasic: eventmodels.EventBasic{
  663. ServerId: s.nodeId,
  664. },
  665. }
  666. db := mdb.LogstashDB.Model(&eventmodels.TurntableEventContent{}).Where(where).Order("create_at")
  667. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  668. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  669. }
  670. pages := Paginate(db, page, pageSize)
  671. err := db.Scopes(pages.Limit).Find(&records).Error
  672. if err != nil && err != gorm.ErrRecordNotFound {
  673. mhayaLogger.Warnf("Turntable Find error:%v", err)
  674. return nil, common.NewResult(code.InternalError)
  675. }
  676. var results []*entity.TurntableDetail
  677. for _, v := range records {
  678. results = append(results, &entity.TurntableDetail{
  679. UserName: v.UserName,
  680. OpenId: v.TgId,
  681. TurntableExecutionTime: v.CreateAt,
  682. PrizeName: v.PrizeName,
  683. PrizeNumber: v.PrizeNumber,
  684. })
  685. }
  686. return &entity.TurntableResp{
  687. Details: results,
  688. Total: pages.Count,
  689. }, nil
  690. }
  691. // 资产统计
  692. func (s *Synthesis) Assets(req entity.AssetsReq) (*entity.AssetsResp, *code.Result) {
  693. // TODO Assets
  694. // page, pageSize := checkPageParam(req.Page, req.Size)
  695. // var records []*eventmodels.TurntableEventContent
  696. // where := &eventmodels.TurntableEventContent{
  697. // UserBasic: eventmodels.UserBasic{
  698. // UserId: req.UserName,
  699. // TgId: req.OpenId,
  700. // },
  701. // EventBasic: eventmodels.EventBasic{
  702. // ServerId: s.nodeId,
  703. // },
  704. // }
  705. // db := mdb.LogstashDB.Model(&eventmodels.TurntableEventContent{}).Where(where).Order("create_at")
  706. // pages := Paginate(db, page, pageSize)
  707. // err := db.Scopes(pages.Limit).Find(&records).Error
  708. // if err != nil && err != gorm.ErrRecordNotFound {
  709. // mhayaLogger.Warnf("Turntable Find error:%v", err)
  710. // return nil, common.NewResult(code.InternalError)
  711. // }
  712. // var results []*entity.TurntableDetail
  713. // for _, v := range records {
  714. // results = append(results, &entity.TurntableDetail{
  715. // UserName: v.UserName,
  716. // OpenId: v.TgId,
  717. // TurntableExecutionTime: v.CreateAt,
  718. // PrizeName: v.PrizeName,
  719. // PrizeNumber: v.PrizeNumber,
  720. // })
  721. // }
  722. count, codeResult := NewPlayerManage().GetPlayerTotalCount()
  723. if codeResult != nil {
  724. mhayaLogger.Warnf("Assets GetPlayerTotalCount codeResult:%v", codeResult)
  725. return nil, codeResult
  726. }
  727. return &entity.AssetsResp{
  728. // Details: results,
  729. Total: int64(count),
  730. }, nil
  731. }
  732. // 资产变动记录
  733. func (s *Synthesis) AssetsRecord(req entity.AssetsRecordReq) (*entity.AssetsRecordResp, *code.Result) {
  734. page, pageSize := checkPageParam(req.Page, req.Size)
  735. if !constant.CurrencyValid(req.Currency) {
  736. mhayaLogger.Warnf("AssetsRecord unknow currency:%v", req.Currency)
  737. return nil, common.NewResult(code.ParamError)
  738. }
  739. if req.OperationType != string(constant.IncreaseOp) && req.OperationType != string(constant.DecreaseOp) {
  740. mhayaLogger.Warnf("AssetsRecord unknow OperationType:%v", req.OperationType)
  741. return nil, common.NewResult(code.ParamError)
  742. }
  743. var records []*eventmodels.AssetsEventContent
  744. where := &eventmodels.AssetsEventContent{
  745. UserBasic: eventmodels.UserBasic{
  746. UserId: req.UserName,
  747. TgId: req.OpenId,
  748. },
  749. EventBasic: eventmodels.EventBasic{
  750. ServerId: s.nodeId,
  751. },
  752. OperationType: constant.OperationTypeStr(req.OperationType),
  753. Currency: constant.CurrencyTypeStr(req.Currency),
  754. Amount: req.Amount,
  755. }
  756. db := mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Order("create_at")
  757. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  758. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  759. }
  760. pages := Paginate(db, page, pageSize)
  761. err := db.Scopes(pages.Limit).Find(&records).Error
  762. if err != nil && err != gorm.ErrRecordNotFound {
  763. mhayaLogger.Warnf("Invite Find error:%v", err)
  764. return nil, common.NewResult(code.InternalError)
  765. }
  766. var results []*entity.AssetsRecordDetail
  767. for _, v := range records {
  768. results = append(results, &entity.AssetsRecordDetail{
  769. UserName: v.UserName,
  770. OpenId: v.TgId,
  771. AssetType: string(v.Currency),
  772. AssetsGrowth: v.Amount,
  773. AcquisitionTime: v.CreateAt,
  774. SourceOfAssetAcquisition: v.Reason,
  775. })
  776. }
  777. return &entity.AssetsRecordResp{
  778. Details: results,
  779. Total: pages.Count,
  780. }, nil
  781. }
  782. // 邀请统计
  783. func (s *Synthesis) Invite(req entity.InviteReq) (*entity.InviteResp, *code.Result) {
  784. page, pageSize := checkPageParam(req.Page, req.Size)
  785. // 构建查询条件
  786. filter := bson.M{}
  787. if req.UserName != "" {
  788. filter["userName"] = req.UserName
  789. }
  790. if req.OpenId != "" {
  791. filter["openId"] = req.OpenId
  792. }
  793. // 设置分页选项
  794. findOptions := options.Find()
  795. findOptions.SetSkip(int64((page - 1) * pageSize))
  796. findOptions.SetLimit(int64(pageSize))
  797. findOptions.SetSort(bson.D{{"createTime", -1}})
  798. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  799. defer cancel()
  800. collection := mdb.MDB.Collection(constant.CNamePlayer)
  801. // 查询数据
  802. var results []*entity.InviteDetail
  803. cursor, err := collection.Find(ctx, filter, findOptions)
  804. if err != nil {
  805. mhayaLogger.Warnf("Invite Find error:%v", err)
  806. return nil, common.NewResult(code.InternalError)
  807. }
  808. defer cursor.Close(ctx)
  809. // 解析结果
  810. for cursor.Next(ctx) {
  811. var result entity.InviteDetail
  812. if err := cursor.Decode(&result); err != nil {
  813. mhayaLogger.Warnf("Invite Decode error:%v", err)
  814. return nil, common.NewResult(code.InternalError)
  815. }
  816. results = append(results, &result)
  817. }
  818. if err := cursor.Err(); err != nil {
  819. mhayaLogger.Warnf("Invite cursor error:%v", err)
  820. return nil, common.NewResult(code.InternalError)
  821. }
  822. // TODO 邀请统计相关信息
  823. // TODO 获取总数total
  824. var count int64
  825. return &entity.InviteResp{
  826. Details: results,
  827. Total: count,
  828. }, nil
  829. }
  830. // 活跃统计
  831. func (s *Synthesis) Active(req entity.ActiveReq) (*entity.ActiveResp, *code.Result) {
  832. page, pageSize := checkPageParam(req.Page, req.Size)
  833. // 构建查询条件
  834. filter := bson.M{}
  835. if req.UserName != "" {
  836. filter["userName"] = req.UserName
  837. }
  838. if req.OpenId != "" {
  839. filter["openId"] = req.OpenId
  840. }
  841. // 设置分页选项
  842. findOptions := options.Find()
  843. findOptions.SetSkip(int64((page - 1) * pageSize))
  844. findOptions.SetLimit(int64(pageSize))
  845. findOptions.SetSort(bson.D{{"createTime", -1}})
  846. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  847. defer cancel()
  848. collection := mdb.MDB.Collection(constant.CNamePlayer)
  849. // 查询数据
  850. var results []*entity.ActiveDetail
  851. cursor, err := collection.Find(ctx, filter, findOptions)
  852. if err != nil {
  853. mhayaLogger.Warnf("Active Find error:%v", err)
  854. return nil, common.NewResult(code.InternalError)
  855. }
  856. defer cursor.Close(ctx)
  857. // 解析结果
  858. for cursor.Next(ctx) {
  859. var result entity.ActiveDetail
  860. if err := cursor.Decode(&result); err != nil {
  861. mhayaLogger.Warnf("Active Decode error:%v", err)
  862. return nil, common.NewResult(code.InternalError)
  863. }
  864. results = append(results, &result)
  865. }
  866. if err := cursor.Err(); err != nil {
  867. mhayaLogger.Warnf("Active cursor error:%v", err)
  868. return nil, common.NewResult(code.InternalError)
  869. }
  870. // TODO Active 活跃统计相关信息
  871. // TODO 获取总数total
  872. var count int64
  873. return &entity.ActiveResp{
  874. Details: results,
  875. Total: count,
  876. }, nil
  877. }
  878. // 任务完成度统计
  879. func (s *Synthesis) TaskCompletion(req entity.TaskCompletionReq) (*entity.TaskCompletionResp, *code.Result) {
  880. page, pageSize := checkPageParam(req.Page, req.Size)
  881. // 构建查询条件
  882. filter := bson.M{}
  883. if req.UserName != "" {
  884. filter["userName"] = req.UserName
  885. }
  886. if req.OpenId != "" {
  887. filter["openId"] = req.OpenId
  888. }
  889. // 设置分页选项
  890. findOptions := options.Find()
  891. findOptions.SetSkip(int64((page - 1) * pageSize))
  892. findOptions.SetLimit(int64(pageSize))
  893. findOptions.SetSort(bson.D{{"createTime", -1}})
  894. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  895. defer cancel()
  896. collection := mdb.MDB.Collection(constant.CNamePlayer)
  897. // 查询数据
  898. var results []*entity.TaskCompletionDetail
  899. cursor, err := collection.Find(ctx, filter, findOptions)
  900. if err != nil {
  901. mhayaLogger.Warnf("TaskCompletion Find error:%v", err)
  902. return nil, common.NewResult(code.InternalError)
  903. }
  904. defer cursor.Close(ctx)
  905. // 解析结果
  906. for cursor.Next(ctx) {
  907. var result entity.TaskCompletionDetail
  908. if err := cursor.Decode(&result); err != nil {
  909. mhayaLogger.Warnf("TaskCompletion Decode error:%v", err)
  910. return nil, common.NewResult(code.InternalError)
  911. }
  912. results = append(results, &result)
  913. }
  914. if err := cursor.Err(); err != nil {
  915. mhayaLogger.Warnf("TaskCompletion cursor error:%v", err)
  916. return nil, common.NewResult(code.InternalError)
  917. }
  918. // TODO 任务完成度统计相关信息
  919. // TODO 获取总数total
  920. var count int64
  921. return &entity.TaskCompletionResp{
  922. Details: results,
  923. Total: count,
  924. }, nil
  925. }
  926. // 用户行为检测
  927. func (s *Synthesis) BehaviorMonitoring(req entity.BehaviorMonitoringReq) (*entity.BehaviorMonitoringResp, *code.Result) {
  928. page, pageSize := checkPageParam(req.Page, req.Size)
  929. // 构建查询条件
  930. filter := bson.M{}
  931. if req.UserName != "" {
  932. filter["userName"] = req.UserName
  933. }
  934. if req.OpenId != "" {
  935. filter["openId"] = req.OpenId
  936. }
  937. // 设置分页选项
  938. findOptions := options.Find()
  939. findOptions.SetSkip(int64((page - 1) * pageSize))
  940. findOptions.SetLimit(int64(pageSize))
  941. findOptions.SetSort(bson.D{{"createTime", -1}})
  942. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  943. defer cancel()
  944. collection := mdb.MDB.Collection(constant.CNamePlayer)
  945. // 查询数据
  946. var results []*entity.BehaviorMonitoringDetail
  947. cursor, err := collection.Find(ctx, filter, findOptions)
  948. if err != nil {
  949. mhayaLogger.Warnf("BehaviorMonitoring Find error:%v", err)
  950. return nil, common.NewResult(code.InternalError)
  951. }
  952. defer cursor.Close(ctx)
  953. // 解析结果
  954. for cursor.Next(ctx) {
  955. var result entity.BehaviorMonitoringDetail
  956. if err := cursor.Decode(&result); err != nil {
  957. mhayaLogger.Warnf("BehaviorMonitoring Decode error:%v", err)
  958. return nil, common.NewResult(code.InternalError)
  959. }
  960. results = append(results, &result)
  961. }
  962. if err := cursor.Err(); err != nil {
  963. mhayaLogger.Warnf("BehaviorMonitoring cursor error:%v", err)
  964. return nil, common.NewResult(code.InternalError)
  965. }
  966. // TODO 用户行为检测相关信息
  967. // TODO 获取总数total
  968. var count int64
  969. return &entity.BehaviorMonitoringResp{
  970. Details: results,
  971. Total: count,
  972. }, nil
  973. }