From c51eac62d2d750b2324f464c0aa4562c7b957216 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期五, 15 十一月 2019 18:18:52 +0800
Subject: [PATCH] ---
---
cache/cache.go | 115 ++++++++++++++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 105 insertions(+), 10 deletions(-)
diff --git a/cache/cache.go b/cache/cache.go
index 4a399f8..ffb0912 100644
--- a/cache/cache.go
+++ b/cache/cache.go
@@ -5,23 +5,30 @@
"basic.com/pubsub/cache.git/shardmap"
"basic.com/pubsub/protomsg.git"
"basic.com/valib/gopherdiscovery.git"
+ "basic.com/valib/logger.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_"
+ PREFIX_SO = "SO"
)
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 +50,14 @@
initCameraTaskRules()//鍒濆鍖栨憚鍍忔満浠诲姟瑙勫垯缂撳瓨
+ initCamera()//鍒濆鍖栨憚鍍忔満淇℃伅
+
+ initServerInfo()//鍒濆鍖栨湇鍔″櫒閰嶇疆淇℃伅
+
+ initSdks() //鍒濆鍖杝dk鍒楄〃淇℃伅
+
+ initSoData()
+
initChan <- true
}
@@ -54,31 +69,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 {
- 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
+ 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)
}
}
@@ -101,7 +143,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 +197,32 @@
} 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")
+ }
+}
+
+func initSoData() {
+ var api dbapi.SoApi
+ soinfos := api.FindAll()
+ logger.Debug("=====================娉ㄥ唽琛ㄤ俊鎭細")
+ for _,soinfo :=range soinfos {
+ logger.Debug(soinfo)
+ cMap.Set(PREFIX_SO+soinfo.SdkId,soinfo)
+ }
+}
+
+func GetSoInfoById(sdkId string) (sdk protomsg.SoInfo,err error){
+ obj,b :=cMap.Get(PREFIX_SO + sdkId)
+ if b {
+ return obj.(protomsg.SoInfo),nil
+ } else {
+ return sdk,errors.New("sdk not found")
+ }
}
\ No newline at end of file
--
Gitblit v1.8.0