From e6982607fbbeaa96d3d14409df780266646b793d Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期五, 01 十一月 2019 18:22:44 +0800
Subject: [PATCH] 换一个chan

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

diff --git a/cache/cache.go b/cache/cache.go
index e1120c2..a50f676 100644
--- a/cache/cache.go
+++ b/cache/cache.go
@@ -5,21 +5,26 @@
 	"basic.com/pubsub/cache.git/shardmap"
 	"basic.com/pubsub/protomsg.git"
 	"basic.com/valib/gopherdiscovery.git"
+	"errors"
 	"fmt"
 	"github.com/gogo/protobuf/proto"
 	"github.com/satori/go.uuid"
 	"strconv"
 )
 const (
+	PREFIX_CAMERA = "CAMERA_"
 	PREFIX_POLYGON = "POLYGON_"
 	PREFIX_TIME = "TIME_"
 	PREFIX_RULE = "RULE_"
+	SERVER_KEY = "SERVERINFO"
+	PREFIX_SDK = "SDK_"
 )
 
 var cMap *shardmap.ShardMap
 
 
 func Init(initChan chan bool,dbIp string,surveyPort int,pubSubPort int){
+	cMap = shardmap.New(uint8(32))
 	urlSurvey := "tcp://" + dbIp + ":" + strconv.Itoa(surveyPort)
 	urlPubSub := "tcp://" + dbIp + ":" + strconv.Itoa(pubSubPort)
 	client, _ := gopherdiscovery.ClientWithSub(urlSurvey, urlPubSub, uuid.NewV4().String())
@@ -43,6 +48,12 @@
 
 	initCameraTaskRules()//鍒濆鍖栨憚鍍忔満浠诲姟瑙勫垯缂撳瓨
 
+	initCamera()//鍒濆鍖栨憚鍍忔満淇℃伅
+
+	initServerInfo()//鍒濆鍖栨湇鍔″櫒閰嶇疆淇℃伅
+
+	initSdks() //鍒濆鍖杝dk鍒楄〃淇℃伅
+
 	initChan <- true
 }
 
@@ -54,33 +65,58 @@
 		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")
 
 	}
 }
 
+func initServerInfo() {
+	var api dbapi.SysSetApi
+	b, s := api.GetServerInfo()
+	if b{
+		cMap.Set(SERVER_KEY,s)
+	}
+}
+
+func initCamera() {
+	var api dbapi.CameraApi
+	cameras := api.FindAll()
+	for _,cam := range cameras {
+		cMap.Set(PREFIX_CAMERA + cam.Id,cam)
+	}
+}
+
 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)
 	}
 }
 
@@ -103,7 +139,32 @@
 	}
 }
 
+func initSdks(){
+	var api dbapi.SdkApi
+	sdks := api.FindAll("")
+	for _,sdk :=range sdks {
+		cMap.Set(PREFIX_SDK+sdk.Id,sdk)
+	}
+}
 
+func GetServerInfo() (con protomsg.LocalConfig,err error) {
+	config, b := cMap.Get(SERVER_KEY)
+	if b {
+		return config.(protomsg.LocalConfig),nil
+	} else {
+		return con,errors.New("conf not found")
+	}
+}
+
+//閫氳繃cameraId鑾峰彇鎽勫儚鏈轰俊鎭�
+func GetCameraById(cameraId string) (c protomsg.Camera,err error) {
+	cam, b := cMap.Get(PREFIX_CAMERA + cameraId)
+	if b {
+		return cam.(protomsg.Camera),nil
+	} else {
+		return c,errors.New("camera not found")
+	}
+}
 
 func GetPolygonsByCameraId(cameraId string) []protomsg.CameraPolygon{
 	obj, b := cMap.Get(PREFIX_POLYGON + cameraId)
@@ -132,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