91 lines
3.1 KiB
Go
91 lines
3.1 KiB
Go
package group
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/perfect-panel/server/internal/model/group"
|
|
"github.com/perfect-panel/server/internal/model/node"
|
|
"github.com/perfect-panel/server/internal/model/subscribe"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type DeleteNodeGroupLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewDeleteNodeGroupLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteNodeGroupLogic {
|
|
return &DeleteNodeGroupLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *DeleteNodeGroupLogic) DeleteNodeGroup(req *types.DeleteNodeGroupRequest) error {
|
|
// 查询节点组信息
|
|
var nodeGroup group.NodeGroup
|
|
if err := l.svcCtx.DB.Where("id = ?", req.Id).First(&nodeGroup).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return errors.New("node group not found")
|
|
}
|
|
logger.Errorf("failed to find node group: %v", err)
|
|
return err
|
|
}
|
|
|
|
// 检查是否有关联节点（使用JSON_CONTAINS查询node_group_ids数组）
|
|
var nodeCount int64
|
|
if err := l.svcCtx.DB.Model(&node.Node{}).Where("JSON_CONTAINS(node_group_ids, ?)", fmt.Sprintf("[%d]", nodeGroup.Id)).Count(&nodeCount).Error; err != nil {
|
|
logger.Errorf("failed to count nodes in group: %v", err)
|
|
return err
|
|
}
|
|
if nodeCount > 0 {
|
|
return fmt.Errorf("cannot delete group with %d associated nodes, please migrate nodes first", nodeCount)
|
|
}
|
|
|
|
var defaultSubscribeCount int64
|
|
if err := l.svcCtx.DB.Model(&subscribe.Subscribe{}).Where("node_group_id = ?", nodeGroup.Id).Count(&defaultSubscribeCount).Error; err != nil {
|
|
logger.Errorf("failed to count subscribes with default group: %v", err)
|
|
return err
|
|
}
|
|
if defaultSubscribeCount > 0 {
|
|
return fmt.Errorf("cannot delete group referenced by %d subscribes' default node group", defaultSubscribeCount)
|
|
}
|
|
|
|
var subscribeGroupCount int64
|
|
if err := l.svcCtx.DB.Model(&subscribe.Subscribe{}).Where("JSON_CONTAINS(node_group_ids, ?)", fmt.Sprintf("[%d]", nodeGroup.Id)).Count(&subscribeGroupCount).Error; err != nil {
|
|
logger.Errorf("failed to count subscribes in group: %v", err)
|
|
return err
|
|
}
|
|
if subscribeGroupCount > 0 {
|
|
return fmt.Errorf("cannot delete group referenced by %d subscribes' node group list", subscribeGroupCount)
|
|
}
|
|
|
|
var userSubscribeCount int64
|
|
if err := l.svcCtx.DB.Table("user_subscribe").Where("node_group_id = ?", nodeGroup.Id).Count(&userSubscribeCount).Error; err != nil {
|
|
logger.Errorf("failed to count user subscribes in group: %v", err)
|
|
return err
|
|
}
|
|
if userSubscribeCount > 0 {
|
|
return fmt.Errorf("cannot delete group referenced by %d user subscribes", userSubscribeCount)
|
|
}
|
|
|
|
// 使用 GORM Transaction 删除节点组
|
|
return l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
|
|
// 删除节点组
|
|
if err := tx.Where("id = ?", req.Id).Delete(&group.NodeGroup{}).Error; err != nil {
|
|
logger.Errorf("failed to delete node group: %v", err)
|
|
return err // 自动回滚
|
|
}
|
|
|
|
logger.Infof("deleted node group: id=%d", nodeGroup.Id)
|
|
return nil // 自动提交
|
|
})
|
|
}
|