pig/pig-farm-controller
1
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
修bug
This commit is contained in:
huang
2025-09-23 17:55:33 +08:00
parent eda5c8dedb
commit 9e129a1ac0
2 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
internal/app/service/task/scheduler.go
View File

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

10
internal/infra/repository/plan_repository.go
View File

@@ -62,6 +62,9 @@ type PlanRepository interface {
// StopPlanTransactionally 停止一个计划的执行，包括更新状态、移除待执行任务和更新执行日志 // StopPlanTransactionally 停止一个计划的执行，包括更新状态、移除待执行任务和更新执行日志
StopPlanTransactionally(planID uint) error StopPlanTransactionally(planID uint) error
// UpdatePlanStateAfterExecution 更新计划执行后的状态（计数和状态）
UpdatePlanStateAfterExecution(planID uint, newCount uint, newStatus models.PlanStatus) error
} }
// gormPlanRepository 是 PlanRepository 的 GORM 实现 // gormPlanRepository 是 PlanRepository 的 GORM 实现
@@ -735,3 +738,10 @@ func (r *gormPlanRepository) UpdatePlanStatus(id uint, status models.PlanStatus)
} }
return nil return nil
} }
func (r *gormPlanRepository) UpdatePlanStateAfterExecution(planID uint, newCount uint, newStatus models.PlanStatus) error {
return r.db.Model(&models.Plan{}).Where("id = ?", planID).Updates(map[string]interface{}{
"execute_count": newCount,
"status": newStatus,
}).Error
}