1. 提取触发器逻辑

2. 创建/更新计划时自动生成对应触发器

internal/infra/repository/plan_repository.go

@@ -1,10 +1,12 @@
 package repository
 
 import (
+	"encoding/json"
 	"errors"
 	"fmt"
 
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+	"gorm.io/datatypes"
 	"gorm.io/gorm"
 )
 
@@ -170,6 +172,11 @@ func (r *gormPlanRepository) CreatePlan(plan *models.Plan) error {
 		if err := tx.Create(plan).Error; err != nil {
 			return err
 		}
+
+		// 3. 创建触发器Task
+		if err := r.createPlanAnalysisTask(tx, plan); err != nil {
+			return err
+		}
 		return nil
 	})
 }
@@ -186,7 +193,12 @@ func (r *gormPlanRepository) updatePlanTx(tx *gorm.DB, plan *models.Plan) error
 	if err := r.validatePlanTree(tx, plan); err != nil {
 		return err
 	}
-	return r.reconcilePlanNode(tx, plan)
+	if err := r.reconcilePlanNode(tx, plan); err != nil {
+		return err
+	}
+
+	// 更新Plan触发器
+	return r.updatePlanAnalysisTask(tx, plan)
 }
 
 // validatePlanTree 对整个计划树进行全面的只读健康检查
