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
88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package group
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/perfect-panel/server/internal/logic/admin/group"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
|
|
"github.com/hibiken/asynq"
|
|
)
|
|
|
|
type RecalculateGroupLogic struct {
|
|
svc *svc.ServiceContext
|
|
}
|
|
|
|
func NewRecalculateGroupLogic(svc *svc.ServiceContext) *RecalculateGroupLogic {
|
|
return &RecalculateGroupLogic{
|
|
svc: svc,
|
|
}
|
|
}
|
|
|
|
func (l *RecalculateGroupLogic) ProcessTask(ctx context.Context, t *asynq.Task) error {
|
|
logger.Infof("[RecalculateGroup] Starting scheduled group recalculation: %s", time.Now().Format("2006-01-02 15:04:05"))
|
|
|
|
// 1. Check if group management is enabled
|
|
var enabledConfig struct {
|
|
Value string `gorm:"column:value"`
|
|
}
|
|
err := l.svc.DB.Table("system").
|
|
Where("`category` = ? AND `key` = ?", "group", "enabled").
|
|
Select("value").
|
|
First(&enabledConfig).Error
|
|
if err != nil {
|
|
logger.Errorw("[RecalculateGroup] Failed to read group enabled config", logger.Field("error", err.Error()))
|
|
return err
|
|
}
|
|
|
|
// If not enabled, skip execution
|
|
if enabledConfig.Value != "true" && enabledConfig.Value != "1" {
|
|
logger.Debugf("[RecalculateGroup] Group management is not enabled, skipping")
|
|
return nil
|
|
}
|
|
|
|
// 2. Get grouping mode
|
|
var modeConfig struct {
|
|
Value string `gorm:"column:value"`
|
|
}
|
|
err = l.svc.DB.Table("system").
|
|
Where("`category` = ? AND `key` = ?", "group", "mode").
|
|
Select("value").
|
|
First(&modeConfig).Error
|
|
if err != nil {
|
|
logger.Errorw("[RecalculateGroup] Failed to read group mode config", logger.Field("error", err.Error()))
|
|
return err
|
|
}
|
|
|
|
mode := modeConfig.Value
|
|
if mode == "" {
|
|
mode = "average" // default mode
|
|
}
|
|
|
|
// 3. Only execute if mode is "traffic"
|
|
if mode != "traffic" {
|
|
logger.Debugf("[RecalculateGroup] Group mode is not 'traffic' (current: %s), skipping", mode)
|
|
return nil
|
|
}
|
|
|
|
// 4. Execute traffic-based grouping
|
|
logger.Infof("[RecalculateGroup] Executing traffic-based grouping")
|
|
|
|
logic := group.NewRecalculateGroupLogic(ctx, l.svc)
|
|
req := &types.RecalculateGroupRequest{
|
|
Mode: "traffic",
|
|
TriggerType: "scheduled",
|
|
}
|
|
|
|
if err := logic.RecalculateGroup(req); err != nil {
|
|
logger.Errorw("[RecalculateGroup] Failed to execute traffic grouping", logger.Field("error", err.Error()))
|
|
return err
|
|
}
|
|
|
|
logger.Infof("[RecalculateGroup] Successfully completed traffic-based grouping: %s", time.Now().Format("2006-01-02 15:04:05"))
|
|
return nil
|
|
}
|