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

按天统计流量接口可以预估某日同时间流量

Browse Source
This commit is contained in:
GoEdgeLab
2022-04-10 21:25:24 +08:00
parent 9f33aae08b
commit b5b424d782
3 changed files with 112 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
internal/rpc/services/service_traffic_daily_stat.go
View File

@@ -4,9 +4,11 @@ package services
import (
"context"
"fmt"
"github.com/TeaOSLab/EdgeAPI/internal/db/models/stats"
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
timeutil "github.com/iwind/TeaGo/utils/time"
"github.com/iwind/TeaGo/types"
"regexp"
)
// TrafficDailyStatService 按日统计服务
@@ -22,10 +24,56 @@ func (this *TrafficDailyStatService) FindTrafficDailyStatWithDay(ctx context.Con
}
var tx = this.NullTx()
var day = timeutil.Format("Ymd")
stat, err := stats.SharedTrafficDailyStatDAO.FindDailyStat(tx, day)
if err != nil {
return nil, err
var day = req.Day
var stat = &stats.TrafficDailyStat{
Day: day,
}
if len(req.Minute) > 0 && regexp.MustCompile(`^\d{6}$`).MatchString(req.Minute) {
var hourString = req.Minute[:2]
var hourInt = types.Int(hourString)
var lastHourInt = hourInt - 1
// 过往小时
if lastHourInt >= 0 {
var hourFrom = day + "00"
var hourTo = day + fmt.Sprintf("%02d", lastHourInt)
sumStat, err := stats.SharedTrafficHourlyStatDAO.SumHourlyStats(tx, hourFrom, hourTo)
if err != nil {
return nil, err
}
if sumStat != nil {
stat = &stats.TrafficDailyStat{
Id: 0,
Day: day,
CachedBytes: sumStat.CachedBytes,
Bytes: sumStat.Bytes,
CountRequests: sumStat.CountRequests,
CountCachedRequests: sumStat.CountCachedRequests,
CountAttackRequests: sumStat.CountAttackRequests,
AttackBytes: sumStat.AttackBytes,
}
}
}
// 当前小时
hourStat, err := stats.SharedTrafficHourlyStatDAO.FindHourlyStat(tx, day+hourString)
if err != nil {
return nil, err
}
if hourStat != nil {
var seconds = types.Int(req.Minute[2:4])*60 + types.Int(req.Minute[4:]) + 1 /** 因为是0-59，所以+1 **/
stat.Bytes += hourStat.Bytes * uint64(seconds) / 3600
stat.CachedBytes += hourStat.CachedBytes * uint64(seconds) / 3600
stat.CountRequests += hourStat.CountRequests * uint64(seconds) / 3600
stat.CountCachedRequests += hourStat.CountCachedRequests * uint64(seconds) / 3600
stat.CountAttackRequests += hourStat.CountAttackRequests * uint64(seconds) / 3600
stat.AttackBytes += hourStat.AttackBytes * uint64(seconds) / 3600
}
} else {
stat, err = stats.SharedTrafficDailyStatDAO.FindDailyStat(tx, day)
if err != nil {
return nil, err
}
}
if stat == nil {