重构AnalysisPlanTaskManager

internal/app/service/task/analysis_plan_task_manager.go

@@ -57,9 +57,9 @@ func (m *AnalysisPlanTaskManager) Refresh() error {
 		m.logger.Errorf("清理无效任务时出错: %v", err)
 	}
 
-	// 3. 为应执行但缺失的计划添加新触发器
-	if err := m.addNewTriggers(runnablePlans, pendingTasks); err != nil {
-		return fmt.Errorf("添加新触发器时出错: %w", err)
+	// 3. 添加或更新触发器
+	if err := m.addOrUpdateTriggers(runnablePlans, pendingTasks); err != nil {
+		return fmt.Errorf("添加或更新触发器时出错: %w", err)
 	}
 
 	m.logger.Info("计划任务管理器同步完成.")
@@ -167,18 +167,38 @@ func (m *AnalysisPlanTaskManager) cleanupInvalidTasks(invalidPlanIDs []uint, all
 	return nil
 }
 
-// addNewTriggers 检查并为应执行但缺失的计划添加新触发器。
-func (m *AnalysisPlanTaskManager) addNewTriggers(runnablePlans []*models.Plan, allPendingTasks []models.PendingTask) error {
-	// 创建一个集合，存放所有已在队列中的计划触发器
-	pendingTriggerPlanIDs := make(map[uint]struct{})
+// addOrUpdateTriggers 检查、更新或创建触发器。
+func (m *AnalysisPlanTaskManager) addOrUpdateTriggers(runnablePlans []*models.Plan, allPendingTasks []models.PendingTask) error {
+	// 创建一个映射，存放所有已在队列中的计划触发器
+	pendingTriggersMap := make(map[uint]models.PendingTask)
 	for _, pt := range allPendingTasks {
 		if pt.Task != nil && pt.Task.Type == models.TaskPlanAnalysis {
-			pendingTriggerPlanIDs[pt.Task.PlanID] = struct{}{}
+			pendingTriggersMap[pt.Task.PlanID] = pt
 		}
 	}
 
 	for _, plan := range runnablePlans {
-		if _, exists := pendingTriggerPlanIDs[plan.ID]; !exists {
+		existingTrigger, exists := pendingTriggersMap[plan.ID]
+
+		if exists {
+			// --- 新增逻辑：检查并更新现有触发器 ---
+			// 只对自动计划检查时间更新
+			if plan.ExecutionType == models.PlanExecutionTypeAutomatic {
+				next, err := utils.GetNextCronTime(plan.CronExpression)
+				if err != nil {
+					m.logger.Errorf("为计划 #%d 解析Cron表达式失败，跳过更新: %v", plan.ID, err)
+					continue
+				}
+				// 如果数据库中记录的执行时间与根据当前Cron表达式计算出的下一次时间不一致，则更新
+				if !existingTrigger.ExecuteAt.Equal(next) {
+					m.logger.Infof("计划 #%d 的执行时间已变更，正在更新触发器 #%d 的执行时间从 %v 到 %v...", plan.ID, existingTrigger.ID, existingTrigger.ExecuteAt, next)
+					if err := m.pendingTaskRepo.UpdatePendingTaskExecuteAt(existingTrigger.ID, next); err != nil {
+						m.logger.Errorf("更新触发器 #%d 的执行时间失败: %v", existingTrigger.ID, err)
+					}
+				}
+			}
+		} else {
+			// --- 原有逻辑：为缺失的计划创建新触发器 ---
 			m.logger.Infof("发现应执行但队列中缺失的计划 #%d，正在为其创建触发器...", plan.ID)
 			if err := m.createTriggerTask(plan); err != nil {
 				m.logger.Errorf("为计划 #%d 创建触发器失败: %v", plan.ID, err)
@@ -195,8 +215,16 @@ func (m *AnalysisPlanTaskManager) createTriggerTask(plan *models.Plan) error {
 	if err != nil {
 		return fmt.Errorf("查找计划分析任务失败: %w", err)
 	}
+
+	// --- 如果触发器任务定义不存在，则自动创建 ---
 	if analysisTask == nil {
-		return fmt.Errorf("未找到计划 #%d 关联的 'plan_analysis' 任务", plan.ID)
+		m.logger.Warnf("未找到计划 #%d 关联的 'plan_analysis' 任务定义，将自动创建...", plan.ID)
+		newAnalysisTask, err := m.planRepo.CreatePlanAnalysisTask(plan)
+		if err != nil {
+			return fmt.Errorf("自动创建 'plan_analysis' 任务定义失败: %w", err)
+		}
+		analysisTask = newAnalysisTask
+		m.logger.Infof("已成功为计划 #%d 创建 'plan_analysis' 任务定义 (ID: %d)", plan.ID, analysisTask.ID)
 	}
 
 	var executeAt time.Time