From b7e65bd6d482ce9d95bd423e64c80344938d21d5 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期五, 12 七月 2019 19:58:27 +0800 Subject: [PATCH] 改造定时器 --- insertdata/insertDataToEs.go | 67 ++++++++++++++++++++++----------- 1 files changed, 45 insertions(+), 22 deletions(-) diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go index 026925f..99e8323 100644 --- a/insertdata/insertDataToEs.go +++ b/insertdata/insertDataToEs.go @@ -1,12 +1,13 @@ package insertdata import ( + "ruleprocess/logger" "encoding/json" "errors" "fmt" + "io/ioutil" "net" "ruleprocess/cache" - "ruleprocess/logger" "strings" "time" @@ -15,9 +16,24 @@ "github.com/satori/go.uuid" "ruleprocess/ruleserver" "ruleprocess/util" + "github.com/go-yaml/yaml" ) +var weedfsUrl string +type conf struct { + PhotoUrl string `yaml:"photoUrl"` +} +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) + weedfsUrl = c.PhotoUrl +} -var weedfsUrl = "http://192.168.1.182:6333/submit" // 浜鸿劯鐨勬暟鎹粨鏋� type PerVideoPicture struct { Id string `json:"id"` @@ -43,6 +59,7 @@ AnalyServerName string `json:"analyServerName"` AnalyServerIp string `json:"analyServerIp"` ClusterId string `json:"clusterId"` + DetectScore float64 `json:"detectScore"` IsAlarm int `json:"isAlarm"` IsAckAlarm int `json:"isAckAlarm"` IsCollect int `json:"isCollect"` @@ -102,7 +119,7 @@ // 浜鸿劯妫�娴嬶紝娌℃湁鐩镐技鐨勫簳搴撲汉鍛� localConfig, err := cache.GetServerInfo() if err != nil { - logger.Info("鏌ヨ鏈満淇℃伅澶辫触锛�") + logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�") } serverIp, err := GetLocalIP() // 瑙e帇缂╁苟涓婁紶鍥剧墖 @@ -113,16 +130,16 @@ // 鏌ヨcameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { - logger.Info("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") + logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) - logger.Info("-------------------------------------------鐪嬩笅瀹藉拰楂�", i.Width, i.Height) bytes := util.SubImg(i, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height)) resp, err := util.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String()) if err != nil { - logger.Info("涓婁紶灏忓浘鍑洪敊") + logger.Error("涓婁紶灏忓浘鍑洪敊") } + logger.Info("================灏忓浘鍦板潃:",resp["fileUrl"].(string)) sex := "" if face.ThftRes.Gender == 1 { sex = "鐢�" @@ -149,13 +166,14 @@ race, face.ThftRes.Smile, face.ThftRes.Beauty, - "", + string(face.Feature), []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, "鏆傛棤闆嗙兢", localConfig.ServerId, localConfig.ServerName, serverIp, "", + face.Score, 1, 0, 0, @@ -169,10 +187,14 @@ return } - err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) + err1 := EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) + if err1 != nil { + logger.Error("涓婁紶ES鍑洪敊锛�---",err1) + } //logger.Info(err.Error()) } else { - // 浜鸿劯鎻愬彇 + // 浜鸿劯姣斿 + logger.Warn("___________________________________________杩欐槸鏈塨aseinfo鐨�") localConfig, err := cache.GetServerInfo() if err != nil { logger.Info("鏌ヨ鏈満淇℃伅澶辫触锛�") @@ -186,16 +208,17 @@ // 鏌ヨcameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { - logger.Info("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") + logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) - logger.Info("-------------------------------------------鐪嬩笅瀹藉拰楂�", i.Width, i.Height) + //logger.Info("-------------------------------------------鐪嬩笅瀹藉拰楂�", i.Width, i.Height) bytes := util.SubImg(i, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height)) resp, err := util.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String()) if err != nil { - logger.Info("涓婁紶灏忓浘鍑洪敊") + logger.Error("涓婁紶灏忓浘鍑洪敊") } + logger.Info("================灏忓浘鍦板潃:",resp["fileUrl"].(string)) sex := "" if face.ThftRes.Gender == 1 { sex = "鐢�" @@ -228,6 +251,7 @@ localConfig.ServerName, serverIp, "", + face.Score, 1, 0, 0, @@ -237,14 +261,12 @@ requstbody, err := json.Marshal(pervideo) if err != nil { - logger.Info("json parse error ", err) + logger.Error("json parse error ", err) return } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) - logger.Info("璺熷簳搴撶殑鐩镐技鏁版嵁---------锛�", face.Liker) - - fmt.Println("------------------------------------------鍝堝搱鍝堝搱锛屽簳搴撴湁浜�") + logger.Info("------------------------------------------鍝堝搱鍝堝搱锛屽簳搴撴湁浜�") } } @@ -272,9 +294,9 @@ //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String()) resp, err = util.DrawPolygonOnImage(msg.Cid, i,msg.RuleResult["yolo"].([]ruleserver.Result)) if err != nil { - logger.Info("鐢绘鎴栦笂浼犲浘鐗囨湇鍔″櫒鍑洪敊", err) + logger.Error("鐢绘鎴栦笂浼犲浘鐗囨湇鍔″櫒鍑洪敊", err) } else { - logger.Info("宸叉姤璀﹀苟涓婁紶鏀瑰抚鍥剧墖鍒版湇鍔″櫒") + logger.Info("涓婁紶鐨勫浘鐗囦俊鎭細",resp) } } else { isAlarm = 0 @@ -285,12 +307,12 @@ // 鏌ヨ鏈満淇℃伅 localConfig, err := cache.GetServerInfo() if err != nil { - logger.Info("鏌ヨ鏈満淇℃伅澶辫触锛�") + logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�") } // 鏌ヨcameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { - logger.Info("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") + logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") } serverIp, err := GetLocalIP() peraction := Personaction{ @@ -324,9 +346,10 @@ } err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) if err != nil { - logger.Info("寰�ES鎻掑叆鏁版嵁澶辫触", err) + logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err) } else { - logger.Info("__________________________________________寰�ES鎻掑叆yolo鏁版嵁鎴愬姛") + logger.Warn("__________________________________________寰�ES鎻掑叆yolo鏁版嵁鎴愬姛") + //os.Exit(1) } } } -- Gitblit v1.8.0