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
}
@@ -62,12 +69,20 @@
      return
   }
   switch newUpdateMsg.Table {
   case protomsg.TableChanged_T_Server:
      initServerInfo()
   case protomsg.TableChanged_T_Camera:
      initCamera()
   case protomsg.TableChanged_T_CameraPolygon://更新摄像机多变形
      initPolygons()
   case protomsg.TableChanged_T_TimeRule://更新时间规则
      initTimeRules()
   case protomsg.TableChanged_T_CameraTask://更新摄像机规则配置参数
      initCameraTaskRules()
   case protomsg.TableChanged_T_CameraTaskArgs://更新摄像机规则配置参数
      initCameraTaskRules()
   case protomsg.TableChanged_T_Sdk://更新sdk配置
      initSdks()
   default:
      fmt.Println("unknown operation")
@@ -93,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)
   }
}
@@ -124,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)
   }
}
@@ -173,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")
   }
}