From c6bc736739b2f95c9c9717127c6aad66bb8068d5 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期三, 18 九月 2019 19:49:27 +0800 Subject: [PATCH] 给定时器加锁 --- ruleserver/readyDataForRule.go | 112 ++++++++++++++++++++++++++++++++++++++------------------ 1 files changed, 76 insertions(+), 36 deletions(-) diff --git a/ruleserver/readyDataForRule.go b/ruleserver/readyDataForRule.go index b64d22b..a7ff52d 100644 --- a/ruleserver/readyDataForRule.go +++ b/ruleserver/readyDataForRule.go @@ -2,14 +2,11 @@ import ( "basic.com/dbapi.git" - bigCache "basic.com/pubsub/cache.git" "basic.com/pubsub/protomsg.git" "encoding/json" "errors" "fmt" - "github.com/go-yaml/yaml" "github.com/golang/protobuf/proto" - "io/ioutil" "math" "nanomsg.org/go-mangos" "nanomsg.org/go-mangos/protocol/req" @@ -33,30 +30,13 @@ } func Init() { - data, err := ioutil.ReadFile("./config/conf.yml") - if err != nil { - fmt.Println("璇诲彇閰嶇疆鏂囦欢鍑洪敊--", err) - logger.Error("璇诲彇閰嶇疆鏂囦欢鍑洪敊--", err) - } - c := conf{} - //鎶妝aml褰㈠紡鐨勫瓧绗︿覆瑙f瀽鎴恠truct绫诲瀷 - yaml.Unmarshal(data, &c) - logger.Debug("閰嶇疆鏂囦欢鐨勫�间负锛�", c) - serverIp = c.ServerIp - serverPort = c.ServerPort - dbTablePersons = c.DbTablePersons - localConfig, err1 := cache.GetServerInfo() - if err1 != nil { - logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�") - } - logger.Debug("鏈満淇℃伅鍜宻erver淇℃伅锛�", localConfig, serverIp, serverPort) - bigCache.Init(dbTablePersons, serverIp, serverPort, localConfig.ServerId) - sock, err = req.NewSocket(); - if err != nil { - logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error()) - } + //logger.Debug("鏈満淇℃伅鍜宻erver淇℃伅锛�", localConfig, serverIp, serverPort) + //bigCache.Init(dbTablePersons, serverIp, serverPort, localConfig.ServerId) + go Push1() } +var sender chan *protomsg.CompareArgs = make(chan *protomsg.CompareArgs) +var receiver chan []byte = make(chan []byte) type BaseInfo struct { TableId string `json:"tableId"` TableName string `json:"tableName"` @@ -83,17 +63,21 @@ } } logger.Debug("褰撳墠鏁版嵁甯ц鍖归厤鐨勮鍒欑粍锛�-------------------------","鎽勫儚鏈篿d:",cameraId,"浠诲姟id",taskId) - for _,ruleGroup := range taskGroup.GroupRules{ - logger.Info(ruleGroup.GroupText) + if taskGroup == nil { + return nil + } else { + for _,ruleGroup := range taskGroup.GroupRules{ + logger.Info(ruleGroup.GroupText) + } + return taskGroup } - return taskGroup } // 鏍规嵁鎽勫儚鏈篿d鎷垮埌鎽勫儚鏈烘墍鏈夊尯鍩� func GetPolygons(cameraId string) []protomsg.CameraPolygon { var cameraPolygons []protomsg.CameraPolygon cameraPolygons = cache.GetPolygonsByCameraId(cameraId) - //logger.Debug("------=======鏌ョ湅涓嬪叏閮ㄥ尯鍩燂細",cameraPolygons) + logger.Debug("------=======鏌ョ湅涓嬪叏閮ㄥ尯鍩燂細",cameraPolygons) return cameraPolygons } @@ -175,12 +159,14 @@ CompareThreshold:compareThreshold, Source:false, } - serverIp, _ := GetLocalIP() - bytes := Push("tcp://"+serverIp+":40010",comArg,sock) + + bytes := getCompareMsg(comArg) + //bytes := []byte{} var scResult protomsg.SdkCompareResult err1 := proto.Unmarshal(bytes, &scResult) if err1 != nil { logger.Error("getBaseInfo瑙e帇閿欒", err1) + return } //logger.Info("----------------------------------------map鏄�", m) ids := []string{} @@ -292,7 +278,7 @@ func (a *AreaMap) CountAreaObjs(arg *SdkData) { a.targetNum = 0 - threshold := 0.0 // 鐩镐技搴� + threshold := 80.0 // 鐩镐技搴� intersectionper := 0.2 // 鍗犳瘮 size := 0.0 // 灏哄 @@ -332,7 +318,7 @@ } } -func Push(url string,data *protomsg.CompareArgs,sock mangos.Socket) []byte{ +func Push(data *protomsg.CompareArgs,sock mangos.Socket) []byte{ //var sock mangos.Socket var err error var msg []byte @@ -345,11 +331,13 @@ logger.Error("Failed set MaxRecvSize: %v", err) return nil } - //sock.AddTransport(ipc.NewTransport()) - sock.AddTransport(tcp.NewTransport()) - if err = sock.Dial(url); err != nil { + serverIP, _ := GetLocalIP() + if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil { logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error()) } + //sock.AddTransport(ipc.NewTransport()) + sock.AddTransport(tcp.NewTransport()) + logger.Info("搴忓垪鍖栨暟鎹�") bytes,err1 := proto.Marshal(data) logger.Info("鏁版嵁闀垮害涓猴細",len(bytes)) @@ -370,6 +358,58 @@ return msg } + +func Push1(){ + //var sock mangos.Socket + var err error + var msg []byte + + if sock, err = req.NewSocket(); err != nil { + logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error()) + } + errSize := sock.SetOption(mangos.OptionMaxRecvSize,5*1024*1024) + if errSize != nil { + logger.Error("Failed set MaxRecvSize: %v", err) + } + //sock.AddTransport(ipc.NewTransport()) + sock.AddTransport(tcp.NewTransport()) + serverIP, _ := GetLocalIP() + if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil { + logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error()) + } + logger.Info("搴忓垪鍖栨暟鎹�") + + for { + select { + // case <-ctx.Done(): + // return + case data := <- sender: + bytes,err1 := proto.Marshal(data) + logger.Info("鏁版嵁闀垮害涓猴細",len(bytes)) + if err1 != nil { + logger.Info("搴忓垪鍖栧け璐ワ細",err1) + } + logger.Debug("鎺ㄩ�佹暟鎹�") + //bytes := []byte("ndfasojdfaidsos") + if err = sock.Send(bytes); err != nil { + logger.Error("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error()) + //os.Exit(1) + } + if msg, err = sock.Recv(); err != nil { + logger.Error("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error()) + //os.Exit(1) + } + logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲,鏁版嵁闀垮害涓猴細",len(msg)) + receiver <- msg + default: + + } + } +} +func getCompareMsg(data *protomsg.CompareArgs) []byte{ + sender <- data + return <-receiver +} // 鑾峰彇鏈満ip func GetLocalIP() (ipv4 string, err error) { var ( -- Gitblit v1.8.0