synthesis.go 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
  1. package service
  2. import (
  3. "context"
  4. "time"
  5. "github.com/mhaya/game/game_cluster/internal/mdb"
  6. "github.com/mhaya/game/game_cluster/nodes/webadmin/entity"
  7. mdb2 "github.com/mhaya/game/game_cluster/nodes/webadmin/mdb"
  8. "go.mongodb.org/mongo-driver/bson"
  9. "go.mongodb.org/mongo-driver/mongo"
  10. "go.mongodb.org/mongo-driver/mongo/options"
  11. )
  12. type Synthesis struct {
  13. db *mongo.Database
  14. }
  15. func NewSynthesis() *Synthesis {
  16. return &Synthesis{
  17. db: mdb2.MdbAdmin,
  18. }
  19. }
  20. func (s *Synthesis) FindMDBUserLogDaily(req *entity.UserLogDailyReq) ([]*entity.UserLogDailyResp, error) {
  21. // 定义上下文
  22. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  23. defer cancel()
  24. // 指定集合
  25. collection := s.db.Collection("user_logs_daily")
  26. // 构建查询条件 - 如果查询值为空那就不添加查询条件
  27. filter := bson.M{}
  28. if req.StartTime != 0 {
  29. filter["timestamp"] = bson.M{
  30. "$gte": req.StartTime,
  31. "$lte": req.EndTime,
  32. }
  33. }
  34. if req.Platform != "" {
  35. filter["platform"] = req.Platform
  36. }
  37. if req.Channel != "" {
  38. filter["channel"] = req.Channel
  39. }
  40. // 分页参数
  41. skip := (req.Page - 1) * req.Size
  42. // 执行查询
  43. opts := options.Find()
  44. opts.SetSkip(int64(skip))
  45. opts.SetLimit(int64(req.Size))
  46. cursor, err := collection.Find(ctx, filter, opts)
  47. if err != nil {
  48. return nil, err
  49. }
  50. defer cursor.Close(ctx)
  51. // 解析查询结果
  52. var results []*entity.UserLogDailyResp
  53. for cursor.Next(ctx) {
  54. var result entity.UserLogDailyResp
  55. err := cursor.Decode(&result)
  56. if err != nil {
  57. return nil, err
  58. }
  59. results = append(results, &result)
  60. }
  61. if err := cursor.Err(); err != nil {
  62. return nil, err
  63. }
  64. return results, nil
  65. }
  66. // FindWithdrawal 根据请求查询提现记录
  67. func (s *Synthesis) FindWithdrawal(req *entity.UserWithdrawalReq) ([]*entity.UserWithdrawalResp, error) {
  68. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  69. defer cancel()
  70. collection := mdb.MDB.Collection("CashOutRecord")
  71. // 构建过滤器
  72. filter := bson.M{}
  73. if req.UserName != "" {
  74. filter["userName"] = req.UserName
  75. }
  76. if req.NickName != "" {
  77. filter["nickName"] = req.NickName
  78. }
  79. if req.StartTime > 0 && req.EndTime > 0 {
  80. filter["createdAt"] = bson.M{"$gte": time.Unix(req.StartTime, 0), "$lte": time.Unix(req.EndTime, 0)}
  81. } else if req.StartTime > 0 {
  82. filter["createdAt"] = bson.M{"$gte": time.Unix(req.StartTime, 0)}
  83. } else if req.EndTime > 0 {
  84. filter["createdAt"] = bson.M{"$lte": time.Unix(req.EndTime, 0)}
  85. }
  86. // 设置分页选项
  87. findOptions := options.Find()
  88. findOptions.SetSkip(int64((req.Page - 1) * req.Size))
  89. findOptions.SetLimit(int64(req.Size))
  90. // 查询数据
  91. var results []*entity.UserWithdrawalResp
  92. cursor, err := collection.Find(ctx, filter, findOptions)
  93. if err != nil {
  94. return nil, err
  95. }
  96. defer cursor.Close(ctx)
  97. // 解析结果
  98. for cursor.Next(ctx) {
  99. var result entity.UserWithdrawalResp
  100. if err := cursor.Decode(&result); err != nil {
  101. return nil, err
  102. }
  103. results = append(results, &result)
  104. }
  105. if err := cursor.Err(); err != nil {
  106. return nil, err
  107. }
  108. return results, nil
  109. }