From c775ea17ffd8b064fb5109567e89a169299373a1 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 10 七月 2019 17:24:36 +0800
Subject: [PATCH] 更新cache

---
 insertdata/insertDataToEs.go |   39 ++++++++++++++++++++++++++++++++-------
 1 files changed, 32 insertions(+), 7 deletions(-)

diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index 39335bd..f67c8ed 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"`
@@ -84,6 +101,7 @@
 func InsertToEs(msg ruleserver.ResultMsg) {
 
 	// 鐩存帴浠庤鍒欑殑鏍囩鏁版嵁閲屾嬁绗﹀悎瑙勫垯鐨勪汉鑴哥粨鏋�
+	//logger.Info("------------------------------------浠庨厤缃枃浠惰鍙栫殑鍊硷細",weedfsUrl)
 	if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 {
 		logger.Info("寰�ES鎻掍汉鑴告暟鎹�")
 		for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) {
@@ -117,7 +135,7 @@
 				}
 				i := protomsg.Image{}
 				err = proto.Unmarshal(bdata, &i)
-				//logger.Info("-------------------------------------------鐪嬩笅瀹藉拰楂�", i.Width, i.Height)
+				logger.Info("-------------------------------------------鐪嬩笅瀹藉拰楂�", int(face.Location.X),int(face.Location.Y),int(face.Location.X+face.Location.Width),int(face.Location.Y+face.Location.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 {
@@ -150,13 +168,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,
@@ -170,10 +189,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("鏌ヨ鏈満淇℃伅澶辫触锛�")
@@ -230,6 +253,7 @@
 					localConfig.ServerName,
 					serverIp,
 					"",
+					face.Score,
 					1,
 					0,
 					0,
@@ -328,7 +352,8 @@
 		if err != nil {
 			logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err)
 		} else {
-			logger.Info("__________________________________________寰�ES鎻掑叆yolo鏁版嵁鎴愬姛")
+			logger.Warn("__________________________________________寰�ES鎻掑叆yolo鏁版嵁鎴愬姛")
+			//os.Exit(1)
 		}
 	}
 }

--
Gitblit v1.8.0