bmad 分析师工作

internal/app/service/device_service.go

@@ -38,7 +38,8 @@ type deviceService struct {
 	deviceRepo         repository.DeviceRepository
 	areaControllerRepo repository.AreaControllerRepository
 	deviceTemplateRepo repository.DeviceTemplateRepository
-	deviceDomainSvc    device.Service // 依赖领域服务
+	planRepo           repository.PlanRepository // 新增计划仓库依赖
+	deviceDomainSvc    device.Service            // 依赖领域服务
 }
 
 // NewDeviceService 创建一个新的 DeviceService 实例。
@@ -46,16 +47,48 @@ func NewDeviceService(
 	deviceRepo repository.DeviceRepository,
 	areaControllerRepo repository.AreaControllerRepository,
 	deviceTemplateRepo repository.DeviceTemplateRepository,
+	planRepo repository.PlanRepository,
 	deviceDomainSvc device.Service,
 ) DeviceService {
 	return &deviceService{
 		deviceRepo:         deviceRepo,
 		areaControllerRepo: areaControllerRepo,
 		deviceTemplateRepo: deviceTemplateRepo,
+		planRepo:           planRepo,
 		deviceDomainSvc:    deviceDomainSvc,
 	}
 }
 
+func (s *deviceService) isDeviceInUseByActivePlan(deviceID uint) (bool, error) {
+	plans, err := s.planRepo.FindPlansWithPendingTasks()
+	if err != nil {
+		return false, fmt.Errorf("查询活动计划失败: %w", err)
+	}
+
+	for _, plan := range plans {
+		tasks, err := s.planRepo.FlattenPlanTasks(plan.ID)
+		if err != nil {
+			return false, fmt.Errorf("展开计划 %d 的任务失败: %w", plan.ID, err)
+		}
+
+		for _, task := range tasks {
+			// 假设任务参数中设备ID的键是 "device_id"
+			var params map[string]interface{}
+			if err := json.Unmarshal(task.Parameters, &params); err != nil {
+				// 无法解析参数，跳过此任务
+				continue
+			}
+
+			if paramDeviceID, ok := params["device_id"]; ok {
+				if floatDeviceID, ok := paramDeviceID.(float64); ok && uint(floatDeviceID) == deviceID {
+					return true, nil
+				}
+			}
+		}
+	}
+	return false, nil
+}
+
 // --- Devices ---
 
 func (s *deviceService) CreateDevice(req *dto.CreateDeviceRequest) (*dto.DeviceResponse, error) {
@@ -149,6 +182,13 @@ func (s *deviceService) DeleteDevice(id string) error {
 		return err
 	}
 
+	// 检查设备是否被活动计划使用
+	if inUse, err := s.isDeviceInUseByActivePlan(uint(idUint)); err != nil {
+		return fmt.Errorf("检查设备是否被计划使用失败: %w", err)
+	} else if inUse {
+		return errors.New("设备正在被活动计划使用，无法删除")
+	}
+
 	return s.deviceRepo.Delete(uint(idUint))
 }
 
@@ -263,6 +303,15 @@ func (s *deviceService) DeleteAreaController(id string) error {
 		return err
 	}
 
+	// 检查是否有设备正在使用此区域主控
+	devices, err := s.deviceRepo.ListByAreaControllerID(uint(idUint))
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return err
+	}
+	if len(devices) > 0 {
+		return errors.New("区域主控正在被设备使用，无法删除")
+	}
+
 	return s.areaControllerRepo.Delete(uint(idUint))
 }
 
@@ -369,5 +418,14 @@ func (s *deviceService) DeleteDeviceTemplate(id string) error {
 		return err
 	}
 
+	// 检查是否有设备正在使用此模板
+	devices, err := s.deviceRepo.FindByDeviceTemplateID(uint(idUint))
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return err
+	}
+	if len(devices) > 0 {
+		return errors.New("设备模板正在被设备使用，无法删除")
+	}
+
 	return s.deviceTemplateRepo.Delete(uint(idUint))
 }

internal/infra/repository/device_repository.go

@@ -1,6 +1,7 @@
 package repository
 
 import (
+	"errors"
 	"fmt"
 	"strconv"
 
@@ -44,12 +45,22 @@ type DeviceRepository interface {
 
 // gormDeviceRepository 是 DeviceRepository 的 GORM 实现
 type gormDeviceRepository struct {
-	db *gorm.DB
+	db                    *gorm.DB
+	deviceCommandLogRepo  DeviceCommandLogRepository
+	pendingCollectionRepo PendingCollectionRepository
 }
 
 // NewGormDeviceRepository 创建一个新的 DeviceRepository GORM 实现实例
-func NewGormDeviceRepository(db *gorm.DB) DeviceRepository {
-	return &gormDeviceRepository{db: db}
+func NewGormDeviceRepository(
+	db *gorm.DB,
+	deviceCommandLogRepo DeviceCommandLogRepository,
+	pendingCollectionRepo PendingCollectionRepository,
+) DeviceRepository {
+	return &gormDeviceRepository{
+		db:                    db,
+		deviceCommandLogRepo:  deviceCommandLogRepo,
+		pendingCollectionRepo: pendingCollectionRepo,
+	}
 }
 
 // Create 创建一个新的设备记录
@@ -129,6 +140,24 @@ func (r *gormDeviceRepository) Update(device *models.Device) error {
 // Delete 根据 ID 删除一个设备
 // GORM 使用软删除，记录不会从数据库中物理移除，而是设置 DeletedAt 字段。
 func (r *gormDeviceRepository) Delete(id uint) error {
+	// 检查是否有相关的设备命令日志
+	logs, err := r.deviceCommandLogRepo.FindByDeviceID(id)
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return fmt.Errorf("查询设备 %d 的命令日志失败: %w", id, err)
+	}
+	if len(logs) > 0 {
+		return errors.New("设备有相关的命令日志，不能删除")
+	}
+
+	// 检查是否有相关的待处理采集请求
+	pendingCollections, err := r.pendingCollectionRepo.FindByDeviceID(id)
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return fmt.Errorf("查询设备 %d 的待处理采集请求失败: %w", id, err)
+	}
+	if len(pendingCollections) > 0 {
+		return errors.New("设备有相关的待处理采集请求，不能删除")
+	}
+
 	return r.db.Delete(&models.Device{}, id).Error
 }