TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-08 03:00:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

上传流量数据时同时上传服务所属用户ID

Browse Source
This commit is contained in:
GoEdgeLab
2023-03-22 19:33:25 +08:00
parent f58897bc74
commit 4879c783ed
1 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/db/models/server_daily_stat_dao.go
View File

@@ -61,20 +61,25 @@ func (this *ServerDailyStatDAO) SaveStats(tx *dbs.Tx, stats []*pb.ServerDailySta
var serverUserMap = map[int64]int64{} // serverId => userId var serverUserMap = map[int64]int64{} // serverId => userId
var cacheMap = utils.NewCacheMap() var cacheMap = utils.NewCacheMap()
for _, stat := range stats { for _, stat := range stats {
day := timeutil.FormatTime("Ymd", stat.CreatedAt) var day = timeutil.FormatTime("Ymd", stat.CreatedAt)
hour := timeutil.FormatTime("YmdH", stat.CreatedAt) var hour = timeutil.FormatTime("YmdH", stat.CreatedAt)
timeFrom := timeutil.FormatTime("His", stat.CreatedAt) var timeFrom = timeutil.FormatTime("His", stat.CreatedAt)
timeTo := timeutil.FormatTime("His", stat.CreatedAt+5*60-1) // 5分钟 var timeTo = timeutil.FormatTime("His", stat.CreatedAt+5*60-1) // 5分钟
// 所属用户 // 用户ID
serverUserId, ok := serverUserMap[stat.ServerId] var serverUserId = stat.UserId
if serverUserId == 0 {
var ok bool
serverUserId, ok = serverUserMap[stat.ServerId]
if !ok { if !ok {
userId, err := SharedServerDAO.FindServerUserId(tx, stat.ServerId) userId, err := SharedServerDAO.FindServerUserId(tx, stat.ServerId)
if err != nil { if err != nil {
return err return err
} }
serverUserMap[stat.ServerId] = userId
serverUserId = userId serverUserId = userId
} }
}
_, _, err := this.Query(tx). _, _, err := this.Query(tx).
Param("bytes", stat.Bytes). Param("bytes", stat.Bytes).