86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/perfect-panel/server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/perfect-panel/server/internal/model/traffic"
|
|
"github.com/perfect-panel/server/internal/model/user"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/pkg/constant"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type GetUserSubscribeTrafficLogsLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// Get user subcribe traffic logs
|
|
func NewGetUserSubscribeTrafficLogsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUserSubscribeTrafficLogsLogic {
|
|
return &GetUserSubscribeTrafficLogsLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetUserSubscribeTrafficLogsLogic) GetUserSubscribeTrafficLogs(req *types.GetUserSubscribeTrafficLogsRequest) (resp *types.GetUserSubscribeTrafficLogsResponse, err error) {
|
|
resp = &types.GetUserSubscribeTrafficLogsResponse{}
|
|
u := l.ctx.Value(constant.CtxKeyUser).(*user.User)
|
|
var traffics []traffic.TrafficLog
|
|
err = l.svcCtx.UserModel.Transaction(l.ctx, func(db *gorm.DB) error {
|
|
return db.Model(traffic.TrafficLog{}).Where("user_id = ? and `timestamp` >= ? and `timestamp` < ?", u.Id, time.UnixMilli(req.StartTime), time.UnixMilli(req.EndTime)).Find(&traffics).Error
|
|
})
|
|
|
|
if err != nil {
|
|
l.Errorw("get user subscribe traffic logs failed", logger.Field("error", err.Error()))
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), err.Error())
|
|
}
|
|
|
|
//合并多条记录为以天为单位
|
|
trafficMap := make(map[string]*traffic.TrafficLog)
|
|
for _, traf := range traffics {
|
|
key := traf.Timestamp.Format(time.DateOnly)
|
|
existTraf := trafficMap[key]
|
|
if existTraf == nil {
|
|
trafficMap[key] = &traf
|
|
} else {
|
|
existTraf.Upload = existTraf.Download + traf.Upload
|
|
existTraf.Download = existTraf.Download + traf.Download
|
|
trafficMap[key] = existTraf
|
|
}
|
|
}
|
|
|
|
startTime := time.UnixMilli(req.StartTime)
|
|
EndTime := time.UnixMilli(req.EndTime)
|
|
res := make(map[string]traffic.TrafficLog)
|
|
|
|
// 循环遍历每一天
|
|
for current := startTime; !current.After(EndTime); current = current.AddDate(0, 0, 1) {
|
|
dateStr := current.Format(time.DateOnly) // 格式化为日期字符串
|
|
if trafficMap[dateStr] == nil {
|
|
res[dateStr] = traffic.TrafficLog{
|
|
Timestamp: current,
|
|
}
|
|
} else {
|
|
res[dateStr] = *trafficMap[dateStr]
|
|
}
|
|
resp.List = append(resp.List, types.TrafficLog{
|
|
Id: res[dateStr].Id,
|
|
ServerId: res[dateStr].ServerId,
|
|
Upload: res[dateStr].Upload,
|
|
Download: res[dateStr].Download,
|
|
Timestamp: res[dateStr].Timestamp.UnixMilli(),
|
|
})
|
|
}
|
|
|
|
return
|
|
}
|