From 04dfa328304ea805d93d04b577058f07b669bf14 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 02 七月 2019 18:16:50 +0800
Subject: [PATCH] 调整yolo报警策略
---
cache/cache.go | 179 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 179 insertions(+), 0 deletions(-)
diff --git a/cache/cache.go b/cache/cache.go
index 08bf029..5f4c813 100644
--- a/cache/cache.go
+++ b/cache/cache.go
@@ -1 +1,180 @@
package cache
+
+import (
+ "basic.com/dbapi.git"
+ "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){
+ 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)
+
+ initCacheData(initChan)
+
+ peers, _ := client.Peers()
+ for b := range peers{
+ fmt.Println("peerMsg:",b)
+ updateData(b)
+ }
+}
+
+func initCacheData(initChan chan bool) {
+
+ initPolygons()//鍒濆鍖栨憚鍍忔満澶氳竟褰㈢紦瀛�
+
+ initTimeRules()//鍒濆鍖栨椂闂磋鍒欑紦瀛�
+
+ initCameraTaskRules()//鍒濆鍖栨憚鍍忔満浠诲姟瑙勫垯缂撳瓨
+
+ initCamera()//鍒濆鍖栨憚鍍忔満淇℃伅
+
+ initServerInfo()//鍒濆鍖栨湇鍔″櫒閰嶇疆淇℃伅
+
+ initChan <- true
+}
+
+var newUpdateMsg = &protomsg.DbChangeMessage{}
+
+func updateData(b []byte){
+ if err :=proto.Unmarshal(b,newUpdateMsg);err !=nil{
+ fmt.Println("dbChangeMsg unmarshal err:",err)
+ 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_CameraTaskArgs://鏇存柊鎽勫儚鏈鸿鍒欓厤缃弬鏁�
+ initCameraTaskRules()
+ 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()
+ 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 })
+ }
+ }
+ }
+}
+
+func initTimeRules(){
+ var api dbapi.CameraApi
+ flag, rules := api.FindAllTimeRules()
+ if flag {
+ for _, rule := range rules {
+ cMap.Set(PREFIX_TIME+rule.Id, rule)
+ }
+ }
+}
+
+func initCameraTaskRules(){
+ var api dbapi.CameraTaskArgsApi
+ all := api.FindAll()
+
+ for _, taskArg := range all {
+ cMap.Set(PREFIX_RULE+taskArg.CameraId,taskArg.TaskArgs)
+ }
+}
+
+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)
+ if b {
+ return obj.([]protomsg.CameraPolygon)
+ } else {
+ return nil
+ }
+}
+
+//浠庣紦瀛樹腑鑾峰彇鏃堕棿瑙勫垯
+func GetTimeRuleById(id string) (exist bool,rule protomsg.CameraTimerule){
+ obj, b := cMap.Get(PREFIX_TIME + id)
+ if b {
+ return true,obj.(protomsg.CameraTimerule)
+ } else {
+ return false,rule
+ }
+}
+
+//鏍规嵁鎽勫儚鏈篿d浠庣紦瀛樹腑鑾峰彇鎽勫儚鏈虹殑浠诲姟瑙勫垯璁剧疆
+func GetCameraTaskRulesByCameraId(cameraId string) []*protomsg.TaskGroupArgs {
+ obj, b := cMap.Get(PREFIX_RULE + cameraId)
+ if b {
+ return obj.([]*protomsg.TaskGroupArgs)
+ } else {
+ return nil
+ }
+}
\ No newline at end of file
--
Gitblit v1.8.0