init: 1.0.0

internal/logic/app/user/getusersubscribetrafficlogslogic.go

@@ -0,0 +1,85 @@
+package user
+
+import (
+	"context"
+	"time"
+
+	"github.com/perfect-panel/ppanel-server/pkg/xerr"
+	"github.com/pkg/errors"
+
+	"github.com/perfect-panel/ppanel-server/internal/model/traffic"
+	"github.com/perfect-panel/ppanel-server/internal/model/user"
+	"github.com/perfect-panel/ppanel-server/internal/svc"
+	"github.com/perfect-panel/ppanel-server/internal/types"
+	"github.com/perfect-panel/ppanel-server/pkg/constant"
+	"github.com/perfect-panel/ppanel-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
+}