TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-07 18:50: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

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

@@ -61,19 +61,24 @@ 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 !ok { if serverUserId == 0 {
userId, err := SharedServerDAO.FindServerUserId(tx, stat.ServerId) var ok bool
if err != nil { serverUserId, ok = serverUserMap[stat.ServerId]
return err if !ok {
userId, err := SharedServerDAO.FindServerUserId(tx, stat.ServerId)
if err != nil {
return err
}
serverUserMap[stat.ServerId] = userId
serverUserId = userId
} }
serverUserId = userId
} }
_, _, err := this.Query(tx). _, _, err := this.Query(tx).