synthesis.go 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165
  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. var records []*eventmodels.UserWithdrawEventContent
  214. where := &eventmodels.UserWithdrawEventContent{
  215. UserBasic: eventmodels.UserBasic{
  216. UserId: req.UserName,
  217. UserName: req.NickName,
  218. },
  219. EventBasic: eventmodels.EventBasic{
  220. ServerId: s.nodeId,
  221. },
  222. WithdrawId: req.ID,
  223. Currency: constant.UsdtCurrency,
  224. Address: req.Address,
  225. State: req.State,
  226. WithdrawalState: req.Withdrawal,
  227. }
  228. db := mdb.LogstashDB.Model(&eventmodels.UserWithdrawEventContent{}).Where(where).Order("create_at")
  229. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  230. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  231. }
  232. if req.AmountMin > 0 && req.AmountMax > 0 && req.AmountMin <= req.AmountMax {
  233. db = db.Where("amount >= ? and amount <= ?", req.AmountMin, req.AmountMax)
  234. }
  235. if req.AfterAmountMin > 0 && req.AfterAmountMax > 0 && req.AfterAmountMin <= req.AfterAmountMax {
  236. db = db.Where("after_balance >= ? and after_balance <= ?", req.AfterAmountMin, req.AfterAmountMax)
  237. }
  238. pages := Paginate(db, page, pageSize)
  239. err := db.Scopes(pages.Limit).Find(&records).Error
  240. if err != nil && err != gorm.ErrRecordNotFound {
  241. mhayaLogger.Warnf("FindWithdrawal Find error:%v", err)
  242. return nil, common.NewResult(code.InternalError)
  243. }
  244. var results []*entity.UserWithdrawalDetail
  245. for _, v := range records {
  246. results = append(results, &entity.UserWithdrawalDetail{
  247. // TODO
  248. // Id: v.WithdrawId,
  249. // UserName: v.UserId,
  250. // NickName: v.UserName,
  251. // OpenId: v.TgId,
  252. // Status: v.State,
  253. // Reason: "",
  254. // Withdrawal: 0,
  255. // Amount: 0,
  256. // AfterAmount: 0,
  257. // Type: nil,
  258. // Address: "",
  259. // CreateAt: nil,
  260. // UpdateAt: nil,
  261. })
  262. }
  263. return &entity.UserWithdrawalResp{
  264. Details: results,
  265. Total: pages.Count,
  266. }, nil
  267. }
  268. // 导出提现记录
  269. func (s *Synthesis) WithdrawalExport(req entity.UserWithdrawalExportReq) (*entity.UserWithdrawalResp, *code.Result) {
  270. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  271. defer cancel()
  272. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  273. // 构建过滤器
  274. filter := bson.M{}
  275. if req.UserName != "" {
  276. filter["userName"] = req.UserName
  277. }
  278. if req.NickName != "" {
  279. filter["nickName"] = req.NickName
  280. }
  281. if req.ID != "" {
  282. filter["_id"], _ = primitive.ObjectIDFromHex(req.ID)
  283. }
  284. if req.Address != "" {
  285. filter["address"] = req.Address
  286. }
  287. if req.State > 0 {
  288. filter["state"] = req.State
  289. }
  290. if req.Withdrawal > 0 {
  291. filter["withdrawal"] = req.Withdrawal
  292. }
  293. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  294. filter["createAt"] = bson.M{
  295. "$gte": req.StartTime,
  296. "$lte": req.EndTime,
  297. }
  298. }
  299. if req.AmountMin > 0 && req.AmountMax > 0 && req.AmountMin <= req.AmountMax {
  300. filter["amount"] = bson.M{
  301. "$gte": req.AmountMin,
  302. "$lte": req.AmountMax,
  303. }
  304. }
  305. if req.AfterAmountMin > 0 && req.AfterAmountMax > 0 && req.AfterAmountMin <= req.AfterAmountMax {
  306. filter["after_amount"] = bson.M{
  307. "$gte": req.AfterAmountMin,
  308. "$lte": req.AfterAmountMax,
  309. }
  310. }
  311. findOptions := options.Find()
  312. findOptions.SetSort(bson.D{{"createAt", -1}})
  313. // 获取总数total
  314. count, err := collection.CountDocuments(ctx, filter)
  315. if err != nil {
  316. mhayaLogger.Warnf("WithdrawalExportData CountDocuments error:%v", err)
  317. return nil, common.NewResult(code.InternalError)
  318. }
  319. // 查询数据
  320. var results []*entity.UserWithdrawalDetail
  321. cursor, err := collection.Find(ctx, filter, findOptions)
  322. if err != nil {
  323. mhayaLogger.Warnf("WithdrawalExportData Find error:%v", err)
  324. return nil, common.NewResult(code.InternalError)
  325. }
  326. defer cursor.Close(ctx)
  327. // 解析结果
  328. for cursor.Next(ctx) {
  329. var result entity.UserWithdrawalDetail
  330. if err := cursor.Decode(&result); err != nil {
  331. mhayaLogger.Warnf("WithdrawalExportData Decode error:%v", err)
  332. return nil, common.NewResult(code.InternalError)
  333. }
  334. results = append(results, &result)
  335. }
  336. if err := cursor.Err(); err != nil {
  337. mhayaLogger.Warnf("WithdrawalExportData cursor error:%v", err)
  338. return nil, common.NewResult(code.InternalError)
  339. }
  340. return &entity.UserWithdrawalResp{
  341. Details: results,
  342. Total: count,
  343. }, nil
  344. }
  345. // WithdrawalStatus 更新提现状态
  346. func (s *Synthesis) WithdrawalStatus(req entity.UserWithdrawalStatus) *code.Result {
  347. // ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  348. // defer cancel()
  349. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  350. // 更新条件
  351. updateCondition := bson.M{"userName": req.UserName}
  352. // 更新内容
  353. updateContent := bson.M{"$set": bson.M{"status": req.Status}}
  354. // 设置更新选项
  355. updateOptions := options.Update() // 设置 upsert 选项
  356. // 执行更新操作
  357. _, err := collection.UpdateOne(context.TODO(), updateCondition, updateContent, updateOptions)
  358. if err != nil {
  359. mhayaLogger.Warnf("WithdrawalStatus UpdateOne error:%v", err)
  360. return common.NewResult(code.InternalError)
  361. }
  362. return nil
  363. }
  364. // WithdrawalStatusBatch 更新提现状态
  365. func (s *Synthesis) WithdrawalStatusBatch(req entity.UserWithdrawalStatusBatch) *code.Result {
  366. collection := mdb.MDB.Collection(constant.CNameCashOutRecord)
  367. if len(req.ID) == 0 {
  368. return common.NewResult(code.ParamError)
  369. }
  370. for _, id := range req.ID {
  371. objID := primitive.ObjectID{}
  372. objID, _ = primitive.ObjectIDFromHex(id)
  373. updateCondition := bson.M{"_id": objID}
  374. updateContent := bson.M{}
  375. withdrawal := models.CashOutRecord{}
  376. err := collection.FindOne(context.TODO(), updateCondition).Decode(&withdrawal)
  377. if err != nil {
  378. mhayaLogger.Warnf("WithdrawalStatusBatch FindOne error:%v", err)
  379. continue
  380. }
  381. if req.Withdrawal != 0 {
  382. if withdrawal.Status == 1 {
  383. updateContent = bson.M{"$set": bson.M{"withdrawal": req.Withdrawal}}
  384. } else {
  385. continue
  386. }
  387. }
  388. if req.Status > 0 {
  389. if withdrawal.Status != 0 {
  390. continue
  391. }
  392. updateContent = bson.M{"$set": bson.M{"status": req.Status}}
  393. }
  394. updateOptions := options.Update().SetUpsert(true)
  395. _, err = collection.UpdateOne(context.TODO(), updateCondition, updateContent, updateOptions)
  396. if err != nil {
  397. mhayaLogger.Warnf("WithdrawalStatusBatch UpdateOne error:%v", err)
  398. continue
  399. }
  400. }
  401. return nil
  402. }
  403. // FindUserCountryCount 查询用户国家分布
  404. //
  405. // 返回值为 UserCountryResp 的切片和错误。
  406. func (s *Synthesis) FindUserCountryCount() (*entity.UserCountryResp, *code.Result) {
  407. // 选择数据库和集合
  408. collection := mdb.MDB.Collection(constant.CNamePlayerCountryByIPStat)
  409. // 定义聚合管道
  410. // 定义聚合管道
  411. pipeline := []bson.D{
  412. {
  413. {Key: "$project", Value: bson.D{
  414. {Key: "playerRegisterCountry", Value: bson.D{{Key: "$objectToArray", Value: "$playerRegisterCountry"}}},
  415. }},
  416. },
  417. {
  418. {Key: "$unwind", Value: "$playerRegisterCountry"},
  419. },
  420. {
  421. {Key: "$group", Value: bson.D{
  422. {Key: "_id", Value: "$playerRegisterCountry.k"},
  423. {Key: "totalValue", Value: bson.D{{Key: "$sum", Value: "$playerRegisterCountry.v"}}},
  424. }},
  425. },
  426. {
  427. {Key: "$project", Value: bson.D{
  428. {Key: "_id", Value: 0},
  429. {Key: "countryKey", Value: "$_id"},
  430. {Key: "totalValue", Value: 1},
  431. }},
  432. },
  433. }
  434. // 执行聚合查询
  435. cursor, err := collection.Aggregate(context.TODO(), pipeline)
  436. if err != nil {
  437. mhayaLogger.Warnf("FindUserCountryCount Aggregate error:%v", err)
  438. return nil, common.NewResult(code.InternalError)
  439. }
  440. defer cursor.Close(context.TODO())
  441. // 遍历查询结果
  442. var results []bson.M
  443. if err := cursor.All(context.TODO(), &results); err != nil {
  444. mhayaLogger.Warnf("FindUserCountryCount All error:%v", err)
  445. return nil, common.NewResult(code.InternalError)
  446. }
  447. var totalIPCount int64
  448. var data []*entity.UserCountryDetail
  449. // 将结果转换为 UserCountryDetail
  450. for _, r := range results {
  451. var resp entity.UserCountryDetail
  452. resp.Country = r["countryKey"].(string)
  453. resp.IPCount = int(r["totalValue"].(int32))
  454. totalIPCount += int64(resp.IPCount)
  455. data = append(data, &resp)
  456. }
  457. for _, v := range data {
  458. // 保留小数点后两位
  459. v.Percentage = math.Round(float64(v.IPCount)/float64(totalIPCount)*10000) / 100
  460. }
  461. // 根据阈值过滤结果
  462. otherCount := 0
  463. otherPercentage := 0.00
  464. filteredResults := make([]*entity.UserCountryDetail, 0)
  465. threshold := 1.00
  466. for _, r := range data {
  467. if r.Percentage >= threshold {
  468. filteredResults = append(filteredResults, r)
  469. // 保留小数点后两位
  470. r.Percentage = math.Round(r.Percentage*100) / 100
  471. otherPercentage += r.Percentage
  472. } else {
  473. otherCount += r.IPCount
  474. }
  475. }
  476. // 将其他国家添加到过滤后的结果中
  477. if otherCount > 0 {
  478. p := 100.00 - math.Round(otherPercentage*100)/100
  479. filteredResults = append(filteredResults, &entity.UserCountryDetail{
  480. Country: "other",
  481. IPCount: otherCount,
  482. Percentage: math.Round(p*100) / 100,
  483. })
  484. }
  485. return &entity.UserCountryResp{
  486. Details: filteredResults,
  487. }, nil
  488. }
  489. // FindUserRetention UserRetentionResp 用户留存率
  490. // 1. 获取指定日期范围内的注册用户数量
  491. // 2. 获取指定日期范围内的活跃用户数量
  492. // 3. 计算留存率
  493. // 4. 返回结果
  494. func (s *Synthesis) FindUserRetention(req entity.UserRetentionReq) (*entity.UserRetentionResp, *code.Result) {
  495. playerPreserve := models.GetPlayerPreserve(req.StartTime, req.EndTime)
  496. // 查询数据
  497. var results []*entity.UserRetentionDetail
  498. for key, v := range playerPreserve {
  499. var retention entity.Retention
  500. for _, vv := range v {
  501. if vv.ID == 1 {
  502. retention.Day1 = entity.DayRetention{
  503. LoggedIn: vv.Ratio,
  504. LoginDate: key,
  505. }
  506. }
  507. if vv.ID == 3 {
  508. retention.Day3 = entity.DayRetention{
  509. LoggedIn: vv.Ratio,
  510. LoginDate: key,
  511. }
  512. }
  513. if vv.ID == 7 {
  514. retention.Day7 = entity.DayRetention{
  515. LoggedIn: vv.Ratio,
  516. LoginDate: key,
  517. }
  518. }
  519. if vv.ID == 14 {
  520. retention.Day14 = entity.DayRetention{
  521. LoggedIn: vv.Ratio,
  522. LoginDate: key,
  523. }
  524. }
  525. if vv.ID == 30 {
  526. retention.Day30 = entity.DayRetention{
  527. LoggedIn: vv.Ratio,
  528. LoginDate: key,
  529. }
  530. }
  531. }
  532. results = append(results, &entity.UserRetentionDetail{
  533. RegistrationDate: key,
  534. RetentionData: retention,
  535. })
  536. }
  537. return &entity.UserRetentionResp{
  538. Details: results,
  539. }, nil
  540. }
  541. // FindUserLevel 用户等级统计
  542. func (s *Synthesis) FindUserLevel() (*entity.UserLevelCountResp, *code.Result) {
  543. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  544. defer cancel()
  545. collection := mdb.MDB.Collection(constant.CNamePlayerLevelStat)
  546. // 查询所有文档
  547. cursor, err := collection.Find(ctx, bson.M{})
  548. if err != nil {
  549. mhayaLogger.Warnf("FindUserLevel Find error:%v", err)
  550. return nil, common.NewResult(code.InternalError)
  551. }
  552. defer cursor.Close(ctx)
  553. var results []*entity.UserLevelCountDetail
  554. var reData []*models.PlayerLevelStat
  555. for cursor.Next(ctx) {
  556. var result models.PlayerLevelStat
  557. if err := cursor.Decode(&result); err != nil {
  558. mhayaLogger.Warnf("FindUserLevel Decode error:%v", err)
  559. return nil, common.NewResult(code.InternalError)
  560. }
  561. reData = append(reData, &result)
  562. }
  563. if err := cursor.Err(); err != nil {
  564. mhayaLogger.Warnf("FindUserLevel cursor error:%v", err)
  565. return nil, common.NewResult(code.InternalError)
  566. }
  567. dataMap := make(map[string]int)
  568. for _, v := range reData {
  569. for key, v1 := range v.ServerLevel {
  570. dataMap[key] += v1
  571. }
  572. }
  573. for k, v := range dataMap {
  574. results = append(results, &entity.UserLevelCountDetail{
  575. Level: cast.ToInt(k),
  576. UserCount: v,
  577. })
  578. }
  579. return &entity.UserLevelCountResp{
  580. Details: results,
  581. }, nil
  582. }
  583. func (s *Synthesis) InsertRecord(param model.UserOperationLog) {
  584. mhayaLogger.Warnf("InsertRecord param:%#v", param)
  585. record := new(model.UserOperationLog)
  586. collection := mdb.MDB.Collection(record.TableName())
  587. insertData := bson.M{}
  588. insertData["user_name"] = param.Username
  589. insertData["role_id"] = param.RoleId
  590. insertData["url"] = param.Path
  591. insertData["method"] = param.Method
  592. insertData["status_code"] = param.StatusCode
  593. insertData["dur"] = param.Dur
  594. insertData["client_ip"] = param.ClientIP
  595. insertData["error_message"] = param.ErrorMessage
  596. insertData["created_at"] = mhayaTime.Now().Unix()
  597. _, err := collection.InsertOne(context.Background(), insertData)
  598. if err != nil {
  599. mhayaLogger.Warnf("InsertRecord InsertOne error:%v", err)
  600. return
  601. }
  602. }
  603. func (s *Synthesis) RecordList(req entity.RecordListReq) (*entity.RecordListResp, *code.Result) {
  604. page, pageSize := checkPageParam(req.Page, req.Size)
  605. var records []*eventmodels.BackendOperationEventContent
  606. where := &eventmodels.BackendOperationEventContent{
  607. UserBasic: eventmodels.UserBasic{
  608. UserName: req.UserName,
  609. },
  610. EventBasic: eventmodels.EventBasic{
  611. ServerId: s.nodeId,
  612. },
  613. RoleId: req.RoleId,
  614. Path: req.Path,
  615. StatusCode: req.StatusCode,
  616. ClientIP: req.ClientIP,
  617. }
  618. db := mdb.LogstashDB.Model(&eventmodels.BackendOperationEventContent{}).Where(where).Order("create_at")
  619. if req.StartTime > 0 && req.EndTime > 0 && req.StartTime <= req.EndTime {
  620. db = db.Where("create_at >= ? and create_at <= ?", req.StartTime, req.EndTime)
  621. }
  622. if req.DurMin > 0 && req.DurMax > 0 && req.DurMin <= req.DurMax {
  623. db = db.Where("dur >= ? and dur <= ?", req.DurMin, req.DurMax)
  624. }
  625. pages := Paginate(db, page, pageSize)
  626. err := db.Scopes(pages.Limit).Find(&records).Error
  627. if err != nil && err != gorm.ErrRecordNotFound {
  628. mhayaLogger.Warnf("RecordList Find error:%v", err)
  629. return nil, common.NewResult(code.InternalError)
  630. }
  631. var results []*entity.RecordListDetail
  632. for _, v := range records {
  633. results = append(results, &entity.RecordListDetail{
  634. Username: v.UserName,
  635. RoleId: v.RoleId,
  636. Path: v.Path,
  637. Method: v.Method,
  638. StatusCode: v.StatusCode,
  639. Dur: v.Dur,
  640. ClientIP: v.ClientIP,
  641. ErrorMessage: v.ErrorMessage,
  642. CreatedAt: v.CreateAt,
  643. })
  644. }
  645. return &entity.RecordListResp{
  646. Details: results,
  647. Total: pages.Count,
  648. }, nil
  649. }
  650. // 转盘统计
  651. func (s *Synthesis) Turntable(req entity.TurntableReq) (*entity.TurntableResp, *code.Result) {
  652. page, pageSize := checkPageParam(req.Page, req.Size)
  653. // 构建查询条件
  654. filter := bson.M{}
  655. if req.UserName != "" {
  656. filter["userName"] = req.UserName
  657. }
  658. if req.OpenId != "" {
  659. filter["openId"] = req.OpenId
  660. }
  661. // 设置分页选项
  662. findOptions := options.Find()
  663. findOptions.SetSkip(int64((page - 1) * pageSize))
  664. findOptions.SetLimit(int64(pageSize))
  665. findOptions.SetSort(bson.D{{"createTime", -1}})
  666. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  667. defer cancel()
  668. collection := mdb.MDB.Collection(constant.CNamePlayer)
  669. // 查询数据
  670. var results []*entity.TurntableDetail
  671. cursor, err := collection.Find(ctx, filter, findOptions)
  672. if err != nil {
  673. mhayaLogger.Warnf("Turntable Find error:%v", err)
  674. return nil, common.NewResult(code.InternalError)
  675. }
  676. defer cursor.Close(ctx)
  677. // 解析结果
  678. for cursor.Next(ctx) {
  679. var result entity.TurntableDetail
  680. if err := cursor.Decode(&result); err != nil {
  681. mhayaLogger.Warnf("Turntable Decode error:%v", err)
  682. return nil, common.NewResult(code.InternalError)
  683. }
  684. results = append(results, &result)
  685. }
  686. if err := cursor.Err(); err != nil {
  687. mhayaLogger.Warnf("Turntable cursor error:%v", err)
  688. return nil, common.NewResult(code.InternalError)
  689. }
  690. // TODO 转盘统计相关信息
  691. // TODO 获取总数total
  692. var count int64
  693. return &entity.TurntableResp{
  694. Details: results,
  695. Total: count,
  696. }, nil
  697. }
  698. // 资产统计
  699. func (s *Synthesis) Assets(req entity.AssetsReq) (*entity.AssetsResp, *code.Result) {
  700. page, pageSize := checkPageParam(req.Page, req.Size)
  701. // 构建查询条件
  702. filter := bson.M{}
  703. if req.UserName != "" {
  704. filter["userName"] = req.UserName
  705. }
  706. if req.OpenId != "" {
  707. filter["openId"] = req.OpenId
  708. }
  709. // 设置分页选项
  710. findOptions := options.Find()
  711. findOptions.SetSkip(int64((page - 1) * pageSize))
  712. findOptions.SetLimit(int64(pageSize))
  713. findOptions.SetSort(bson.D{{"createTime", -1}})
  714. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  715. defer cancel()
  716. collection := mdb.MDB.Collection(constant.CNamePlayer)
  717. // 查询数据
  718. var results []*entity.AssetsDetail
  719. cursor, err := collection.Find(ctx, filter, findOptions)
  720. if err != nil {
  721. mhayaLogger.Warnf("Assets Find error:%v", err)
  722. return nil, common.NewResult(code.InternalError)
  723. }
  724. defer cursor.Close(ctx)
  725. // 解析结果
  726. for cursor.Next(ctx) {
  727. var result entity.AssetsDetail
  728. if err := cursor.Decode(&result); err != nil {
  729. mhayaLogger.Warnf("Assets Decode error:%v", err)
  730. return nil, common.NewResult(code.InternalError)
  731. }
  732. results = append(results, &result)
  733. }
  734. if err := cursor.Err(); err != nil {
  735. mhayaLogger.Warnf("Assets cursor error:%v", err)
  736. return nil, common.NewResult(code.InternalError)
  737. }
  738. // TODO Assets 统计相关信息
  739. // TODO 获取总数total
  740. var count int64
  741. return &entity.AssetsResp{
  742. Details: results,
  743. Total: count,
  744. }, nil
  745. }
  746. // 资产变动记录
  747. func (s *Synthesis) AssetsRecord(req entity.AssetsRecordReq) (*entity.AssetsRecordResp, *code.Result) {
  748. page, pageSize := checkPageParam(req.Page, req.Size)
  749. // 构建查询条件
  750. filter := bson.M{}
  751. if req.UserName != "" {
  752. filter["userName"] = req.UserName
  753. }
  754. if req.OpenId != "" {
  755. filter["openId"] = req.OpenId
  756. }
  757. // 设置分页选项
  758. findOptions := options.Find()
  759. findOptions.SetSkip(int64((page - 1) * pageSize))
  760. findOptions.SetLimit(int64(pageSize))
  761. findOptions.SetSort(bson.D{{"createTime", -1}})
  762. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  763. defer cancel()
  764. collection := mdb.MDB.Collection(constant.CNamePlayer)
  765. // 查询数据
  766. var results []*entity.AssetsRecordDetail
  767. cursor, err := collection.Find(ctx, filter, findOptions)
  768. if err != nil {
  769. mhayaLogger.Warnf("AssetsRecord Find error:%v", err)
  770. return nil, common.NewResult(code.InternalError)
  771. }
  772. defer cursor.Close(ctx)
  773. // 解析结果
  774. for cursor.Next(ctx) {
  775. var result entity.AssetsRecordDetail
  776. if err := cursor.Decode(&result); err != nil {
  777. mhayaLogger.Warnf("AssetsRecord Decode error:%v", err)
  778. return nil, common.NewResult(code.InternalError)
  779. }
  780. results = append(results, &result)
  781. }
  782. if err := cursor.Err(); err != nil {
  783. mhayaLogger.Warnf("AssetsRecord cursor error:%v", err)
  784. return nil, common.NewResult(code.InternalError)
  785. }
  786. // TODO 资产变动记录相关信息
  787. // TODO 获取总数total
  788. var count int64
  789. return &entity.AssetsRecordResp{
  790. Details: results,
  791. Total: count,
  792. }, nil
  793. }
  794. // 邀请统计
  795. func (s *Synthesis) Invite(req entity.InviteReq) (*entity.InviteResp, *code.Result) {
  796. page, pageSize := checkPageParam(req.Page, req.Size)
  797. // 构建查询条件
  798. filter := bson.M{}
  799. if req.UserName != "" {
  800. filter["userName"] = req.UserName
  801. }
  802. if req.OpenId != "" {
  803. filter["openId"] = req.OpenId
  804. }
  805. // 设置分页选项
  806. findOptions := options.Find()
  807. findOptions.SetSkip(int64((page - 1) * pageSize))
  808. findOptions.SetLimit(int64(pageSize))
  809. findOptions.SetSort(bson.D{{"createTime", -1}})
  810. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  811. defer cancel()
  812. collection := mdb.MDB.Collection(constant.CNamePlayer)
  813. // 查询数据
  814. var results []*entity.InviteDetail
  815. cursor, err := collection.Find(ctx, filter, findOptions)
  816. if err != nil {
  817. mhayaLogger.Warnf("Invite Find error:%v", err)
  818. return nil, common.NewResult(code.InternalError)
  819. }
  820. defer cursor.Close(ctx)
  821. // 解析结果
  822. for cursor.Next(ctx) {
  823. var result entity.InviteDetail
  824. if err := cursor.Decode(&result); err != nil {
  825. mhayaLogger.Warnf("Invite Decode error:%v", err)
  826. return nil, common.NewResult(code.InternalError)
  827. }
  828. results = append(results, &result)
  829. }
  830. if err := cursor.Err(); err != nil {
  831. mhayaLogger.Warnf("Invite cursor error:%v", err)
  832. return nil, common.NewResult(code.InternalError)
  833. }
  834. // TODO 邀请统计相关信息
  835. // TODO 获取总数total
  836. var count int64
  837. return &entity.InviteResp{
  838. Details: results,
  839. Total: count,
  840. }, nil
  841. }
  842. // 活跃统计
  843. func (s *Synthesis) Active(req entity.ActiveReq) (*entity.ActiveResp, *code.Result) {
  844. page, pageSize := checkPageParam(req.Page, req.Size)
  845. // 构建查询条件
  846. filter := bson.M{}
  847. if req.UserName != "" {
  848. filter["userName"] = req.UserName
  849. }
  850. if req.OpenId != "" {
  851. filter["openId"] = req.OpenId
  852. }
  853. // 设置分页选项
  854. findOptions := options.Find()
  855. findOptions.SetSkip(int64((page - 1) * pageSize))
  856. findOptions.SetLimit(int64(pageSize))
  857. findOptions.SetSort(bson.D{{"createTime", -1}})
  858. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  859. defer cancel()
  860. collection := mdb.MDB.Collection(constant.CNamePlayer)
  861. // 查询数据
  862. var results []*entity.ActiveDetail
  863. cursor, err := collection.Find(ctx, filter, findOptions)
  864. if err != nil {
  865. mhayaLogger.Warnf("Active Find error:%v", err)
  866. return nil, common.NewResult(code.InternalError)
  867. }
  868. defer cursor.Close(ctx)
  869. // 解析结果
  870. for cursor.Next(ctx) {
  871. var result entity.ActiveDetail
  872. if err := cursor.Decode(&result); err != nil {
  873. mhayaLogger.Warnf("Active Decode error:%v", err)
  874. return nil, common.NewResult(code.InternalError)
  875. }
  876. results = append(results, &result)
  877. }
  878. if err := cursor.Err(); err != nil {
  879. mhayaLogger.Warnf("Active cursor error:%v", err)
  880. return nil, common.NewResult(code.InternalError)
  881. }
  882. // TODO Active 活跃统计相关信息
  883. // TODO 获取总数total
  884. var count int64
  885. return &entity.ActiveResp{
  886. Details: results,
  887. Total: count,
  888. }, nil
  889. }
  890. // 任务完成度统计
  891. func (s *Synthesis) TaskCompletion(req entity.TaskCompletionReq) (*entity.TaskCompletionResp, *code.Result) {
  892. page, pageSize := checkPageParam(req.Page, req.Size)
  893. // 构建查询条件
  894. filter := bson.M{}
  895. if req.UserName != "" {
  896. filter["userName"] = req.UserName
  897. }
  898. if req.OpenId != "" {
  899. filter["openId"] = req.OpenId
  900. }
  901. // 设置分页选项
  902. findOptions := options.Find()
  903. findOptions.SetSkip(int64((page - 1) * pageSize))
  904. findOptions.SetLimit(int64(pageSize))
  905. findOptions.SetSort(bson.D{{"createTime", -1}})
  906. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  907. defer cancel()
  908. collection := mdb.MDB.Collection(constant.CNamePlayer)
  909. // 查询数据
  910. var results []*entity.TaskCompletionDetail
  911. cursor, err := collection.Find(ctx, filter, findOptions)
  912. if err != nil {
  913. mhayaLogger.Warnf("TaskCompletion Find error:%v", err)
  914. return nil, common.NewResult(code.InternalError)
  915. }
  916. defer cursor.Close(ctx)
  917. // 解析结果
  918. for cursor.Next(ctx) {
  919. var result entity.TaskCompletionDetail
  920. if err := cursor.Decode(&result); err != nil {
  921. mhayaLogger.Warnf("TaskCompletion Decode error:%v", err)
  922. return nil, common.NewResult(code.InternalError)
  923. }
  924. results = append(results, &result)
  925. }
  926. if err := cursor.Err(); err != nil {
  927. mhayaLogger.Warnf("TaskCompletion cursor error:%v", err)
  928. return nil, common.NewResult(code.InternalError)
  929. }
  930. // TODO 任务完成度统计相关信息
  931. // TODO 获取总数total
  932. var count int64
  933. return &entity.TaskCompletionResp{
  934. Details: results,
  935. Total: count,
  936. }, nil
  937. }
  938. // 用户行为检测
  939. func (s *Synthesis) BehaviorMonitoring(req entity.BehaviorMonitoringReq) (*entity.BehaviorMonitoringResp, *code.Result) {
  940. page, pageSize := checkPageParam(req.Page, req.Size)
  941. // 构建查询条件
  942. filter := bson.M{}
  943. if req.UserName != "" {
  944. filter["userName"] = req.UserName
  945. }
  946. if req.OpenId != "" {
  947. filter["openId"] = req.OpenId
  948. }
  949. // 设置分页选项
  950. findOptions := options.Find()
  951. findOptions.SetSkip(int64((page - 1) * pageSize))
  952. findOptions.SetLimit(int64(pageSize))
  953. findOptions.SetSort(bson.D{{"createTime", -1}})
  954. ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
  955. defer cancel()
  956. collection := mdb.MDB.Collection(constant.CNamePlayer)
  957. // 查询数据
  958. var results []*entity.BehaviorMonitoringDetail
  959. cursor, err := collection.Find(ctx, filter, findOptions)
  960. if err != nil {
  961. mhayaLogger.Warnf("BehaviorMonitoring Find error:%v", err)
  962. return nil, common.NewResult(code.InternalError)
  963. }
  964. defer cursor.Close(ctx)
  965. // 解析结果
  966. for cursor.Next(ctx) {
  967. var result entity.BehaviorMonitoringDetail
  968. if err := cursor.Decode(&result); err != nil {
  969. mhayaLogger.Warnf("BehaviorMonitoring Decode error:%v", err)
  970. return nil, common.NewResult(code.InternalError)
  971. }
  972. results = append(results, &result)
  973. }
  974. if err := cursor.Err(); err != nil {
  975. mhayaLogger.Warnf("BehaviorMonitoring cursor error:%v", err)
  976. return nil, common.NewResult(code.InternalError)
  977. }
  978. // TODO 用户行为检测相关信息
  979. // TODO 获取总数total
  980. var count int64
  981. return &entity.BehaviorMonitoringResp{
  982. Details: results,
  983. Total: count,
  984. }, nil
  985. }