From fc87731788af22601270789e935cf068edc80f79 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 09 七月 2019 10:50:59 +0800
Subject: [PATCH] ---

---
 cache/cache.go |   69 +++++++++++++++++++++++++++++-----
 1 files changed, 59 insertions(+), 10 deletions(-)

diff --git a/cache/cache.go b/cache/cache.go
index 4a399f8..b0578cf 100644
--- a/cache/cache.go
+++ b/cache/cache.go
@@ -5,23 +5,27 @@
 	"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"
 )
 
 var cMap *shardmap.ShardMap
 
 
 func Init(initChan chan bool,dbIp string,surveyPort int,pubSubPort int){
-	urlSurvey := dbIp + strconv.Itoa(surveyPort)
-	urlPubSub := dbIp + strconv.Itoa(pubSubPort)
+	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())
 	recvMsg := client.HeartBeatMsg()
 	fmt.Println(<-recvMsg)
@@ -43,6 +47,10 @@
 
 	initCameraTaskRules()//鍒濆鍖栨憚鍍忔満浠诲姟瑙勫垯缂撳瓨
 
+	initCamera()//鍒濆鍖栨憚鍍忔満淇℃伅
+
+	initServerInfo()//鍒濆鍖栨湇鍔″櫒閰嶇疆淇℃伅
+
 	initChan <- true
 }
 
@@ -54,10 +62,16 @@
 		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()
 	default:
@@ -66,18 +80,36 @@
 	}
 }
 
+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()
 	for _, item := range data {
-		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 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 })
+			}
 		}
 	}
 }
@@ -101,7 +133,24 @@
 	}
 }
 
+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)

--
Gitblit v1.8.0