From 4d21ccf808d91239f9200b15a46977f79b5cb199 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期一, 01 七月 2019 18:46:20 +0800
Subject: [PATCH] cache

---
 cache/cache.go |  132 ++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 132 insertions(+), 0 deletions(-)

diff --git a/cache/cache.go b/cache/cache.go
index 08bf029..4a399f8 100644
--- a/cache/cache.go
+++ b/cache/cache.go
@@ -1 +1,133 @@
 package cache
+
+import (
+	"basic.com/dbapi.git"
+	"basic.com/pubsub/cache.git/shardmap"
+	"basic.com/pubsub/protomsg.git"
+	"basic.com/valib/gopherdiscovery.git"
+	"fmt"
+	"github.com/gogo/protobuf/proto"
+	"github.com/satori/go.uuid"
+	"strconv"
+)
+const (
+	PREFIX_POLYGON = "POLYGON_"
+	PREFIX_TIME = "TIME_"
+	PREFIX_RULE = "RULE_"
+)
+
+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)
+	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()//鍒濆鍖栨憚鍍忔満浠诲姟瑙勫垯缂撳瓨
+
+	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_CameraPolygon://鏇存柊鎽勫儚鏈哄鍙樺舰
+		initPolygons()
+	case protomsg.TableChanged_T_TimeRule://鏇存柊鏃堕棿瑙勫垯
+		initTimeRules()
+	case protomsg.TableChanged_T_CameraTaskArgs://鏇存柊鎽勫儚鏈鸿鍒欓厤缃弬鏁�
+		initCameraTaskRules()
+	default:
+		fmt.Println("unknown operation")
+
+	}
+}
+
+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 })
+		}
+	}
+}
+
+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 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