Files
cloudpods/pkg/cloudcommon/db/taskman/worker.go
Qiu Jian df41b7cdcd use separate worker for long running tasks, such as ImageConvertTask,
ImageCheckTask, CloudProviderSyncInfoTask
2018-12-27 16:19:24 +08:00

32 lines
718 B
Go

package taskman
import (
"yunion.io/x/jsonutils"
"yunion.io/x/log"
"yunion.io/x/onecloud/pkg/appsrv"
)
var taskWorkMan *appsrv.SWorkerManager
var taskWorkerTable map[string]*appsrv.SWorkerManager
func init() {
taskWorkMan = appsrv.NewWorkerManager("TaskWorkerManager", 4, 1024, true)
taskWorkerTable = make(map[string]*appsrv.SWorkerManager)
}
func runTask(taskId string, data jsonutils.JSONObject) {
taskName := TaskManager.getTaskName(taskId)
if len(taskName) == 0 {
log.Errorf("no such task??? task_id=%s", taskId)
return
}
worker := taskWorkMan
if workerMan, ok := taskWorkerTable[taskName]; ok {
worker = workerMan
}
worker.Run(func() {
TaskManager.execTask(taskId, data)
}, nil, nil)
}