shanshanzhong
c582087c0f
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/hibiken/asynq"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"github.com/perfect-panel/server/pkg/tool"
|
|
task "github.com/perfect-panel/server/queue/types"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
//goland:noinspection GoNameStartsWithPackageName
|
|
type ServerPushUserTrafficLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// NewServerPushUserTrafficLogic Push user Traffic
|
|
func NewServerPushUserTrafficLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ServerPushUserTrafficLogic {
|
|
return &ServerPushUserTrafficLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *ServerPushUserTrafficLogic) ServerPushUserTraffic(req *types.ServerPushUserTrafficRequest) error {
|
|
// Find server info
|
|
serverInfo, err := l.svcCtx.NodeModel.FindOneServer(l.ctx, req.ServerId)
|
|
if err != nil {
|
|
l.Errorw("[PushOnlineUsers] FindOne error", logger.Field("error", err))
|
|
return errors.New("server not found")
|
|
}
|
|
|
|
// Create traffic task
|
|
var request task.TrafficStatistics
|
|
request.ServerId = serverInfo.Id
|
|
request.Protocol = req.Protocol
|
|
tool.DeepCopy(&request.Logs, req.Traffic)
|
|
|
|
// Push traffic task
|
|
val, _ := json.Marshal(request)
|
|
t := asynq.NewTask(task.ForthwithTrafficStatistics, val, asynq.MaxRetry(3))
|
|
info, err := l.svcCtx.Queue.EnqueueContext(l.ctx, t)
|
|
if err != nil {
|
|
l.Errorw("[ServerPushUserTraffic] Push traffic task error", logger.Field("error", err.Error()), logger.Field("task", t))
|
|
} else {
|
|
l.Infow("[ServerPushUserTraffic] Push traffic task success", logger.Field("task", t), logger.Field("info", info))
|
|
}
|
|
|
|
// Update server last reported time
|
|
now := time.Now()
|
|
serverInfo.LastReportedAt = &now
|
|
|
|
err = l.svcCtx.NodeModel.UpdateServer(l.ctx, serverInfo)
|
|
if err != nil {
|
|
l.Errorw("[ServerPushUserTraffic] UpdateServer error", logger.Field("error", err))
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|