panlei
2019-11-16 32abe72b2cba9b8a7f1e407ac25d181118157f8a
cache/cache.go
@@ -5,6 +5,7 @@
   "basic.com/pubsub/cache.git/shardmap"
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/gopherdiscovery.git"
   "basic.com/valib/logger.git"
   "errors"
   "fmt"
   "github.com/gogo/protobuf/proto"
@@ -18,6 +19,7 @@
   PREFIX_RULE = "RULE_"
   SERVER_KEY = "SERVERINFO"
   PREFIX_SDK = "SDK_"
   PREFIX_SO = "SO"
)
var cMap *shardmap.ShardMap
@@ -53,6 +55,8 @@
   initServerInfo()//初始化服务器配置信息
   initSdks() //初始化sdk列表信息
   initSoData()
   initChan <- true
}
@@ -104,18 +108,19 @@
func initPolygons(){
   var api dbapi.CameraApi
   data := api.FindAllPolygons()
   pMap :=make(map[string][]protomsg.CameraPolygon,0)
   for _, item := range data {
      if item.Type != "line" {
         cameraId :=item.CameraId
         arr, b := cMap.Get(PREFIX_POLYGON + cameraId)
         if b {
            list := arr.([]protomsg.CameraPolygon)
            list = append(list,item)
            cMap.Set(PREFIX_POLYGON + cameraId,list)
         } else {
            cMap.Set(PREFIX_POLYGON + cameraId,[]protomsg.CameraPolygon{ item })
         if _,ok :=pMap[cameraId];ok{
            pMap[cameraId] = append(pMap[cameraId],item)
         } else{
            pMap[cameraId]=[]protomsg.CameraPolygon{item}
         }
      }
   }
   for k,v :=range pMap{
      cMap.Set(PREFIX_POLYGON + k,v)
   }
}
@@ -201,4 +206,23 @@
   } else {
      return sdk,errors.New("sdk not found")
   }
}
func initSoData() {
   var api dbapi.SoApi
   soinfos := api.FindAll()
   logger.Debug("=====================注册表信息:")
   for _,soinfo :=range soinfos {
      logger.Debug(soinfo)
      cMap.Set(PREFIX_SO+soinfo.SdkId,soinfo)
   }
}
func GetSoInfoById(sdkId string) (sdk protomsg.SoInfo,err error){
   obj,b :=cMap.Get(PREFIX_SO + sdkId)
   if b {
      return obj.(protomsg.SoInfo),nil
   } else {
      return sdk,errors.New("sdk not found")
   }
}