From ca177f11067d4886e615a2179e943ed6acc687b4 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期一, 15 七月 2019 10:54:07 +0800
Subject: [PATCH] 把es插入路径换成配置文件里的

---
 insertdata/insertDataToEs.go |  145 +++++++++++++++++++++++++-----------------------
 config/conf.yml              |    4 +
 2 files changed, 78 insertions(+), 71 deletions(-)

diff --git a/config/conf.yml b/config/conf.yml
index 2bb51ae..1e2c4df 100644
--- a/config/conf.yml
+++ b/config/conf.yml
@@ -1 +1,3 @@
-photoUrl: "http://192.168.1.182:6333/submit"
\ No newline at end of file
+photoUrl: "http://192.168.1.182:6333/submit"
+videopersons: "http://192.168.1.182:9200/videopersons/perVideoPicture"
+personaction: "http://192.168.1.182:9200/personaction/perVideoAction"
\ No newline at end of file
diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index f967650..60fea44 100644
--- a/insertdata/insertDataToEs.go
+++ b/insertdata/insertDataToEs.go
@@ -19,10 +19,12 @@
 	"ruleprocess/util"
 )
 
-var weedfsUrl string
+var weedfsUrl,videoPersonUrl,personAction string
 
 type conf struct {
-	PhotoUrl string `yaml:"photoUrl"`
+	PhotoUrl     string `yaml:"photoUrl"`
+	Videopersons string `yaml:"videopersons"`
+	Personaction string `yaml:"personaction"`
 }
 
 func init() {
@@ -35,6 +37,8 @@
 	//鎶妝aml褰㈠紡鐨勫瓧绗︿覆瑙f瀽鎴恠truct绫诲瀷
 	yaml.Unmarshal(data, &c)
 	weedfsUrl = c.PhotoUrl
+	videoPersonUrl = c.Videopersons
+	personAction = c.Personaction
 }
 
 // 浜鸿劯鐨勬暟鎹粨鏋�
@@ -202,7 +206,7 @@
 				logger.Info("json parse error ", err)
 				return
 			}
-			err1 := EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
+			err1 := EsReq("POST", videoPersonUrl, requstbody)
 			if err1 != nil {
 				logger.Error("涓婁紶ES鍑洪敊锛�---", err1)
 			}
@@ -211,73 +215,73 @@
 }
 func ChangeStatusFace(msg ruleserver.ResultMsg) {
 	logger.Info("寰�ES鎻掑叆浜鸿劯闈炴姤璀︿絾鏄姸鎬佽浆鎹㈡暟鎹�")
-		// 涓婁紶澶у浘
-		// 瑙e帇缂╁苟涓婁紶鍥剧墖
-		bdata, err := util.UnCompress(msg.Data)
-		if err != nil {
-			panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
-		}
-		i := protomsg.Image{}
-		err = proto.Unmarshal(bdata, &i)
-		bigPhotoUrl := make(map[string]interface{})
-		bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
-		logger.Debug("========澶у浘璺緞锛�", bigPhotoUrl)
-		// 浜鸿劯妫�娴嬶紝娌℃湁鐩镐技鐨勫簳搴撲汉鍛�
-		localConfig, err := cache.GetServerInfo()
-		if err != nil {
-			logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�")
-		}
-		serverIp, err := GetLocalIP()
-		// 鏌ヨcameraName
-		camera, err := cache.GetCameraById(msg.Cid)
-		if err != nil {
-			logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�")
-		}
+	// 涓婁紶澶у浘
+	// 瑙e帇缂╁苟涓婁紶鍥剧墖
+	bdata, err := util.UnCompress(msg.Data)
+	if err != nil {
+		panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
+	}
+	i := protomsg.Image{}
+	err = proto.Unmarshal(bdata, &i)
+	bigPhotoUrl := make(map[string]interface{})
+	bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
+	logger.Debug("========澶у浘璺緞锛�", bigPhotoUrl)
+	// 浜鸿劯妫�娴嬶紝娌℃湁鐩镐技鐨勫簳搴撲汉鍛�
+	localConfig, err := cache.GetServerInfo()
+	if err != nil {
+		logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�")
+	}
+	serverIp, err := GetLocalIP()
+	// 鏌ヨcameraName
+	camera, err := cache.GetCameraById(msg.Cid)
+	if err != nil {
+		logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�")
+	}
 
-		pervideo := PerVideoPicture{
-			uuid.NewV4().String(),
-			msg.Cid,
-			camera.Addr,
-			i.Timestamp,
-			strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1],
-			msg.Tasklab.Taskid,
-			msg.Tasklab.Taskname,
-			"浜鸿劯",
-			"鐘舵�佽浆鎹㈡暟鎹紝闈炴姤璀︽暟鎹�",
-			time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿
-			"",
-			0,
-			"",
-			"",
-			0,
-			0,
-			"",
-			[]string{""},
-			"鏆傛棤闆嗙兢",
-			localConfig.ServerId,
-			localConfig.ServerName,
-			serverIp,
-			"",
-			0,
-			1,
-			0,
-			0,
-			0,
-			[]*protomsg.Baseinfo{},
-		}
-		requstbody, err := json.Marshal(pervideo)
+	pervideo := PerVideoPicture{
+		uuid.NewV4().String(),
+		msg.Cid,
+		camera.Addr,
+		i.Timestamp,
+		strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1],
+		msg.Tasklab.Taskid,
+		msg.Tasklab.Taskname,
+		"浜鸿劯",
+		"鐘舵�佽浆鎹㈡暟鎹紝闈炴姤璀︽暟鎹�",
+		time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿
+		"",
+		0,
+		"",
+		"",
+		0,
+		0,
+		"",
+		[]string{""},
+		"鏆傛棤闆嗙兢",
+		localConfig.ServerId,
+		localConfig.ServerName,
+		serverIp,
+		"",
+		0,
+		1,
+		0,
+		0,
+		0,
+		[]*protomsg.Baseinfo{},
+	}
+	requstbody, err := json.Marshal(pervideo)
 
-		if err != nil {
-			logger.Info("json parse error ", err)
-			return
-		}
-		err1 := EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
-		if err1 != nil {
-			logger.Error("涓婁紶ES鍑洪敊锛�---", err1)
-		}
-		//if msg.RuleResult["cacheData"] != nil {
-		//	InsertFace(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
-		//}
+	if err != nil {
+		logger.Info("json parse error ", err)
+		return
+	}
+	err1 := EsReq("POST", videoPersonUrl, requstbody)
+	if err1 != nil {
+		logger.Error("涓婁紶ES鍑洪敊锛�---", err1)
+	}
+	//if msg.RuleResult["cacheData"] != nil {
+	//	InsertFace(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
+	//}
 }
 
 // 寰�es涓彃鍏olo鏁版嵁
@@ -355,7 +359,7 @@
 			return
 
 		}
-		err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
+		err = EsReq("POST", personAction, requstbody)
 		if err != nil {
 			logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err)
 		} else {
@@ -425,7 +429,7 @@
 		return
 
 	}
-	err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
+	err = EsReq("POST", personAction, requstbody)
 	if err != nil {
 		logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err)
 	} else {
@@ -433,6 +437,7 @@
 		//os.Exit(1)
 	}
 }
+
 // 鑾峰彇鏈満ip
 func GetLocalIP() (ipv4 string, err error) {
 	var (

--
Gitblit v1.8.0