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
56 lines
1.7 KiB
Go
56 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"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"
|
|
"github.com/perfect-panel/server/pkg/tool"
|
|
"github.com/perfect-panel/server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type UpdateNodeLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// NewUpdateNodeLogic Update Node
|
|
func NewUpdateNodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateNodeLogic {
|
|
return &UpdateNodeLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *UpdateNodeLogic) UpdateNode(req *types.UpdateNodeRequest) error {
|
|
data, err := l.svcCtx.NodeModel.FindOneNode(l.ctx, req.Id)
|
|
if err != nil {
|
|
l.Errorw("[UpdateNode] Query Database Error: ", logger.Field("error", err.Error()))
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "[UpdateNode] Query Database Error")
|
|
}
|
|
data.Name = req.Name
|
|
data.Tags = tool.StringSliceToString(req.Tags)
|
|
data.ServerId = req.ServerId
|
|
data.Port = req.Port
|
|
data.Address = req.Address
|
|
data.Protocol = req.Protocol
|
|
data.Enabled = req.Enabled
|
|
data.NodeGroupIds = node.JSONInt64Slice(req.NodeGroupIds)
|
|
err = l.svcCtx.NodeModel.UpdateNode(l.ctx, data)
|
|
if err != nil {
|
|
l.Errorw("[UpdateNode] Update Database Error: ", logger.Field("error", err.Error()))
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "[UpdateNode] Update Database Error")
|
|
}
|
|
return l.svcCtx.NodeModel.ClearNodeCache(l.ctx, &node.FilterNodeParams{
|
|
Page: 1,
|
|
Size: 1000,
|
|
ServerId: []int64{data.ServerId},
|
|
Search: "",
|
|
})
|
|
}
|