定义task接口和delay_task实现
This commit is contained in:
@@ -1,48 +1,63 @@
|
||||
package task
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
)
|
||||
|
||||
const (
|
||||
ParamsDelayDuration = "delay_duration"
|
||||
)
|
||||
|
||||
// DelayTask 是一个用于模拟延迟的 Task 实现
|
||||
type DelayTask struct {
|
||||
id string
|
||||
duration time.Duration
|
||||
priority int
|
||||
done bool
|
||||
executionTask *models.TaskExecutionLog
|
||||
duration time.Duration
|
||||
logger *logs.Logger
|
||||
}
|
||||
|
||||
// NewDelayTask 创建一个新的 DelayTask 实例
|
||||
func NewDelayTask(id string, duration time.Duration, priority int) *DelayTask {
|
||||
func NewDelayTask(logger *logs.Logger, executionTask *models.TaskExecutionLog) *DelayTask {
|
||||
return &DelayTask{
|
||||
id: id,
|
||||
duration: duration,
|
||||
priority: priority,
|
||||
done: false,
|
||||
executionTask: executionTask,
|
||||
logger: logger,
|
||||
}
|
||||
}
|
||||
|
||||
// Execute 执行延迟任务,等待指定的时间
|
||||
func (d *DelayTask) Execute() error {
|
||||
fmt.Printf("任务 %s (%s): 开始延迟 %s...\n", d.id, d.GetDescription(), d.duration)
|
||||
d.logger.Infof("任务 %v: 开始延迟 %v...", d.executionTask.TaskID, d.duration)
|
||||
time.Sleep(d.duration)
|
||||
fmt.Printf("任务 %s (%s): 延迟结束。\n", d.id, d.GetDescription())
|
||||
d.done = true
|
||||
d.logger.Infof("任务 %v: 延迟结束。\n", d.executionTask.TaskID)
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetID 获取任务ID
|
||||
func (d *DelayTask) GetID() string {
|
||||
return d.id
|
||||
func (d *DelayTask) ParseParams() error {
|
||||
if d.executionTask.Task.Parameters == nil {
|
||||
d.logger.Errorf("任务 %v: 缺少参数", d.executionTask.TaskID)
|
||||
return fmt.Errorf("任务 %v: 参数不全", d.executionTask.TaskID)
|
||||
}
|
||||
|
||||
var params map[string]interface{}
|
||||
if err := json.Unmarshal(d.executionTask.Task.Parameters, ¶ms); err != nil {
|
||||
d.logger.Errorf("任务 %v: 解析参数失败: %v", d.executionTask.TaskID, err)
|
||||
return fmt.Errorf("任务 %v: 解析参数失败: %v", d.executionTask.TaskID, err)
|
||||
}
|
||||
|
||||
duration, ok := params[ParamsDelayDuration].(float64)
|
||||
if !ok {
|
||||
d.logger.Errorf("任务 %v: 参数 %v 不是数字或不存在", d.executionTask.TaskID, ParamsDelayDuration)
|
||||
return fmt.Errorf("任务 %v: 参数 %v 不是数字或不存在", d.executionTask.TaskID, ParamsDelayDuration)
|
||||
}
|
||||
|
||||
d.duration = time.Duration(duration)
|
||||
return nil
|
||||
}
|
||||
|
||||
// IsDone 检查任务是否已完成
|
||||
func (d *DelayTask) IsDone() bool {
|
||||
return d.done
|
||||
}
|
||||
|
||||
// GetDescription 获取任务说明,根据任务ID和延迟时间生成
|
||||
func (d *DelayTask) GetDescription() string {
|
||||
return fmt.Sprintf("延迟任务,ID: %s,延迟时间: %s", d.id, d.duration)
|
||||
func (d *DelayTask) OnFailure(executeErr error) {
|
||||
d.logger.Errorf("任务 %v: 执行失败: %v", d.executionTask.TaskID, executeErr)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user