06a2425474
- Add subscribe traffic_limit schema and migration\n- Support traffic_limit in admin create/update and list/details\n- Apply traffic_limit when building server user list speed limits\n- Add public user traffic stats API
104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
package group
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
"github.com/perfect-panel/server/internal/model/group"
|
||
"github.com/perfect-panel/server/internal/model/node"
|
||
"github.com/perfect-panel/server/internal/svc"
|
||
"github.com/perfect-panel/server/internal/types"
|
||
"github.com/perfect-panel/server/pkg/logger"
|
||
)
|
||
|
||
type GetNodeGroupListLogic struct {
|
||
logger.Logger
|
||
ctx context.Context
|
||
svcCtx *svc.ServiceContext
|
||
}
|
||
|
||
func NewGetNodeGroupListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetNodeGroupListLogic {
|
||
return &GetNodeGroupListLogic{
|
||
Logger: logger.WithContext(ctx),
|
||
ctx: ctx,
|
||
svcCtx: svcCtx,
|
||
}
|
||
}
|
||
|
||
func (l *GetNodeGroupListLogic) GetNodeGroupList(req *types.GetNodeGroupListRequest) (resp *types.GetNodeGroupListResponse, err error) {
|
||
var nodeGroups []group.NodeGroup
|
||
var total int64
|
||
|
||
// 构建查询
|
||
query := l.svcCtx.DB.Model(&group.NodeGroup{})
|
||
|
||
// 获取总数
|
||
if err := query.Count(&total).Error; err != nil {
|
||
logger.Errorf("failed to count node groups: %v", err)
|
||
return nil, err
|
||
}
|
||
|
||
// 分页查询
|
||
offset := (req.Page - 1) * req.Size
|
||
if err := query.Order("sort ASC").Offset(offset).Limit(req.Size).Find(&nodeGroups).Error; err != nil {
|
||
logger.Errorf("failed to find node groups: %v", err)
|
||
return nil, err
|
||
}
|
||
|
||
// 转换为响应格式
|
||
var list []types.NodeGroup
|
||
for _, ng := range nodeGroups {
|
||
// 统计该组的节点数（JSON数组查询）
|
||
var nodeCount int64
|
||
l.svcCtx.DB.Model(&node.Node{}).Where("JSON_CONTAINS(node_group_ids, ?)", fmt.Sprintf("[%d]", ng.Id)).Count(&nodeCount)
|
||
|
||
// 处理指针类型的字段
|
||
var forCalculation bool
|
||
if ng.ForCalculation != nil {
|
||
forCalculation = *ng.ForCalculation
|
||
} else {
|
||
forCalculation = true // 默认值
|
||
}
|
||
|
||
var isExpiredGroup bool
|
||
if ng.IsExpiredGroup != nil {
|
||
isExpiredGroup = *ng.IsExpiredGroup
|
||
}
|
||
|
||
var minTrafficGB, maxTrafficGB, maxTrafficGBExpired int64
|
||
if ng.MinTrafficGB != nil {
|
||
minTrafficGB = *ng.MinTrafficGB
|
||
}
|
||
if ng.MaxTrafficGB != nil {
|
||
maxTrafficGB = *ng.MaxTrafficGB
|
||
}
|
||
if ng.MaxTrafficGBExpired != nil {
|
||
maxTrafficGBExpired = *ng.MaxTrafficGBExpired
|
||
}
|
||
|
||
list = append(list, types.NodeGroup{
|
||
Id: ng.Id,
|
||
Name: ng.Name,
|
||
Description: ng.Description,
|
||
Sort: ng.Sort,
|
||
ForCalculation: forCalculation,
|
||
IsExpiredGroup: isExpiredGroup,
|
||
ExpiredDaysLimit: ng.ExpiredDaysLimit,
|
||
MaxTrafficGBExpired: maxTrafficGBExpired,
|
||
SpeedLimit: ng.SpeedLimit,
|
||
MinTrafficGB: minTrafficGB,
|
||
MaxTrafficGB: maxTrafficGB,
|
||
NodeCount: nodeCount,
|
||
CreatedAt: ng.CreatedAt.Unix(),
|
||
UpdatedAt: ng.UpdatedAt.Unix(),
|
||
})
|
||
}
|
||
|
||
resp = &types.GetNodeGroupListResponse{
|
||
Total: total,
|
||
List: list,
|
||
}
|
||
|
||
return resp, nil
|
||
}
|