abstract.go

@@ -15,8 +15,8 @@ import "context"
 //
 // Date : 14:22 2022/6/23
 type ITask interface {
-	// Name 任务名称标识, 全局唯一
-	Name() string
+	// GetFlag 任务名称标识, 全局唯一
+	GetFlag() string
 	// Description 任务描述
 	Description() string
 	// GetRunID 获取任务ID
@@ -24,5 +24,5 @@ type ITask interface {
 	// Callback 任务执行成功的回调
 	Callback(result *Result) error
 	// Execute 执行任务
-	Execute(ctx context.Context, cfg *Config) *Result
+	Execute(ctx context.Context, cfg *Config) (map[string]interface{}, error)
 }

define.go

@@ -19,6 +19,8 @@ type Config struct {
 	ForbiddenCallback bool
 	// Param 任务执行参数
 	Param map[string]interface{}
+	// TaskFlag 任务标识
+	TaskFlag string
 }
 
 // Result 执行结果

task.go

@@ -51,10 +51,10 @@ func (d *dispatch) Register(taskInstanceList ...ITask) error {
 		if nil == taskInstance {
 			continue
 		}
-		if _, exist := d.taskTable[taskInstance.Name()]; exist {
-			return fmt.Errorf("%s 任务重复注册! ", taskInstance.Name())
+		if _, exist := d.taskTable[taskInstance.GetFlag()]; exist {
+			return fmt.Errorf("%s 任务重复注册! ", taskInstance.GetFlag())
 		}
-		d.taskTable[taskInstance.Name()] = taskInstance
+		d.taskTable[taskInstance.GetFlag()] = taskInstance
 	}
 	return nil
 }
@@ -109,20 +109,19 @@ func (d *dispatch) Run(ctx context.Context, cfg *Config) *Result {
 		TaskConfig:      cfg,
 		Data:            nil,
 		Err:             nil,
-		Async:           cfg.Async,
 	}
 	defer func() {
 		result.FinishTime = time.Now().UnixNano()
 		result.Used = result.FinishTime - result.StartTime
 	}()
 	d.lock.RLock()
-	if taskInstance, exist = d.taskTable[cfg.TaskName]; !exist {
-		result.Err = fmt.Errorf("%v 任务未注册", cfg.TaskName)
+	if taskInstance, exist = d.taskTable[cfg.TaskFlag]; !exist {
+		result.Err = fmt.Errorf("%v 任务未注册", cfg.TaskFlag)
+		return result
 	}
 	d.lock.RUnlock()
 	result.TaskRunID = taskInstance.GetRunID()
 	result.TaskDescription = taskInstance.Description()
-	if cfg.Async {
 	// 异步运行
 	go func() {
 		if e := recover(); nil != e {
@@ -136,9 +135,5 @@ func (d *dispatch) Run(ctx context.Context, cfg *Config) *Result {
 		result.Data, result.Err = taskInstance.Execute(ctx, cfg)
 		_ = taskInstance.Callback(result)
 	}()
-	} else {
-		result.Data, result.Err = taskInstance.Execute(ctx, cfg)
-		_ = taskInstance.Callback(result)
-	}
 	return result
 }