synthesis.go 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197
  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: 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. page, pageSize := checkPageParam(req.Page, req.Size)
  694. playerMgr := NewPlayerManage()
  695. listResp, codeResult := playerMgr.List(context.Background(), entity.PlayerListReq{
  696. Page: page,
  697. Size: pageSize,
  698. })
  699. if codeResult != nil {
  700. mhayaLogger.Warnf("Assets List error:%v", codeResult)
  701. return nil, codeResult
  702. }
  703. if req.UserName != "" || req.OpenId != "" {
  704. var results []*entity.AssetsDetail
  705. var totalUsdtAmount int64
  706. var totalTonAmount int64
  707. where := &eventmodels.AssetsEventContent{
  708. UserBasic: eventmodels.UserBasic{
  709. UserId: req.UserName,
  710. TgId: req.OpenId,
  711. },
  712. EventBasic: eventmodels.EventBasic{
  713. ServerId: s.nodeId,
  714. },
  715. }
  716. where.Currency = string(constant.UsdtCurrency)
  717. err := mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &totalUsdtAmount).Error
  718. if err != nil {
  719. mhayaLogger.Warnf("Assets Pluck UsdtCurrency error:%v", err)
  720. return nil, common.NewResult(code.InternalError)
  721. }
  722. where.Currency = string(constant.TonCurrency)
  723. err = mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &totalTonAmount).Error
  724. if err != nil {
  725. mhayaLogger.Warnf("Assets Pluck TonCurrency error:%v", err)
  726. return nil, common.NewResult(code.InternalError)
  727. }
  728. results = append(results, &entity.AssetsDetail{
  729. UserName: req.UserName,
  730. OpenId: req.OpenId,
  731. TonValue: totalTonAmount,
  732. UsdtValue: totalUsdtAmount,
  733. StatisticalDate: mhayaTime.Now().Unix(),
  734. })
  735. return &entity.AssetsResp{
  736. Details: results,
  737. Total: 1,
  738. }, nil
  739. }
  740. var results []*entity.AssetsDetail
  741. for _, detail := range listResp.Details {
  742. var totalUsdtAmount int64
  743. var totalTonAmount int64
  744. where := &eventmodels.AssetsEventContent{
  745. UserBasic: eventmodels.UserBasic{
  746. UserId: detail.UserName,
  747. TgId: detail.OpenId,
  748. },
  749. EventBasic: eventmodels.EventBasic{
  750. ServerId: s.nodeId,
  751. },
  752. }
  753. where.Currency = string(constant.UsdtCurrency)
  754. err := mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &totalUsdtAmount).Error
  755. if err != nil {
  756. mhayaLogger.Warnf("Assets Pluck UsdtCurrency error:%v", err)
  757. return nil, common.NewResult(code.InternalError)
  758. }
  759. where.Currency = string(constant.TonCurrency)
  760. err = mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &totalTonAmount).Error
  761. if err != nil {
  762. mhayaLogger.Warnf("Assets Pluck TonCurrency error:%v", err)
  763. return nil, common.NewResult(code.InternalError)
  764. }
  765. results = append(results, &entity.AssetsDetail{
  766. UserName: detail.UserName,
  767. OpenId: detail.OpenId,
  768. TonValue: totalTonAmount,
  769. UsdtValue: totalUsdtAmount,
  770. StatisticalDate: mhayaTime.Now().Unix(),
  771. })
  772. }
  773. count, codeResult := playerMgr.GetPlayerTotalCount()
  774. if codeResult != nil {
  775. mhayaLogger.Warnf("Assets GetPlayerTotalCount codeResult:%v", codeResult)
  776. return nil, codeResult
  777. }
  778. return &entity.AssetsResp{
  779. Details: results,
  780. Total: int64(count),
  781. }, nil
  782. }
  783. // 资产变动记录
  784. func (s *Synthesis) AssetsRecord(req entity.AssetsRecordReq) (*entity.AssetsRecordResp, *code.Result) {
  785. page, pageSize := checkPageParam(req.Page, req.Size)
  786. if !constant.CurrencyValid(req.Currency) {
  787. mhayaLogger.Warnf("AssetsRecord unknow currency:%v", req.Currency)
  788. return nil, common.NewResult(code.ParamError)
  789. }
  790. if req.OperationType != string(constant.IncreaseOp) && req.OperationType != string(constant.DecreaseOp) {
  791. mhayaLogger.Warnf("AssetsRecord unknow OperationType:%v", req.OperationType)
  792. return nil, common.NewResult(code.ParamError)
  793. }
  794. var records []*eventmodels.AssetsEventContent
  795. where := &eventmodels.AssetsEventContent{
  796. UserBasic: eventmodels.UserBasic{
  797. UserId: req.UserName,
  798. TgId: req.OpenId,
  799. },
  800. EventBasic: eventmodels.EventBasic{
  801. ServerId: s.nodeId,
  802. },
  803. OperationType: req.OperationType,
  804. Currency: req.Currency,
  805. Amount: req.Amount,
  806. }
  807. db := mdb.LogstashDB.Model(&eventmodels.AssetsEventContent{}).Where(where).Order("create_at")
  808. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  809. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  810. }
  811. pages := Paginate(db, page, pageSize)
  812. err := db.Scopes(pages.Limit).Find(&records).Error
  813. if err != nil && err != gorm.ErrRecordNotFound {
  814. mhayaLogger.Warnf("Invite Find error:%v", err)
  815. return nil, common.NewResult(code.InternalError)
  816. }
  817. var results []*entity.AssetsRecordDetail
  818. for _, v := range records {
  819. results = append(results, &entity.AssetsRecordDetail{
  820. UserName: v.UserName,
  821. OpenId: v.TgId,
  822. AssetType: string(v.Currency),
  823. AssetsGrowth: v.Amount,
  824. AcquisitionTime: v.CreateAt,
  825. SourceOfAssetAcquisition: v.Reason,
  826. })
  827. }
  828. return &entity.AssetsRecordResp{
  829. Details: results,
  830. Total: pages.Count,
  831. }, nil
  832. }
  833. // 邀请统计
  834. func (s *Synthesis) Invite(req entity.InviteReq) (*entity.InviteResp, *code.Result) {
  835. page, pageSize := checkPageParam(req.Page, req.Size)
  836. // 构建查询条件
  837. filter := bson.M{}
  838. if req.UserName != "" {
  839. filter["userName"] = req.UserName
  840. }
  841. if req.OpenId != "" {
  842. filter["openId"] = req.OpenId
  843. }
  844. // 设置分页选项
  845. findOptions := options.Find()
  846. findOptions.SetSkip(int64((page - 1) * pageSize))
  847. findOptions.SetLimit(int64(pageSize))
  848. findOptions.SetSort(bson.D{{"createTime", -1}})
  849. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  850. defer cancel()
  851. collection := mdb.MDB.Collection(constant.CNamePlayer)
  852. // 查询数据
  853. var results []*entity.InviteDetail
  854. cursor, err := collection.Find(ctx, filter, findOptions)
  855. if err != nil {
  856. mhayaLogger.Warnf("Invite Find error:%v", err)
  857. return nil, common.NewResult(code.InternalError)
  858. }
  859. defer cursor.Close(ctx)
  860. // 解析结果
  861. for cursor.Next(ctx) {
  862. var result entity.InviteDetail
  863. if err := cursor.Decode(&result); err != nil {
  864. mhayaLogger.Warnf("Invite Decode error:%v", err)
  865. return nil, common.NewResult(code.InternalError)
  866. }
  867. results = append(results, &result)
  868. }
  869. if err := cursor.Err(); err != nil {
  870. mhayaLogger.Warnf("Invite cursor error:%v", err)
  871. return nil, common.NewResult(code.InternalError)
  872. }
  873. // TODO 邀请统计相关信息
  874. // TODO 获取总数total
  875. var count int64
  876. return &entity.InviteResp{
  877. Details: results,
  878. Total: count,
  879. }, nil
  880. }
  881. // 活跃统计
  882. func (s *Synthesis) Active(req entity.ActiveReq) (*entity.ActiveResp, *code.Result) {
  883. page, pageSize := checkPageParam(req.Page, req.Size)
  884. // 构建查询条件
  885. filter := bson.M{}
  886. if req.UserName != "" {
  887. filter["userName"] = req.UserName
  888. }
  889. if req.OpenId != "" {
  890. filter["openId"] = req.OpenId
  891. }
  892. // 设置分页选项
  893. findOptions := options.Find()
  894. findOptions.SetSkip(int64((page - 1) * pageSize))
  895. findOptions.SetLimit(int64(pageSize))
  896. findOptions.SetSort(bson.D{{"createTime", -1}})
  897. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  898. defer cancel()
  899. collection := mdb.MDB.Collection(constant.CNamePlayer)
  900. // 查询数据
  901. var results []*entity.ActiveDetail
  902. cursor, err := collection.Find(ctx, filter, findOptions)
  903. if err != nil {
  904. mhayaLogger.Warnf("Active Find error:%v", err)
  905. return nil, common.NewResult(code.InternalError)
  906. }
  907. defer cursor.Close(ctx)
  908. // 解析结果
  909. for cursor.Next(ctx) {
  910. var result entity.ActiveDetail
  911. if err := cursor.Decode(&result); err != nil {
  912. mhayaLogger.Warnf("Active Decode error:%v", err)
  913. return nil, common.NewResult(code.InternalError)
  914. }
  915. results = append(results, &result)
  916. }
  917. if err := cursor.Err(); err != nil {
  918. mhayaLogger.Warnf("Active cursor error:%v", err)
  919. return nil, common.NewResult(code.InternalError)
  920. }
  921. // TODO Active 活跃统计相关信息
  922. // TODO 获取总数total
  923. var count int64
  924. return &entity.ActiveResp{
  925. Details: results,
  926. Total: count,
  927. }, nil
  928. }
  929. // 任务完成度统计
  930. func (s *Synthesis) TaskCompletion(req entity.TaskCompletionReq) (*entity.TaskCompletionResp, *code.Result) {
  931. page, pageSize := checkPageParam(req.Page, req.Size)
  932. // 构建查询条件
  933. filter := bson.M{}
  934. if req.UserName != "" {
  935. filter["userName"] = req.UserName
  936. }
  937. if req.OpenId != "" {
  938. filter["openId"] = req.OpenId
  939. }
  940. // 设置分页选项
  941. findOptions := options.Find()
  942. findOptions.SetSkip(int64((page - 1) * pageSize))
  943. findOptions.SetLimit(int64(pageSize))
  944. findOptions.SetSort(bson.D{{"createTime", -1}})
  945. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  946. defer cancel()
  947. collection := mdb.MDB.Collection(constant.CNamePlayer)
  948. // 查询数据
  949. var results []*entity.TaskCompletionDetail
  950. cursor, err := collection.Find(ctx, filter, findOptions)
  951. if err != nil {
  952. mhayaLogger.Warnf("TaskCompletion Find error:%v", err)
  953. return nil, common.NewResult(code.InternalError)
  954. }
  955. defer cursor.Close(ctx)
  956. // 解析结果
  957. for cursor.Next(ctx) {
  958. var result entity.TaskCompletionDetail
  959. if err := cursor.Decode(&result); err != nil {
  960. mhayaLogger.Warnf("TaskCompletion Decode error:%v", err)
  961. return nil, common.NewResult(code.InternalError)
  962. }
  963. results = append(results, &result)
  964. }
  965. if err := cursor.Err(); err != nil {
  966. mhayaLogger.Warnf("TaskCompletion cursor error:%v", err)
  967. return nil, common.NewResult(code.InternalError)
  968. }
  969. // TODO 任务完成度统计相关信息
  970. // TODO 获取总数total
  971. var count int64
  972. return &entity.TaskCompletionResp{
  973. Details: results,
  974. Total: count,
  975. }, nil
  976. }
  977. // 用户行为检测
  978. func (s *Synthesis) BehaviorMonitoring(req entity.BehaviorMonitoringReq) (*entity.BehaviorMonitoringResp, *code.Result) {
  979. page, pageSize := checkPageParam(req.Page, req.Size)
  980. // 构建查询条件
  981. filter := bson.M{}
  982. if req.UserName != "" {
  983. filter["userName"] = req.UserName
  984. }
  985. if req.OpenId != "" {
  986. filter["openId"] = req.OpenId
  987. }
  988. // 设置分页选项
  989. findOptions := options.Find()
  990. findOptions.SetSkip(int64((page - 1) * pageSize))
  991. findOptions.SetLimit(int64(pageSize))
  992. findOptions.SetSort(bson.D{{"createTime", -1}})
  993. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  994. defer cancel()
  995. collection := mdb.MDB.Collection(constant.CNamePlayer)
  996. // 查询数据
  997. var results []*entity.BehaviorMonitoringDetail
  998. cursor, err := collection.Find(ctx, filter, findOptions)
  999. if err != nil {
  1000. mhayaLogger.Warnf("BehaviorMonitoring Find error:%v", err)
  1001. return nil, common.NewResult(code.InternalError)
  1002. }
  1003. defer cursor.Close(ctx)
  1004. // 解析结果
  1005. for cursor.Next(ctx) {
  1006. var result entity.BehaviorMonitoringDetail
  1007. if err := cursor.Decode(&result); err != nil {
  1008. mhayaLogger.Warnf("BehaviorMonitoring Decode error:%v", err)
  1009. return nil, common.NewResult(code.InternalError)
  1010. }
  1011. results = append(results, &result)
  1012. }
  1013. if err := cursor.Err(); err != nil {
  1014. mhayaLogger.Warnf("BehaviorMonitoring cursor error:%v", err)
  1015. return nil, common.NewResult(code.InternalError)
  1016. }
  1017. // TODO 用户行为检测相关信息
  1018. // TODO 获取总数total
  1019. var count int64
  1020. return &entity.BehaviorMonitoringResp{
  1021. Details: results,
  1022. Total: count,
  1023. }, nil
  1024. }