Przeglądaj źródła

修改定时任务方法为带次数的方法

liuxiulin 1 miesiąc temu
rodzic
commit
267cf131c6

+ 1 - 0
pkg/rule/task_message.go

@@ -10,6 +10,7 @@ type TaskLifecycleMessage struct {
 type TimerTaskMessage struct {
 	TaskId  string        `json:"task_id"` // 任务Id
 	Cron    string        `json:"cron"`    // 任务执行的cron表达式
+	Times   int           `json:"time"`    // 执行次数
 	Actions []*TaskAction `json:"actions"` // 执行动作列表
 }
 

+ 1 - 1
services/timer-service/internal/scheduler.go

@@ -36,7 +36,7 @@ func (t *TaskSchedule) AddTask(msg []byte) error {
 	newJson := gjson.New(task.Actions)
 	server.Log.Debugf("add task: taskId:%s,corn:%s,actions:%s", task.TaskId, task.Cron, newJson.MustToJsonString())
 	// 创建任务
-	entity, err := t.cron.Add(context.Background(), task.Cron, func(ctx context.Context) {
+	entity, err := t.cron.AddTimes(context.Background(), task.Cron, task.Times, func(ctx context.Context) {
 		if err = NewTaskExecutor(task.Actions).Do(); err != nil {
 			server.Log.Errorf("do taskid :%s error:%s", task.TaskId, err.Error())
 		}