重构AnalysisPlanTaskManager

internal/infra/repository/plan_repository.go

@@ -42,6 +42,12 @@ type PlanRepository interface {
 	DeleteTask(id int) error
 	// FindPlanAnalysisTaskByParamsPlanID 根据Parameters中的ParamsPlanID字段值查找TaskPlanAnalysis类型的Task
 	FindPlanAnalysisTaskByParamsPlanID(paramsPlanID uint) (*models.Task, error)
+	// FindRunnablePlans 获取所有应执行的计划
+	FindRunnablePlans() ([]*models.Plan, error)
+	// FindDisabledAndStoppedPlans 获取所有已禁用或已停止的计划
+	FindDisabledAndStoppedPlans() ([]*models.Plan, error)
+	// FindPlanAnalysisTaskByPlanID 根据 PlanID 找到其关联的 'plan_analysis' 任务
+	FindPlanAnalysisTaskByPlanID(planID uint) (*models.Task, error)
 }
 
 // gormPlanRepository 是 PlanRepository 的 GORM 实现
@@ -565,15 +571,56 @@ func (r *gormPlanRepository) createPlanAnalysisTask(tx *gorm.DB, plan *models.Pl
 	return tx.Create(task).Error
 }
 
-// updatePlanAnalysisTask 使用简单粗暴的删除再创建方式实现更新, 以控制AnalysisPlanTask的定义全部在createPlanAnalysisTask方法中
+// updatePlanAnalysisTask 使用更安全的方式更新触发器任务
 func (r *gormPlanRepository) updatePlanAnalysisTask(tx *gorm.DB, plan *models.Plan) error {
-	task, err := r.findPlanAnalysisTaskByParamsPlanID(tx, plan.ID)
-	if err != nil {
-		return err
+	task, err := r.findPlanAnalysisTask(tx, plan.ID)
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return fmt.Errorf("查找现有计划分析任务失败: %w", err)
 	}
-	err = r.deleteTask(tx, task.ID)
-	if err != nil {
-		return err
+
+	// 如果触发器任务不存在，则创建一个
+	if task == nil {
+		return r.createPlanAnalysisTask(tx, plan)
 	}
-	return r.createPlanAnalysisTask(tx, plan)
+
+	// 如果存在，则更新它的名称和描述以反映计划的最新信息
+	task.Name = fmt.Sprintf("'%s'计划触发器", plan.Name)
+	task.Description = fmt.Sprintf("计划名: %s, 计划ID: %d", plan.Name, plan.ID)
+
+	return tx.Save(task).Error
+}
+
+func (r *gormPlanRepository) FindRunnablePlans() ([]*models.Plan, error) {
+	var plans []*models.Plan
+	err := r.db.
+		Where("status = ?", models.PlanStatusEnabled).
+		Where(
+			r.db.Where("execution_type = ?", models.PlanExecutionTypeManual).
+				Or("execution_type = ? AND (execute_num = 0 OR execute_count < execute_num)", models.PlanExecutionTypeAutomatic),
+		).
+		Find(&plans).Error
+	return plans, err
+}
+
+func (r *gormPlanRepository) FindDisabledAndStoppedPlans() ([]*models.Plan, error) {
+	var plans []*models.Plan
+	err := r.db.
+		Where("status = ? OR status = ?", models.PlanStatusDisabled, models.PlanStatusStopeed).
+		Find(&plans).Error
+	return plans, err
+}
+
+// findPlanAnalysisTask 是一个内部使用的、更高效的查找方法
+func (r *gormPlanRepository) findPlanAnalysisTask(tx *gorm.DB, planID uint) (*models.Task, error) {
+	var task models.Task
+	err := tx.Where("plan_id = ? AND type = ?", planID, models.TaskPlanAnalysis).First(&task).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, nil // 未找到不是错误，返回nil, nil
+	}
+	return &task, err
+}
+
+// FindPlanAnalysisTaskByPlanID 是暴露给外部的公共方法
+func (r *gormPlanRepository) FindPlanAnalysisTaskByPlanID(planID uint) (*models.Task, error) {
+	return r.findPlanAnalysisTask(r.db, planID)
 }