liulei/jx-callback
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

- BaseTask.GetID

Browse Source
This commit is contained in:
gazebo
2018-10-20 10:49:32 +08:00
parent 1c74a43c1b
commit 042d3b5616
2 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
business/jxutils/tasksch/task.go
View File

@@ -25,6 +25,7 @@ type TaskList []ITask
type ITask interface { type ITask interface {
Run() Run()
GetID() string
GetResult(duration time.Duration) (retVal []interface{}, err error) GetResult(duration time.Duration) (retVal []interface{}, err error)
Cancel() Cancel()
GetTotalItemCount() int GetTotalItemCount() int
@@ -104,6 +105,10 @@ func (t *BaseTask) Init(parallelCount, batchSize int, isContinueWhenError bool,
t.C = t.finishChan t.C = t.finishChan
} }
func (t *BaseTask) GetID() string {
return t.ID
}
func (t *BaseTask) GetResult(duration time.Duration) (retVal []interface{}, err error) { func (t *BaseTask) GetResult(duration time.Duration) (retVal []interface{}, err error) {
if t.GetStatus() >= TaskStatusEndBegin { if t.GetStatus() >= TaskStatusEndBegin {
return t.Result, t.Err return t.Result, t.Err

11
business/jxutils/tasksch/task_man.go
View File

@@ -20,7 +20,7 @@ func init() {
func (m *TaskMan) RunTask(taskName string, isContinueWhenError bool, resultHandler ResultHandlerFunc, parallelCount, batchSize int, userName string, worker WorkFunc, itemList interface{}, params ...interface{}) *ParallelTask { func (m *TaskMan) RunTask(taskName string, isContinueWhenError bool, resultHandler ResultHandlerFunc, parallelCount, batchSize int, userName string, worker WorkFunc, itemList interface{}, params ...interface{}) *ParallelTask {
task := RunTask(taskName, isContinueWhenError, resultHandler, parallelCount, batchSize, userName, worker, itemList, params...) task := RunTask(taskName, isContinueWhenError, resultHandler, parallelCount, batchSize, userName, worker, itemList, params...)
m.taskList[task.ID] = task m.ManageTask(task)
return task return task
} }
@@ -39,6 +39,11 @@ func (m *TaskMan) GetTasks(taskID string, fromStatus, toStatus int, lastHours in
return taskList return taskList
} }
func (m *TaskMan) ManageTask(task ITask) ITask {
m.taskList[task.GetID()] = task
return task
}
func RunManagedTask(taskName string, isContinueWhenError bool, resultHandler ResultHandlerFunc, parallelCount, batchSize int, userName string, worker WorkFunc, itemList interface{}, params ...interface{}) *ParallelTask { func RunManagedTask(taskName string, isContinueWhenError bool, resultHandler ResultHandlerFunc, parallelCount, batchSize int, userName string, worker WorkFunc, itemList interface{}, params ...interface{}) *ParallelTask {
return defTaskMan.RunTask(taskName, isContinueWhenError, resultHandler, parallelCount, batchSize, userName, worker, itemList, params...) return defTaskMan.RunTask(taskName, isContinueWhenError, resultHandler, parallelCount, batchSize, userName, worker, itemList, params...)
} }
@@ -46,3 +51,7 @@ func RunManagedTask(taskName string, isContinueWhenError bool, resultHandler Res
func GetTasks(taskID string, fromStatus, toStatus int, lastHours int) (taskList TaskList) { func GetTasks(taskID string, fromStatus, toStatus int, lastHours int) (taskList TaskList) {
return defTaskMan.GetTasks(taskID, fromStatus, toStatus, lastHours) return defTaskMan.GetTasks(taskID, fromStatus, toStatus, lastHours)
} }
func ManageTask(task ITask) ITask {
return defTaskMan.ManageTask(task)
}