panlei
2019-08-30 77a05fa27d55ea7e39aee47f71cb5f74308e5e0e
cache/cache.go
@@ -9,6 +9,7 @@
   "fmt"
   "github.com/gogo/protobuf/proto"
   "github.com/satori/go.uuid"
   "ruleprocess/logger"
   "strconv"
)
const (
@@ -17,6 +18,8 @@
   PREFIX_TIME = "TIME_"
   PREFIX_RULE = "RULE_"
   SERVER_KEY = "SERVERINFO"
   PREFIX_SDK = "SDK_"
   PREFIX_SO = "SO"
)
var cMap *shardmap.ShardMap
@@ -51,6 +54,10 @@
   initServerInfo()//初始化服务器配置信息
   initSdks() //初始化sdk列表信息
   initSoData()
   initChan <- true
}
@@ -74,6 +81,8 @@
      initCameraTaskRules()
   case protomsg.TableChanged_T_CameraTaskArgs://更新摄像机规则配置参数
      initCameraTaskRules()
   case protomsg.TableChanged_T_Sdk://更新sdk配置
      initSdks()
   default:
      fmt.Println("unknown operation")
@@ -99,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)
   }
}
@@ -130,6 +140,14 @@
   for _, taskArg := range all {
      cMap.Set(PREFIX_RULE+taskArg.CameraId,taskArg.TaskArgs)
   }
}
func initSdks(){
   var api dbapi.SdkApi
   sdks := api.FindAll("")
   for _,sdk :=range sdks {
      cMap.Set(PREFIX_SDK+sdk.Id,sdk)
   }
}
@@ -179,4 +197,32 @@
   } else {
      return nil
   }
}
func GetSdkById(sdkId string) (sdk protomsg.Sdk,err error){
   obj,b :=cMap.Get(PREFIX_SDK + sdkId)
   if b {
      return obj.(protomsg.Sdk),nil
   } 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")
   }
}