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