From d831bb9c39728d9e0137dd719889e0c7f315f768 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期四, 11 七月 2019 17:49:44 +0800 Subject: [PATCH] === --- cache/cache.go | 91 ++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 81 insertions(+), 10 deletions(-) diff --git a/cache/cache.go b/cache/cache.go index 4a399f8..764b264 100644 --- a/cache/cache.go +++ b/cache/cache.go @@ -5,23 +5,28 @@ "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){ - 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 +48,12 @@ initCameraTaskRules()//鍒濆鍖栨憚鍍忔満浠诲姟瑙勫垯缂撳瓨 + initCamera()//鍒濆鍖栨憚鍍忔満淇℃伅 + + initServerInfo()//鍒濆鍖栨湇鍔″櫒閰嶇疆淇℃伅 + + initSdks() //鍒濆鍖杝dk鍒楄〃淇℃伅 + initChan <- true } @@ -54,15 +65,39 @@ 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) } } @@ -70,14 +105,16 @@ 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 +138,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) @@ -130,4 +192,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