调整任务进度跟踪器, 改为从数据库获取执行进度:

修bug
This commit is contained in:
2025-09-23 17:55:33 +08:00
parent eda5c8dedb
commit 9e129a1ac0
2 changed files with 23 additions and 12 deletions

View File

@@ -256,25 +256,26 @@ func (s *Scheduler) processTask(claimedLog *models.TaskExecutionLog) {
}
planID := planExecutionLog.PlanID // 这才是正确的顶层计划ID
// 获取计划的最新数据
plan, err := s.planRepo.GetPlanByID(planID)
// 获取计划的最新数据,这里我们只需要基本信息来判断执行类型和次数
plan, err := s.planRepo.GetBasicPlanByID(planID)
if err != nil {
s.logger.Errorf("获取计划 %d 的完整信息失败: %v", planID, err)
s.logger.Errorf("获取计划 %d 的基本信息失败: %v", planID, err)
return
}
// 更新计划的执行计数器
plan.ExecuteCount++
// 在内存中计算新的计数值和状态
newExecuteCount := plan.ExecuteCount + 1
newStatus := plan.Status // 默认为当前状态
// 如果是自动计划且达到执行次数上限,则更新计划状态为已停止
if (plan.ExecutionType == models.PlanExecutionTypeAutomatic && plan.ExecuteNum > 0 && plan.ExecuteCount >= plan.ExecuteNum) || plan.ExecutionType == models.PlanExecutionTypeManual {
plan.Status = models.PlanStatusStopeed
s.logger.Infof("计划 %d (自动执行) 已达到最大执行次数 %d,状态更新为 '执行完毕'。", planID, plan.ExecuteNum)
// 如果是自动计划且达到执行次数上限,或计划是手动类型,则更新计划状态为已停止
if (plan.ExecutionType == models.PlanExecutionTypeAutomatic && plan.ExecuteNum > 0 && newExecuteCount >= plan.ExecuteNum) || plan.ExecutionType == models.PlanExecutionTypeManual {
newStatus = models.PlanStatusStopeed
s.logger.Infof("计划 %d 已完成执行,状态更新为 '执行完毕'。", planID)
}
// 保存更新后的计划状态和执行计数
if err := s.planRepo.UpdatePlan(plan); err != nil {
s.logger.Errorf("更新计划 %d 的执行计数和状态失败: %v", planID, err)
// 使用新的、专门的方法来原子性地更新计数值和状态
if err := s.planRepo.UpdatePlanStateAfterExecution(planID, newExecuteCount, newStatus); err != nil {
s.logger.Errorf("更新计划 %d 的执行状态失败: %v", planID, err)
return
}