shanshanzhong147
82eff47f38
- 新增 migration 02153: user_subscribe 表添加 speed_limit、traffic_limit 列（幂等） - model 层 Subscribe/SubscribeDetails 新增用户级限速字段（*string 类型支持 nil 回退） - server 用户列表和管理员详情接口支持用户级限速覆盖优先级计算 - 更新管理员订阅更新接口支持写入/清除用户级限速 Co-authored-by: multica-agent <github@multica.ai>
78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"github.com/perfect-panel/server/internal/model/group"
|
|
"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/speedlimit"
|
|
"github.com/perfect-panel/server/pkg/tool"
|
|
"github.com/perfect-panel/server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type GetUserSubscribeByIdLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// Get user subcribe by id
|
|
func NewGetUserSubscribeByIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUserSubscribeByIdLogic {
|
|
return &GetUserSubscribeByIdLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetUserSubscribeByIdLogic) GetUserSubscribeById(req *types.GetUserSubscribeByIdRequest) (resp *types.UserSubscribeDetail, err error) {
|
|
sub, err := l.svcCtx.UserModel.FindOneSubscribeDetailsById(l.ctx, req.Id)
|
|
if err != nil {
|
|
l.Errorw("[GetUserSubscribeByIdLogic] FindOneSubscribeDetailsById error", logger.Field("error", err.Error()))
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "FindOneSubscribeDetailsById error: %v", err.Error())
|
|
}
|
|
var subscribeDetails types.UserSubscribeDetail
|
|
tool.DeepCopy(&subscribeDetails, sub)
|
|
subscribeDetails.SpeedLimit = sub.SpeedLimit
|
|
if sub.TrafficLimit != nil && *sub.TrafficLimit != "" {
|
|
_ = json.Unmarshal([]byte(*sub.TrafficLimit), &subscribeDetails.TrafficLimit)
|
|
}
|
|
if sub.Subscribe != nil {
|
|
subscribeDetails.PlanSpeedLimit = sub.Subscribe.SpeedLimit
|
|
}
|
|
|
|
// 填充分组名
|
|
if sub.NodeGroupId > 0 {
|
|
var ng group.NodeGroup
|
|
if err := l.svcCtx.DB.WithContext(l.ctx).First(&ng, sub.NodeGroupId).Error; err == nil {
|
|
subscribeDetails.NodeGroupName = ng.Name
|
|
}
|
|
}
|
|
|
|
// Calculate speed limit status
|
|
if sub.Subscribe != nil && sub.Status == 1 {
|
|
baseSpeed := sub.Subscribe.SpeedLimit
|
|
if sub.SpeedLimit > 0 {
|
|
baseSpeed = sub.SpeedLimit
|
|
}
|
|
|
|
trafficLimit := sub.Subscribe.TrafficLimit
|
|
if sub.TrafficLimit != nil && *sub.TrafficLimit != "" {
|
|
trafficLimit = *sub.TrafficLimit
|
|
}
|
|
|
|
result := speedlimit.Calculate(l.ctx, l.svcCtx.DB, sub.UserId, sub.Id, baseSpeed, trafficLimit)
|
|
subscribeDetails.EffectiveSpeed = result.EffectiveSpeed
|
|
subscribeDetails.IsThrottled = result.IsThrottled
|
|
subscribeDetails.ThrottleRule = result.ThrottleRule
|
|
subscribeDetails.ThrottleStart = result.ThrottleStart
|
|
subscribeDetails.ThrottleEnd = result.ThrottleEnd
|
|
}
|
|
|
|
return &subscribeDetails, nil
|
|
}
|