75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package tasksch
|
|
|
|
import (
|
|
"time"
|
|
|
|
"git.rosy.net.cn/baseapi/utils"
|
|
"git.rosy.net.cn/jx-callback/globals"
|
|
)
|
|
|
|
const (
|
|
StepBegin = "begin"
|
|
StepEnd = "End"
|
|
)
|
|
|
|
type SeqWorkFunc func(step int, params ...interface{}) (result interface{}, err error) // 只有最后一次返回结果保留
|
|
|
|
type SeqTask struct {
|
|
BaseTask
|
|
worker SeqWorkFunc
|
|
}
|
|
|
|
func NewSeqTask(taskName string, userName string, worker SeqWorkFunc, stepCount int, params ...interface{}) *SeqTask {
|
|
task := &SeqTask{
|
|
worker: worker,
|
|
}
|
|
task.Init(1, 1, false, params, taskName, userName, stepCount, stepCount)
|
|
return task
|
|
}
|
|
|
|
func (task *SeqTask) Run() {
|
|
task.run(func() {
|
|
globals.SugarLogger.Debugf("SeqTask.Run %s", task.Name)
|
|
var taskErr error
|
|
var taskResult []interface{}
|
|
for i := 0; i < task.TotalItemCount; i++ {
|
|
select {
|
|
case <-task.quitChan:
|
|
goto EndFor
|
|
default:
|
|
}
|
|
result, err := task.worker(i, task.params...)
|
|
task.finishedOneJob(1, err)
|
|
if taskErr = err; taskErr != nil {
|
|
globals.SugarLogger.Infof("SeqTask.Run %s step:%d failed with error:%v", task.Name, i, err)
|
|
if !task.IsContinueWhenError {
|
|
break
|
|
}
|
|
} else if result != nil {
|
|
taskResult = append(taskResult, utils.Interface2Slice(result)...)
|
|
}
|
|
}
|
|
EndFor:
|
|
task.locker.Lock()
|
|
if taskErr != nil { // 如果有错误，肯定就是失败了
|
|
task.Status = TaskStatusFailed
|
|
} else {
|
|
if task.FinishedJobCount < task.TotalJobCount {
|
|
taskErr = ErrTaskIsCanceled
|
|
task.Status = TaskStatusCanceled
|
|
} else {
|
|
task.Status = TaskStatusFinished
|
|
}
|
|
}
|
|
task.Err = taskErr
|
|
task.Result = taskResult
|
|
task.TerminatedAt = time.Now()
|
|
task.locker.Unlock()
|
|
|
|
globals.SugarLogger.Debugf("SeqTask.Run %s, result:%v, err:%v", task.Name, taskResult, taskErr)
|
|
|
|
close(task.finishChan)
|
|
close(task.quitChan)
|
|
})
|
|
}
|