diff --git a/internal/core/data_initializer.go b/internal/core/data_initializer.go index cbca13a..ab157ef 100644 --- a/internal/core/data_initializer.go +++ b/internal/core/data_initializer.go @@ -33,9 +33,9 @@ func (app *Application) initializeState() error { return nil } -// initializeSystemPlans 确保预定义的系统计划在数据库中存在。 +// initializeSystemPlans 确保预定义的系统计划在数据库中存在并保持最新。 func (app *Application) initializeSystemPlans() error { - app.Logger.Info("开始检查并创建预定义的系统计划...") + app.Logger.Info("开始检查并更新预定义的系统计划...") // 动态构建预定义计划列表 predefinedSystemPlans := app.getPredefinedSystemPlans() @@ -43,24 +43,39 @@ func (app *Application) initializeSystemPlans() error { // 1. 获取所有已存在的系统计划 existingPlans, _, err := app.Infra.Repos.PlanRepo.ListPlans(repository.ListPlansOptions{ PlanType: repository.PlanTypeFilterSystem, - }, 1, 999) // 使用一个较大的 pageSize 来获取所有系统计划 + }, 1, 99999) // 使用一个较大的 pageSize 来获取所有系统计划 if err != nil { return fmt.Errorf("获取现有系统计划失败: %w", err) } // 2. 为了方便查找, 将现有计划名放入一个 map - existingPlanNames := make(map[string]bool) - for _, p := range existingPlans { - existingPlanNames[p.Name] = true + existingPlanMap := make(map[string]*models.Plan) + for i := range existingPlans { + existingPlanMap[existingPlans[i].Name] = &existingPlans[i] } // 3. 遍历预定义的计划列表 - for _, predefinedPlan := range predefinedSystemPlans { - // 4. 如果计划不存在, 则创建 - if !existingPlanNames[predefinedPlan.Name] { + for i := range predefinedSystemPlans { + predefinedPlan := &predefinedSystemPlans[i] // 获取可修改的指针 + + if foundExistingPlan, ok := existingPlanMap[predefinedPlan.Name]; ok { + // 如果计划存在,则进行无差别更新 + app.Logger.Infof("预定义计划 '%s' 已存在,正在进行无差别更新...", predefinedPlan.Name) + + // 将数据库中已存在的计划的ID和运行时状态字段赋值给预定义计划 + predefinedPlan.ID = foundExistingPlan.ID + predefinedPlan.ExecuteCount = foundExistingPlan.ExecuteCount + predefinedPlan.Status = foundExistingPlan.Status + + if err := app.Infra.Repos.PlanRepo.UpdatePlan(predefinedPlan); err != nil { + return fmt.Errorf("更新预定义计划 '%s' 失败: %w", predefinedPlan.Name, err) + } else { + app.Logger.Infof("成功更新预定义计划 '%s'。", predefinedPlan.Name) + } + } else { + // 如果计划不存在, 则创建 app.Logger.Infof("预定义计划 '%s' 不存在,正在创建...", predefinedPlan.Name) - if err := app.Infra.Repos.PlanRepo.CreatePlan(&predefinedPlan); err != nil { - // 错误现在是致命的 + if err := app.Infra.Repos.PlanRepo.CreatePlan(predefinedPlan); err != nil { return fmt.Errorf("创建预定义计划 '%s' 失败: %w", predefinedPlan.Name, err) } else { app.Logger.Infof("成功创建预定义计划 '%s'。", predefinedPlan.Name) diff --git a/internal/domain/scheduler/scheduler.go b/internal/domain/scheduler/scheduler.go index 9ef01a9..4c4863d 100644 --- a/internal/domain/scheduler/scheduler.go +++ b/internal/domain/scheduler/scheduler.go @@ -388,12 +388,27 @@ func (s *Scheduler) handlePlanTermination(planLogID uint, reason string) { s.logger.Errorf("取消计划 %d 的后续任务日志时出错: %v", planLogID, err) } - // 4. 将计划本身的状态更新为失败 + // 4. 获取计划执行日志以获取顶层 PlanID planLog, err := s.executionLogRepo.FindPlanExecutionLogByID(planLogID) if err != nil { s.logger.Errorf("无法找到计划执行日志 %d 以更新父计划状态: %v", planLogID, err) return } + + // 5. 获取顶层计划的详细信息,以检查其类型 + topLevelPlan, err := s.planRepo.GetBasicPlanByID(planLog.PlanID) + if err != nil { + s.logger.Errorf("获取顶层计划 %d 的基本信息失败: %v", planLog.PlanID, err) + return + } + + // 6. 如果是系统任务,则不修改计划状态 + if topLevelPlan.PlanType == models.PlanTypeSystem { + s.logger.Warnf("系统任务 %d (日志ID: %d) 执行失败,但根据策略不修改其计划状态。", topLevelPlan.ID, planLogID) + return + } + + // 7. 将计划本身的状态更新为失败 (仅对非系统任务执行) if err := s.planRepo.UpdatePlanStatus(planLog.PlanID, models.PlanStatusFailed); err != nil { s.logger.Errorf("更新计划 %d 状态为 '失败' 时出错: %v", planLog.PlanID, err) }