83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
package task
|
||
|
||
import (
|
||
"context"
|
||
|
||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
|
||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/utils"
|
||
)
|
||
|
||
// AnalysisPlanTaskManager 封装了创建和更新计划分析任务（即触发器）的逻辑。
|
||
// 这是一个可被 Scheduler 和其他应用服务（如 PlanService）共享的无状态组件。
|
||
type AnalysisPlanTaskManager struct {
|
||
planRepo repository.PlanRepository
|
||
pendingTaskRepo repository.PendingTaskRepository
|
||
executionLogRepo repository.ExecutionLogRepository
|
||
logger *logs.Logger
|
||
}
|
||
|
||
// NewAnalysisPlanTaskManager 是 AnalysisPlanTaskManager 的构造函数。
|
||
func NewAnalysisPlanTaskManager(
|
||
planRepo repository.PlanRepository,
|
||
pendingTaskRepo repository.PendingTaskRepository,
|
||
executionLogRepo repository.ExecutionLogRepository,
|
||
logger *logs.Logger,
|
||
) *AnalysisPlanTaskManager {
|
||
return &AnalysisPlanTaskManager{
|
||
planRepo: planRepo,
|
||
pendingTaskRepo: pendingTaskRepo,
|
||
executionLogRepo: executionLogRepo,
|
||
logger: logger,
|
||
}
|
||
}
|
||
|
||
// CreateOrUpdateTrigger 为给定的 planID 创建或更新其关联的下一次触发任务。
|
||
// 这个方法是幂等的，可以安全地被多次调用。
|
||
func (m *AnalysisPlanTaskManager) CreateOrUpdateTrigger(ctx context.Context, planID uint) error {
|
||
// 获取计划信息
|
||
plan, err := m.planRepo.GetBasicPlanByID(planID)
|
||
if err != nil {
|
||
m.logger.Errorf("[严重] 获取计划失败, 错误: %v", err)
|
||
return err
|
||
}
|
||
|
||
// 获取触发任务
|
||
task, err := m.planRepo.FindPlanAnalysisTaskByParamsPlanID(planID)
|
||
if err != nil {
|
||
m.logger.Errorf("[严重] 获取计划解析任务失败, 错误: %v", err)
|
||
return err
|
||
}
|
||
|
||
// 写入执行日志
|
||
taskLog := &models.TaskExecutionLog{
|
||
TaskID: task.ID,
|
||
Status: models.ExecutionStatusWaiting,
|
||
}
|
||
if err := m.executionLogRepo.CreateTaskExecutionLogsInBatch([]*models.TaskExecutionLog{taskLog}); err != nil {
|
||
m.logger.Errorf("[严重] 创建任务执行日志失败, 错误: %v", err)
|
||
return err
|
||
}
|
||
|
||
// 写入待执行队列
|
||
next, err := utils.GetNextCronTime(plan.CronExpression)
|
||
if err != nil {
|
||
m.logger.Errorf("[严重] 执行时间解析失败, 错误: %v", err)
|
||
return err
|
||
}
|
||
pendingTask := &models.PendingTask{
|
||
TaskID: task.ID,
|
||
ExecuteAt: next,
|
||
TaskExecutionLogID: taskLog.ID,
|
||
}
|
||
err = m.pendingTaskRepo.CreatePendingTasksInBatch([]*models.PendingTask{pendingTask})
|
||
if err != nil {
|
||
m.logger.Errorf("[严重] 创建待执行任务失败, 错误: %v", err)
|
||
return err
|
||
}
|
||
|
||
m.logger.Infof("成功为 Plan %d 创建/更新了下一次的触发任务，执行时间: %v", planID, next)
|
||
return nil
|
||
}
|