76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/domain/plan"
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
|
)
|
|
|
|
type DelayTaskParams struct {
|
|
DelayDuration float64 `json:"delay_duration"`
|
|
}
|
|
|
|
// DelayTask 是一个用于模拟延迟的 Task 实现
|
|
type DelayTask struct {
|
|
ctx context.Context
|
|
executionTask *models.TaskExecutionLog
|
|
duration time.Duration
|
|
}
|
|
|
|
func NewDelayTask(ctx context.Context, executionTask *models.TaskExecutionLog) plan.Task {
|
|
return &DelayTask{
|
|
ctx: ctx,
|
|
executionTask: executionTask,
|
|
}
|
|
}
|
|
|
|
// Execute 执行延迟任务，等待指定的时间
|
|
func (d *DelayTask) Execute(ctx context.Context) error {
|
|
taskCtx, logger := logs.Trace(ctx, d.ctx, "Execute")
|
|
if err := d.parseParameters(taskCtx); err != nil {
|
|
return err
|
|
}
|
|
|
|
logger.Infof("任务 %v: 开始延迟 %v...", d.executionTask.TaskID, d.duration)
|
|
time.Sleep(d.duration)
|
|
logger.Infof("任务 %v: 延迟结束。", d.executionTask.TaskID)
|
|
return nil
|
|
}
|
|
|
|
func (d *DelayTask) parseParameters(ctx context.Context) error {
|
|
logger := logs.TraceLogger(ctx, d.ctx, "parseParameters")
|
|
if d.executionTask.Task.Parameters == nil {
|
|
logger.Errorf("任务 %v: 缺少参数", d.executionTask.TaskID)
|
|
return fmt.Errorf("任务 %v: 参数不全", d.executionTask.TaskID)
|
|
}
|
|
|
|
var params DelayTaskParams
|
|
err := d.executionTask.Task.ParseParameters(¶ms)
|
|
if err != nil {
|
|
logger.Errorf("任务 %v: 解析参数失败: %v", d.executionTask.TaskID, err)
|
|
return fmt.Errorf("任务 %v: 解析参数失败: %v", d.executionTask.TaskID, err)
|
|
}
|
|
|
|
if params.DelayDuration <= 0 {
|
|
logger.Errorf("任务 %v: 参数 delay_duration 缺失或无效 (必须大于0)", d.executionTask.TaskID)
|
|
return fmt.Errorf("任务 %v: 参数 delay_duration 缺失或无效 (必须大于0)", d.executionTask.TaskID)
|
|
}
|
|
|
|
d.duration = time.Duration(params.DelayDuration) * time.Second
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *DelayTask) OnFailure(ctx context.Context, executeErr error) {
|
|
logger := logs.TraceLogger(ctx, d.ctx, "OnFailure")
|
|
logger.Errorf("任务 %v: 执行失败: %v", d.executionTask.TaskID, executeErr)
|
|
}
|
|
|
|
func (d *DelayTask) ResolveDeviceIDs(ctx context.Context) ([]uint, error) {
|
|
return []uint{}, nil
|
|
}
|