package tasktag
|
|
import (
|
"sync"
|
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/logger.git"
|
"taskpubsub/util"
|
)
|
|
var TaskLabelMap sync.Map
|
|
func Init() {
|
|
logger.Info("============= init tasktag info =====================")
|
genTaskLabelMap()
|
go func(taskflag chan bool) {
|
for _ = range taskflag {
|
genTaskLabelMap()
|
logger.Info("update task finished!")
|
}
|
}(util.TaskSdkflag)
|
}
|
|
// 从sqlite 接口拿到所有的任务, 每一个任务都有自己的几个算法
|
//以 taskid 作为key, 对应的算法组合作为 value
|
func genTaskLabelMap() {
|
var newtls []protomsg.TaskLabel
|
for _, taskSdk := range util.TaskSdks {
|
|
var tl protomsg.TaskLabel
|
tl.Taskid = taskSdk.Task.Taskid
|
tl.Taskname = taskSdk.Task.Taskname
|
|
for _, sdkinfo := range taskSdk.Sdks {
|
sdkinfowithtask := new(protomsg.SdkmsgWithTask)
|
sdkinfowithtask.Ipcid = sdkinfo.IpcId
|
sdkinfowithtask.Sdktype = sdkinfo.SdkType
|
logger.Info("======sdkinfowithtask.Sdktype:", sdkinfowithtask.Sdktype)
|
sdkinfowithtask.Sdkdata = make([]byte, 1)
|
tl.Sdkinfos = append(tl.Sdkinfos, sdkinfowithtask)
|
}
|
tl.Index = int32(0)
|
newtls = append(newtls, tl)
|
}
|
updateTaskLabelMap(newtls)
|
TaskLabelMap.Range(func(k, v interface{}) bool {
|
logger.Info(k, v)
|
return true
|
})
|
}
|
|
func updateTaskLabelMap(taskLabel []protomsg.TaskLabel) {
|
TaskLabelMap.Range(func(key interface{}, value interface{}) bool {
|
TaskLabelMap.Delete(key)
|
return true
|
})
|
|
for _, value := range taskLabel {
|
pv := value
|
TaskLabelMap.Store(value.Taskid, &pv)
|
}
|
}
|
