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 |  168 +++++++++++++++++++++++++++----------------------------
 1 files changed, 83 insertions(+), 85 deletions(-)

diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index 10cafc5..99e8323 100644
--- a/insertdata/insertDataToEs.go
+++ b/insertdata/insertDataToEs.go
@@ -101,7 +101,6 @@
 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) {
@@ -135,7 +134,6 @@
 				}
 				i := protomsg.Image{}
 				err = proto.Unmarshal(bdata, &i)
-				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 {
@@ -196,6 +194,7 @@
 				//logger.Info(err.Error())
 			} else {
 				// 浜鸿劯姣斿
+				logger.Warn("___________________________________________杩欐槸鏈塨aseinfo鐨�")
 				localConfig, err := cache.GetServerInfo()
 				if err != nil {
 					logger.Info("鏌ヨ鏈満淇℃伅澶辫触锛�")
@@ -267,93 +266,92 @@
 
 				}
 				err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
-				logger.Info("璺熷簳搴撶殑鐩镐技鏁版嵁---------锛�", face.Liker)
-
-				fmt.Println("------------------------------------------鍝堝搱鍝堝搱锛屽簳搴撴湁浜�")
+				logger.Info("------------------------------------------鍝堝搱鍝堝搱锛屽簳搴撴湁浜�")
 
 			}
 		}
 	}
-	//if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 {
-	//	logger.Info("寰�ES鎻抷olo鏁版嵁")
-	//	var sdkNames string = ""
-	//	alarmRules := []AlarmRule{}
-	//	for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
-	//		sdkNames = sdkNames + yoloResult.SdkName
-	//		alarm := ChangeToString(yoloResult.AlarmLevel)
-	//		alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
-	//	}
-	//	isAlarm := 0
-	//	resp := make(map[string]interface{})
-	//	if len(alarmRules) > 0 {
-	//		isAlarm = 1
-	//		// 瑙e帇缂╁苟涓婁紶鍥剧墖
-	//		bdata, err := util.UnCompress(msg.Data)
-	//		if err != nil {
-	//			panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
-	//		}
-	//		i := protomsg.Image{}
-	//		err = proto.Unmarshal(bdata, &i)
-	//		//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.Error("鐢绘鎴栦笂浼犲浘鐗囨湇鍔″櫒鍑洪敊", err)
-	//		} else {
-	//			logger.Info("宸叉姤璀﹀苟涓婁紶鏀瑰抚鍥剧墖鍒版湇鍔″櫒")
-	//		}
-	//	} else {
-	//		isAlarm = 0
-	//		// 涓嶆槸鎶ヨ鏁版嵁涓嶅瓨
-	//		return
-	//	}
-	//	// logger.Println("鍥剧墖涓婁紶杩斿洖鍊硷細", resp)
-	//	// 鏌ヨ鏈満淇℃伅
-	//	localConfig, err := cache.GetServerInfo()
-	//	if err != nil {
-	//		logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�")
-	//	}
-	//	// 鏌ヨcameraName
-	//	camera, err := cache.GetCameraById(msg.Cid)
-	//	if err != nil {
-	//		logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�")
-	//	}
-	//	serverIp, err := GetLocalIP()
-	//	peraction := Personaction{
-	//		uuid.NewV4().String(),
-	//		msg.Cid,
-	//		camera.Name,
-	//		camera.Addr,
-	//		msg.Tasklab.Taskid,
-	//		msg.Tasklab.Taskname,
-	//		sdkNames,
-	//		"",
-	//		alarmRules,
-	//		localConfig.ServerId,
-	//		localConfig.ServerName,
-	//		serverIp,
-	//		"",
-	//		[]string{strings.Split(resp["fileUrl"].(string), "/")[1]},
-	//		time.Now().Format("2006-01-02 15:04:05"),
-	//		"",
-	//		isAlarm,
-	//		0,
-	//		0,
-	//		0,
-	//	}
-	//	requstbody, err := json.Marshal(peraction)
-	//
-	//	if err != nil {
-	//		logger.Info("json parse error ", err)
-	//		return
-	//
-	//	}
-	//	err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
-	//	if err != nil {
-	//		logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err)
-	//	} else {
-	//		logger.Info("__________________________________________寰�ES鎻掑叆yolo鏁版嵁鎴愬姛")
-	//	}
-	//}
+	if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 {
+		logger.Info("寰�ES鎻抷olo鏁版嵁")
+		var sdkNames string = ""
+		alarmRules := []AlarmRule{}
+		for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
+			sdkNames = sdkNames + yoloResult.SdkName
+			alarm := ChangeToString(yoloResult.AlarmLevel)
+			alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
+		}
+		isAlarm := 0
+		resp := make(map[string]interface{})
+		if len(alarmRules) > 0 {
+			isAlarm = 1
+			// 瑙e帇缂╁苟涓婁紶鍥剧墖
+			bdata, err := util.UnCompress(msg.Data)
+			if err != nil {
+				panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
+			}
+			i := protomsg.Image{}
+			err = proto.Unmarshal(bdata, &i)
+			//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.Error("鐢绘鎴栦笂浼犲浘鐗囨湇鍔″櫒鍑洪敊", err)
+			} else {
+				logger.Info("涓婁紶鐨勫浘鐗囦俊鎭細",resp)
+			}
+		} else {
+			isAlarm = 0
+			// 涓嶆槸鎶ヨ鏁版嵁涓嶅瓨
+			return
+		}
+		// logger.Println("鍥剧墖涓婁紶杩斿洖鍊硷細", resp)
+		// 鏌ヨ鏈満淇℃伅
+		localConfig, err := cache.GetServerInfo()
+		if err != nil {
+			logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�")
+		}
+		// 鏌ヨcameraName
+		camera, err := cache.GetCameraById(msg.Cid)
+		if err != nil {
+			logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�")
+		}
+		serverIp, err := GetLocalIP()
+		peraction := Personaction{
+			uuid.NewV4().String(),
+			msg.Cid,
+			camera.Name,
+			camera.Addr,
+			msg.Tasklab.Taskid,
+			msg.Tasklab.Taskname,
+			sdkNames,
+			"",
+			alarmRules,
+			localConfig.ServerId,
+			localConfig.ServerName,
+			serverIp,
+			"",
+			[]string{strings.Split(resp["fileUrl"].(string), "/")[1]},
+			time.Now().Format("2006-01-02 15:04:05"),
+			"",
+			isAlarm,
+			0,
+			0,
+			0,
+		}
+		requstbody, err := json.Marshal(peraction)
+
+		if err != nil {
+			logger.Info("json parse error ", err)
+			return
+
+		}
+		err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
+		if err != nil {
+			logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err)
+		} else {
+			logger.Warn("__________________________________________寰�ES鎻掑叆yolo鏁版嵁鎴愬姛")
+			//os.Exit(1)
+		}
+	}
 }
 
 // 鑾峰彇鏈満ip

--
Gitblit v1.8.0