init: 1.0.0

2025-04-25 12:08:29 +09:00
commit 8addcc584b
1031 changed files with 76472 additions and 0 deletions
@@ -0,0 +1,166 @@
+package traffic
+
+import (
+	"context"
+	"encoding/json"
+	"time"
+
+	"github.com/perfect-panel/ppanel-server/pkg/logger"
+
+	"github.com/hibiken/asynq"
+	"github.com/perfect-panel/ppanel-server/internal/config"
+	"github.com/perfect-panel/ppanel-server/internal/svc"
+	"github.com/perfect-panel/ppanel-server/internal/types"
+)
+
+type ServerDataLogic struct {
+	svc *svc.ServiceContext
+}
+
+func NewServerDataLogic(svc *svc.ServiceContext) *ServerDataLogic {
+	return &ServerDataLogic{
+		svc: svc,
+	}
+}
+
+func (l *ServerDataLogic) ProcessTask(ctx context.Context, _ *asynq.Task) error {
+	serverData := types.ServerTotalDataResponse{}
+
+	top10ServerToday, top10ServerYesterday, top10UserToday, top10UserYesterday := l.getRanking(ctx)
+	if len(top10ServerToday) == 0 {
+		top10ServerToday = make([]types.ServerTrafficData, 0)
+	}
+	if len(top10ServerYesterday) == 0 {
+		top10ServerYesterday = make([]types.ServerTrafficData, 0)
+	}
+	if len(top10UserToday) == 0 {
+		top10UserToday = make([]types.UserTrafficData, 0)
+	}
+	if len(top10UserYesterday) == 0 {
+		top10UserYesterday = make([]types.UserTrafficData, 0)
+	}
+	serverData.ServerTrafficRankingToday = top10ServerToday
+	serverData.ServerTrafficRankingYesterday = top10ServerYesterday
+	serverData.UserTrafficRankingToday = top10UserToday
+	serverData.UserTrafficRankingYesterday = top10UserYesterday
+	totalUploadToday, totalDownloadToday, totalDownloadMonthly, totalUploadMonthly := l.trafficCount(ctx)
+	serverData.TodayUpload = totalUploadToday
+	serverData.TodayDownload = totalDownloadToday
+	serverData.MonthlyUpload = totalUploadMonthly
+	serverData.MonthlyDownload = totalDownloadMonthly
+	serverData.UpdatedAt = time.Now().UnixMilli()
+	data, err := json.Marshal(serverData)
+	if err != nil {
+		logger.Error("[ServerDataLogic] Marshal server data failed", logger.Field("error", err.Error()), logger.Field("data", serverData))
+		return err
+	}
+	if err := l.svc.Redis.Set(ctx, config.ServerCountCacheKey, data, -1).Err(); err != nil {
+		logger.Error("[ServerDataLogic] Set server data failed", logger.Field("error", err.Error()))
+		return err
+	}
+	logger.Info("[ServerDataLogic] Update server data success")
+	return nil
+}
+
+func (l *ServerDataLogic) getRanking(ctx context.Context) (top10ServerToday, top10ServerYesterday []types.ServerTrafficData, top10UserToday, top10UserYesterday []types.UserTrafficData) {
+	now := time.Now()
+	// 获取服务器流量排行榜
+	serverToday, err := l.svc.TrafficLogModel.TopServersTrafficByDay(ctx, now, 10)
+	if err != nil {
+		logger.Error("[ServerDataLogic] Get top servers traffic by day failed", logger.Field("error", err.Error()))
+	} else {
+		for _, s := range serverToday {
+			if s.ServerId == 0 {
+				continue
+			}
+			serverInfo, err := l.svc.ServerModel.FindOne(ctx, s.ServerId)
+			if err != nil {
+				logger.Error("[ServerDataLogic] Find server failed", logger.Field("error", err.Error()))
+				continue
+			}
+			top10ServerToday = append(top10ServerToday, types.ServerTrafficData{
+				ServerId: s.ServerId,
+				Name:     serverInfo.Name,
+				Upload:   s.Upload,
+				Download: s.Download,
+			})
+		}
+	}
+
+	serverYesterday, err := l.svc.TrafficLogModel.TopServersTrafficByDay(ctx, now.AddDate(0, 0, -1), 10)
+	if err != nil {
+		logger.Error("[ServerDataLogic] Get top servers traffic by day failed", logger.Field("error", err.Error()))
+	} else {
+		for _, s := range serverYesterday {
+			serverInfo, err := l.svc.ServerModel.FindOne(ctx, s.ServerId)
+			if err != nil {
+				logger.Error("[ServerDataLogic] Find server failed", logger.Field("error", err.Error()))
+				continue
+			}
+			top10ServerYesterday = append(top10ServerYesterday, types.ServerTrafficData{
+				ServerId: s.ServerId,
+				Name:     serverInfo.Name,
+				Upload:   s.Upload,
+				Download: s.Download,
+			})
+		}
+	}
+
+	// 获取用户流量排行榜
+	userToday, err := l.svc.TrafficLogModel.TopUsersTrafficByDay(ctx, now, 10)
+	if err != nil {
+		logger.Error("[ServerDataLogic] Get top users traffic by day failed", logger.Field("error", err.Error()))
+	} else {
+		for _, u := range userToday {
+			//userInfo, err := l.svc.UserModel.FindOne(ctx, u.UserId)
+			//if err != nil {
+			//	logx.Error("[ServerDataLogic] Find user failed", logx.Field("error", err.Error()))
+			//	continue
+			//}
+			top10UserToday = append(top10UserToday, types.UserTrafficData{
+				SID:      u.UserId,
+				Upload:   u.Upload,
+				Download: u.Download,
+			})
+		}
+	}
+
+	userYesterday, err := l.svc.TrafficLogModel.TopUsersTrafficByDay(ctx, now.AddDate(0, 0, -1), 10)
+	if err != nil {
+		logger.Error("[ServerDataLogic] Get top users traffic by day failed", logger.Field("error", err.Error()))
+	} else {
+		for _, u := range userYesterday {
+			//userInfo, err := l.svc.UserModel.FindOne(ctx, u.UserId)
+			//if err != nil {
+			//	logx.Error("[ServerDataLogic] Find user failed", logx.Field("error", err.Error()))
+			//	continue
+			//}
+			top10UserYesterday = append(top10UserYesterday, types.UserTrafficData{
+				SID:      u.UserId,
+				Upload:   u.Upload,
+				Download: u.Download,
+			})
+		}
+	}
+	return
+}
+
+func (l *ServerDataLogic) trafficCount(ctx context.Context) (totalUploadToday, totalDownloadToday, totalDownloadMonthly, totalUploadMonthly int64) {
+	now := time.Now()
+	today, err := l.svc.TrafficLogModel.QueryTrafficByDay(ctx, now)
+	if err != nil {
+		logger.Error("[ServerDataLogic] Query traffic by day failed", logger.Field("error", err.Error()))
+	} else {
+		totalUploadToday = today.Upload
+		totalDownloadToday = today.Download
+	}
+
+	monthly, err := l.svc.TrafficLogModel.QueryTrafficByMonthly(ctx, now)
+	if err != nil {
+		logger.Error("[ServerDataLogic] Query traffic by monthly failed", logger.Field("error", err.Error()))
+	} else {
+		totalUploadMonthly = monthly.Upload
+		totalDownloadMonthly = monthly.Download
+	}
+	return
+}
@@ -0,0 +1,98 @@
+package traffic
+
+import (
+	"context"
+	"encoding/json"
+	"time"
+
+	"github.com/perfect-panel/ppanel-server/pkg/logger"
+
+	"github.com/hibiken/asynq"
+	"github.com/perfect-panel/ppanel-server/internal/model/traffic"
+	"github.com/perfect-panel/ppanel-server/internal/svc"
+	"github.com/perfect-panel/ppanel-server/queue/types"
+)
+
+//goland:noinspection GoNameStartsWithPackageName
+type TrafficStatisticsLogic struct {
+	svc *svc.ServiceContext
+}
+
+func NewTrafficStatisticsLogic(svc *svc.ServiceContext) *TrafficStatisticsLogic {
+	return &TrafficStatisticsLogic{
+		svc: svc,
+	}
+}
+
+func (l *TrafficStatisticsLogic) ProcessTask(ctx context.Context, task *asynq.Task) error {
+	var payload types.TrafficStatistics
+	if err := json.Unmarshal(task.Payload(), &payload); err != nil {
+		logger.WithContext(ctx).Error("[TrafficStatistics] Unmarshal payload failed",
+			logger.Field("error", err.Error()),
+			logger.Field("payload", string(task.Payload())),
+		)
+		return nil
+	}
+	if len(payload.Logs) == 0 {
+		logger.WithContext(ctx).Error("[TrafficStatistics] Payload is empty")
+		return nil
+	}
+	// query server info
+	serverInfo, err := l.svc.ServerModel.FindOne(ctx, payload.ServerId)
+	if err != nil {
+		logger.WithContext(ctx).Error("[TrafficStatistics] Find server info failed",
+			logger.Field("serverId", payload.ServerId),
+			logger.Field("error", err.Error()),
+		)
+		return nil
+	}
+	if serverInfo.TrafficRatio == 0 {
+		logger.WithContext(ctx).Error("[TrafficStatistics] Server log ratio is 0",
+			logger.Field("serverId", payload.ServerId),
+		)
+		return nil
+	}
+	now := time.Now()
+	realTimeMultiplier := l.svc.NodeMultiplierManager.GetMultiplier(now)
+	for _, log := range payload.Logs {
+		// update user subscribe with log
+		d := int64(float32(log.Download) * serverInfo.TrafficRatio * realTimeMultiplier)
+		u := int64(float32(log.Upload) * serverInfo.TrafficRatio * realTimeMultiplier)
+		if err := l.svc.UserModel.UpdateUserSubscribeWithTraffic(ctx, log.SID, d, u); err != nil {
+			logger.WithContext(ctx).Error("[TrafficStatistics] Update user subscribe with log failed",
+				logger.Field("sid", log.SID),
+				logger.Field("download", float32(log.Download)*serverInfo.TrafficRatio),
+				logger.Field("upload", float32(log.Upload)*serverInfo.TrafficRatio),
+				logger.Field("error", err.Error()),
+			)
+			continue
+		}
+		// query user Subscribe Info
+		sub, err := l.svc.UserModel.FindOneSubscribe(ctx, log.SID)
+		if err != nil {
+			logger.WithContext(ctx).Error("[TrafficStatistics] Find user Subscribe Info failed",
+				logger.Field("uid", log.SID),
+				logger.Field("error", err.Error()),
+			)
+			continue
+		}
+
+		// create log log
+		if err := l.svc.TrafficLogModel.Insert(ctx, &traffic.TrafficLog{
+			ServerId:    payload.ServerId,
+			SubscribeId: log.SID,
+			UserId:      sub.UserId,
+			Upload:      u,
+			Download:    d,
+			Timestamp:   now,
+		}); err != nil {
+			logger.WithContext(ctx).Error("[TrafficStatistics] Create log log failed",
+				logger.Field("uid", log.SID),
+				logger.Field("download", float32(log.Download)*serverInfo.TrafficRatio),
+				logger.Field("upload", float32(log.Upload)*serverInfo.TrafficRatio),
+				logger.Field("error", err.Error()),
+			)
+		}
+	}
+	return nil
+}