重构代码

chenshijun

2019-07-27 b5560d3cae1164f016ca9339592eda6b9008fb9f

 tasktag/tasktag.go

@@ -4,27 +4,28 @@
   "sync"

   "basic.com/pubsub/protomsg.git"
   "taskpubsub/util"
    "taskpubsub/logger"
   "taskpubsub/util"
)

var TaskMapLab sync.Map
var TaskLabelMap sync.Map

func Init() {

   logger.Info("============= init tasktag info =====================")
   GenTaskMap()
   genTaskLabelMap()
   go func(taskflag chan bool) {
      for _ = range taskflag {
         GenTaskMap()
         genTaskLabelMap()
         logger.Info("update task finished!")
      }

   }(util.TaskSdkflag)
}

func GenTaskMap() {
   var tls []protomsg.TaskLabel
// 从sqlite 接口拿到所有的任务, 每一个任务都有自己的几个算法
//以 taskid 作为key, 对应的算法组合作为 value
func genTaskLabelMap() {
   var newtls []protomsg.TaskLabel
   for _, taskSdk := range util.TaskSdks {

      var tl protomsg.TaskLabel
@@ -39,25 +40,23 @@
            tl.Sdkinfos = append(tl.Sdkinfos, sdkinfowithtask)
      }
      tl.Index = int32(0)
      tls = append(tls, tl)
      newtls = append(newtls, tl)
   }
   GenTasklab(tls)
   TaskMapLab.Range(func(k, v interface{}) bool {
   updateTaskLabelMap(newtls)
   TaskLabelMap.Range(func(k, v interface{}) bool {
      logger.Info(k, v)
      return true
   })
}

// 从sqlite 接口拿到所有的任务, 每一个任务都有自己的几个算法
//以 taskid 作为key, 对应的算法组合作为 value
func GenTasklab(tasklab []protomsg.TaskLabel) {
   TaskMapLab.Range(func(key interface{}, value interface{}) bool {
      TaskMapLab.Delete(key)
func updateTaskLabelMap(taskLabel []protomsg.TaskLabel) {
   TaskLabelMap.Range(func(key interface{}, value interface{}) bool {
      TaskLabelMap.Delete(key)
      return true
   })

   for _, value := range tasklab {
   for _, value := range taskLabel {
      pv := value
      TaskMapLab.Store(value.Taskid, &pv)
      TaskLabelMap.Store(value.Taskid, &pv)
   }
}