@@ -461,51 +473,61 @@ func (r *gormPlanRepository) flattenPlanTasksRecursive(plan *models.Plan) ([]mod
 func (r *gormPlanRepository) DeleteTask(id int) error {
 	// 使用事务确保操作的原子性
 	return r.db.Transaction(func(tx *gorm.DB) error {
-		// 1. 检查是否有待执行任务引用了这个任务
-		var pendingTaskCount int64
-		if err := tx.Model(&models.PendingTask{}).Where("task_id = ?", id).Count(&pendingTaskCount).Error; err != nil {
-			return fmt.Errorf("检查待执行任务时出错: %w", err)
-		}
-
-		// 如果有待执行任务引用该任务，不能删除
-		if pendingTaskCount > 0 {
-			return fmt.Errorf("无法删除任务(ID: %d)，因为存在 %d 条待执行任务引用该任务", id, pendingTaskCount)
-		}
-
-		// 2. 检查是否有计划仍在使用这个任务
-		var planCount int64
-		if err := tx.Model(&models.Plan{}).Joins("JOIN tasks ON plans.id = tasks.plan_id").Where("tasks.id = ?", id).Count(&planCount).Error; err != nil {
-			return fmt.Errorf("检查计划引用任务时出错: %w", err)
-		}
-
-		// 如果有计划在使用该任务，不能删除
-		if planCount > 0 {
-			return fmt.Errorf("无法删除任务(ID: %d)，因为存在 %d 个计划仍在使用该任务", id, planCount)
-		}
-
-		// 3. 执行删除操作
-		result := tx.Delete(&models.Task{}, id)
-		if result.Error != nil {
-			return fmt.Errorf("删除任务失败: %w", result.Error)
-		}
-
-		// 检查是否实际删除了记录
-		if result.RowsAffected == 0 {
-			return gorm.ErrRecordNotFound
-		}
-
-		return nil
+		return r.deleteTask(tx, id)
 	})
 }
 
+// deleteTask 根据ID删除任务
+func (r *gormPlanRepository) deleteTask(tx *gorm.DB, id int) error {
+	// 1. 检查是否有待执行任务引用了这个任务
+	var pendingTaskCount int64
+	if err := tx.Model(&models.PendingTask{}).Where("task_id = ?", id).Count(&pendingTaskCount).Error; err != nil {
+		return fmt.Errorf("检查待执行任务时出错: %w", err)
+	}
+
+	// 如果有待执行任务引用该任务，不能删除
+	if pendingTaskCount > 0 {
+		return fmt.Errorf("无法删除任务(ID: %d)，因为存在 %d 条待执行任务引用该任务", id, pendingTaskCount)
+	}
+
+	// 2. 检查是否有计划仍在使用这个任务
+	var planCount int64
+	if err := tx.Model(&models.Plan{}).Joins("JOIN tasks ON plans.id = tasks.plan_id").Where("tasks.id = ?", id).Count(&planCount).Error; err != nil {
+		return fmt.Errorf("检查计划引用任务时出错: %w", err)
+	}
+
+	// 如果有计划在使用该任务，不能删除
+	if planCount > 0 {
+		return fmt.Errorf("无法删除任务(ID: %d)，因为存在 %d 个计划仍在使用该任务", id, planCount)
+	}
+
+	// 3. 执行删除操作
+	result := tx.Delete(&models.Task{}, id)
+	if result.Error != nil {
+		return fmt.Errorf("删除任务失败: %w", result.Error)
+	}
+
+	// 检查是否实际删除了记录
+	if result.RowsAffected == 0 {
+		return gorm.ErrRecordNotFound
+	}
+
+	return nil
+}
+
 // FindPlanAnalysisTaskByParamsPlanID 根据Parameters中的ParamsPlanID字段值查找TaskPlanAnalysis类型的Task
 func (r *gormPlanRepository) FindPlanAnalysisTaskByParamsPlanID(paramsPlanID uint) (*models.Task, error) {
+	return r.findPlanAnalysisTaskByParamsPlanID(r.db, paramsPlanID)
+}
+
+// findPlanAnalysisTaskByParamsPlanID 使用指定db根据Parameters中的ParamsPlanID字段值查找TaskPlanAnalysis类型的Task
+func (r *gormPlanRepository) findPlanAnalysisTaskByParamsPlanID(tx *gorm.DB, paramsPlanID uint) (*models.Task, error) {
 	var task models.Task
 
 	// 构造JSON查询条件，查找Parameters中包含指定ParamsPlanID且Type为TaskPlanAnalysis的任务
 	// TODO 在JSON字段中查找特定键值的语法取决于数据库类型，这里使用PostgreSQL的语法
 	// TODO 如果使用的是MySQL，则需要相应调整查询条件
-	result := r.db.Where(
+	result := tx.Where(
 		"type = ? AND parameters->>'plan_id' = ?",
 		models.TaskPlanAnalysis,
 		fmt.Sprintf("%d", paramsPlanID),
@@ -520,3 +542,38 @@ func (r *gormPlanRepository) FindPlanAnalysisTaskByParamsPlanID(paramsPlanID uin
 
 	return &task, nil
 }
+
+// createPlanAnalysisTask 用于创建一个TaskPlanAnalysis类型的Task
+func (r *gormPlanRepository) createPlanAnalysisTask(tx *gorm.DB, plan *models.Plan) error {
+	m := map[string]interface{}{
+		models.ParamsPlanID: plan.ID,
+	}
+	parameters, err := json.Marshal(m)
+	if err != nil {
+		return err
+	}
+
+	task := &models.Task{
+		PlanID:         plan.ID,
+		Name:           fmt.Sprintf("'%v'计划触发器", plan.Name),
+		Description:    fmt.Sprintf("计划名: %v, 计划ID: %v", plan.Name, plan.ID),
+		ExecutionOrder: 0,
+		Type:           models.TaskPlanAnalysis,
+		Parameters:     datatypes.JSON(parameters),
+	}
+
+	return tx.Create(task).Error
+}
+
+// updatePlanAnalysisTask 使用简单粗暴的删除再创建方式实现更新, 以控制AnalysisPlanTask的定义全部在createPlanAnalysisTask方法中
+func (r *gormPlanRepository) updatePlanAnalysisTask(tx *gorm.DB, plan *models.Plan) error {
+	task, err := r.findPlanAnalysisTaskByParamsPlanID(tx, plan.ID)
+	if err != nil {
+		return err
+	}
+	err = r.deleteTask(tx, task.ID)
+	if err != nil {
+		return err
+	}
+	return r.createPlanAnalysisTask(tx, plan)
+}