实现根据区域ID或设备ID清空对应阈值告警任务

internal/app/service/threshold_alarm_service.go

@@ -34,6 +34,8 @@ type ThresholdAlarmService interface {
 	GetDeviceThresholdAlarm(ctx context.Context, taskID int) (*dto.DeviceThresholdAlarmDTO, error)
 	// DeleteDeviceThresholdAlarm 删除一个设备阈值告警。
 	DeleteDeviceThresholdAlarm(ctx context.Context, taskID int, req *dto.DeleteDeviceThresholdAlarmDTO) error
+	// DeleteDeviceThresholdAlarmByDeviceID 实现了根据设备ID删除一个设备下所有设备阈值告警的逻辑。
+	DeleteDeviceThresholdAlarmByDeviceID(ctx context.Context, deviceID uint) error
 
 	// CreateAreaThresholdAlarm 创建一个区域阈值告警。
 	CreateAreaThresholdAlarm(ctx context.Context, req *dto.CreateAreaThresholdAlarmDTO) error
@@ -43,6 +45,8 @@ type ThresholdAlarmService interface {
 	GetAreaThresholdAlarm(ctx context.Context, taskID int) (*dto.AreaThresholdAlarmDTO, error)
 	// DeleteAreaThresholdAlarm 实现了删除一个区域阈值告警的逻辑。
 	DeleteAreaThresholdAlarm(ctx context.Context, taskID int) error
+	// DeleteAreaThresholdAlarmByAreaControllerID 实现了根据区域ID删除一个区域下所有区域阈值告警的逻辑。
+	DeleteAreaThresholdAlarmByAreaControllerID(ctx context.Context, areaControllerID uint) error
 }
 
 // thresholdAlarmService 是 ThresholdAlarmService 接口的具体实现。
@@ -370,6 +374,76 @@ func (s *thresholdAlarmService) DeleteDeviceThresholdAlarm(ctx context.Context,
 	return err
 }
 
+// DeleteDeviceThresholdAlarmByDeviceID 实现了根据设备ID删除一个设备下所有设备阈值告警的逻辑。
+func (s *thresholdAlarmService) DeleteDeviceThresholdAlarmByDeviceID(ctx context.Context, deviceID uint) error {
+	serviceCtx, logger := logs.Trace(ctx, s.ctx, "DeleteDeviceThresholdAlarmByDeviceID")
+	tasks, err := s.planRepo.ListTasksByDeviceID(serviceCtx, deviceID)
+	if err != nil {
+		return fmt.Errorf("获取任务列表失败: %w", err)
+	}
+	device, err := s.deviceRepo.FindByID(serviceCtx, deviceID)
+	if err != nil {
+		return fmt.Errorf("获取设备 %d 失败: %w", deviceID, err)
+	}
+	plan, err := s.planRepo.GetSystemPlanByName(serviceCtx, models.PlanNamePeriodicSystemHealthCheck)
+	if err != nil {
+		return fmt.Errorf("获取系统计划失败: %w", err)
+	}
+	if plan == nil {
+		logger.Panicf("系统计划 %v 不存在", models.PlanNamePeriodicSystemHealthCheck)
+	}
+
+	deleteNums := []int{}
+	for i, t := range plan.Tasks {
+		for _, dt := range tasks {
+			if t.ID == dt.ID && t.Type == models.TaskTypeDeviceThresholdCheck {
+				deleteNums = append(deleteNums, i)
+				break
+			}
+		}
+
+		if t.Type == models.TaskTypeAreaCollectorThresholdCheck {
+			var areaParams task.AreaThresholdCheckParams
+			if err := t.ParseParameters(&areaParams); err != nil {
+				return fmt.Errorf("任务 %d: 解析参数失败: %w", t.ID, err)
+			}
+			if areaParams.AreaControllerID == device.AreaControllerID {
+				for ai, ae := range areaParams.ExcludeDeviceIDs {
+					if ae == deviceID {
+						areaParams.ExcludeDeviceIDs = append(areaParams.ExcludeDeviceIDs[:ai], areaParams.ExcludeDeviceIDs[ai+1:]...)
+						break
+					}
+				}
+				err = plan.Tasks[i].SaveParameters(areaParams)
+				if err != nil {
+					return fmt.Errorf("任务 %d: 保存参数失败: %w", t.ID, err)
+				}
+			}
+		}
+	}
+
+	// 为了高效地判断一个索引是否需要被删除，先将 deleteNums 转换为一个 map。
+	deleteMap := make(map[int]struct{}, len(deleteNums))
+	for _, index := range deleteNums {
+		deleteMap[index] = struct{}{}
+	}
+
+	// 创建一个新的任务切片，只包含不需要删除的任务。
+	newTasks := make([]models.Task, 0, len(plan.Tasks)-len(deleteNums))
+	for i, t := range plan.Tasks {
+		if _, found := deleteMap[i]; !found {
+			newTasks = append(newTasks, t)
+		}
+	}
+	plan.Tasks = newTasks
+
+	// 重新排序任务并更新计划
+	plan.ReorderSteps()
+	_, err = s.planService.UpdatePlan(serviceCtx, plan, models.PlanTypeSystem)
+	return err
+
+}
+
 // CreateAreaThresholdAlarm 实现了创建一个区域阈值告警的逻辑。
 func (s *thresholdAlarmService) CreateAreaThresholdAlarm(ctx context.Context, req *dto.CreateAreaThresholdAlarmDTO) error {
 	serviceCtx, logger := logs.Trace(ctx, s.ctx, "CreateAreaThresholdAlarm")
@@ -555,3 +629,60 @@ func (s *thresholdAlarmService) DeleteAreaThresholdAlarm(ctx context.Context, ta
 	_, err = s.planService.UpdatePlan(serviceCtx, plan, models.PlanTypeSystem)
 	return err
 }
+
+// DeleteAreaThresholdAlarmByAreaControllerID 实现了根据区域ID删除一个区域下所有区域阈值告警的逻辑。
+func (s *thresholdAlarmService) DeleteAreaThresholdAlarmByAreaControllerID(ctx context.Context, areaControllerID uint) error {
+	serviceCtx, logger := logs.Trace(ctx, s.ctx, "DeleteAreaThresholdAlarmByAreaControllerID")
+
+	// 1. 获取系统健康检查计划
+	plan, err := s.planRepo.GetSystemPlanByName(serviceCtx, models.PlanNamePeriodicSystemHealthCheck)
+	if err != nil {
+		return fmt.Errorf("获取系统健康检查计划失败: %w", err)
+	}
+	if plan == nil {
+		logger.Panicf("系统计划 %v 不存在", models.PlanNamePeriodicSystemHealthCheck)
+	}
+
+	// 2. 收集所有与指定 areaControllerID 相关的区域阈值告警任务的索引
+	var deleteIndices []int
+	for i, t := range plan.Tasks {
+		// 只关心区域阈值检查任务
+		if t.Type != models.TaskTypeAreaCollectorThresholdCheck {
+			continue
+		}
+
+		var params task.AreaThresholdCheckParams
+		if err := t.ParseParameters(&params); err != nil {
+			return fmt.Errorf("任务 %d: 解析参数失败: %w", t.ID, err)
+		}
+
+		// 如果 AreaControllerID 匹配，则记录其索引以待删除
+		if params.AreaControllerID == areaControllerID {
+			deleteIndices = append(deleteIndices, i)
+		}
+	}
+
+	// 如果没有找到要删除的任务，则直接返回
+	if len(deleteIndices) == 0 {
+		return nil
+	}
+
+	// 3. 使用 map 和新切片的方式安全地删除多个任务
+	deleteMap := make(map[int]struct{}, len(deleteIndices))
+	for _, index := range deleteIndices {
+		deleteMap[index] = struct{}{}
+	}
+
+	newTasks := make([]models.Task, 0, len(plan.Tasks)-len(deleteMap))
+	for i, t := range plan.Tasks {
+		if _, found := deleteMap[i]; !found {
+			newTasks = append(newTasks, t)
+		}
+	}
+	plan.Tasks = newTasks
+
+	// 4. 重新排序任务并更新计划
+	plan.ReorderSteps()
+	_, err = s.planService.UpdatePlan(serviceCtx, plan, models.PlanTypeSystem)
+	return err
+}