ario_server/queue/logic/traffic/serverDataLogic.go

package traffic

import (
	"context"
	"encoding/json"
	"time"

	"github.com/perfect-panel/server/pkg/logger"

	"github.com/hibiken/asynq"
	"github.com/perfect-panel/server/internal/config"
	"github.com/perfect-panel/server/internal/svc"
	"github.com/perfect-panel/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.NodeModel.FindOneServer(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.NodeModel.FindOneServer(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
}