From 86813e43325646066009b95446c9deba6dda28b4 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期一, 04 十一月 2019 13:42:27 +0800 Subject: [PATCH] --- --- cache/cache.go | 37 ++++++++++++++++++++++++++++++------- 1 files changed, 30 insertions(+), 7 deletions(-) diff --git a/cache/cache.go b/cache/cache.go index b0578cf..a50f676 100644 --- a/cache/cache.go +++ b/cache/cache.go @@ -17,6 +17,7 @@ PREFIX_TIME = "TIME_" PREFIX_RULE = "RULE_" SERVER_KEY = "SERVERINFO" + PREFIX_SDK = "SDK_" ) var cMap *shardmap.ShardMap @@ -51,6 +52,8 @@ initServerInfo()//鍒濆鍖栨湇鍔″櫒閰嶇疆淇℃伅 + initSdks() //鍒濆鍖杝dk鍒楄〃淇℃伅 + initChan <- true } @@ -74,6 +77,8 @@ initCameraTaskRules() case protomsg.TableChanged_T_CameraTaskArgs://鏇存柊鎽勫儚鏈鸿鍒欓厤缃弬鏁� initCameraTaskRules() + case protomsg.TableChanged_T_Sdk://鏇存柊sdk閰嶇疆 + initSdks() default: fmt.Println("unknown operation") @@ -99,18 +104,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 +136,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 +193,13 @@ } 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") + } } \ No newline at end of file -- Gitblit v1.8.0