From db16d6383bfec3388b091b734007bd29cf8bbf20 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期六, 02 十一月 2019 16:55:56 +0800
Subject: [PATCH] ---

---
 cache/cache.go |   39 ++++++++++++++++++++++++++++++++-------
 1 files changed, 32 insertions(+), 7 deletions(-)

diff --git a/cache/cache.go b/cache/cache.go
index 5f4c813..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
 }
 
@@ -70,8 +73,12 @@
 		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")
 
@@ -97,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)
 	}
 }
 
@@ -128,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)
 	}
 }
 
@@ -177,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