EUForest
39310d5b9a
- Node group CRUD operations with traffic-based filtering - Three grouping modes: average distribution, subscription-based, and traffic-based - Automatic and manual group recalculation with history tracking - Group assignment preview before applying changes - User subscription group locking to prevent automatic reassignment - Subscribe-to-group mapping configuration - Group calculation history and detailed reports - System configuration for group management (enabled/mode/auto_create) Database: - Add node_group table for group definitions - Add group_history and group_history_detail tables for tracking - Add node_group_ids (JSON) to nodes and subscribe tables - Add node_group_id and group_locked fields to user_subscribe table - Add migration files for schema changes
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package group
|
|
|
|
import (
|
|
"context"
|
|
|
|
"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/pkg/errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type GetRecalculationStatusLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// Get recalculation status
|
|
func NewGetRecalculationStatusLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetRecalculationStatusLogic {
|
|
return &GetRecalculationStatusLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetRecalculationStatusLogic) GetRecalculationStatus() (resp *types.RecalculationState, err error) {
|
|
// 返回最近的一条 GroupHistory 记录
|
|
var history group.GroupHistory
|
|
err = l.svcCtx.DB.Order("id desc").First(&history).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
// 如果没有历史记录，返回空闲状态
|
|
resp = &types.RecalculationState{
|
|
State: "idle",
|
|
Progress: 0,
|
|
Total: 0,
|
|
}
|
|
return resp, nil
|
|
}
|
|
l.Errorw("failed to get group history", logger.Field("error", err.Error()))
|
|
return nil, err
|
|
}
|
|
|
|
// 转换为 RecalculationState 格式
|
|
// Progress = 已处理的用户数（成功+失败），Total = 总用户数
|
|
processedUsers := history.SuccessCount + history.FailedCount
|
|
resp = &types.RecalculationState{
|
|
State: history.State,
|
|
Progress: processedUsers,
|
|
Total: history.TotalUsers,
|
|
}
|
|
|
|
return resp, nil
|
|
}
|