synthesis.go 32 KB

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