From a3a788801fcd2efb16d58e91f393bad8d7730b96 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 26 六月 2019 18:35:19 +0800
Subject: [PATCH] 加上人脸,过完规则的标签区分人脸和yolo

---
 insertdata/insertDataToEs.go |  144 +++++++++++++++++++++++++----------------------
 1 files changed, 76 insertions(+), 68 deletions(-)

diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index 00fe813..5cc012e 100644
--- a/insertdata/insertDataToEs.go
+++ b/insertdata/insertDataToEs.go
@@ -93,76 +93,84 @@
 // 寰�ES鎻掓暟鎹�
 func InsertToEs(msg ruleserver.ResultMsg) {
 	log.Println("寰�ES鎻掓暟鎹�")
-
-	for _, sdkinfo := range msg.Tasklab.Sdkinfos {
-		if sdkinfo.Sdktype == "FaceDetect" {
-			if len(sdkinfo.Sdkdata) > 1 {
-
-				faceParam := protomsg.ParamFacePos{}
-				err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
-				if err1 != nil {
-					log.Println("瑙f瀽FACE sdk鏈夎", err1)
-					continue
-				}
-				for _, face := range faceParam.Faces {
-					pervideo := PerVideoPicture{
-						uuid.NewV4().String(),
-						msg.Cid,
-						msg.Caddr,
-						time.Now().Format("2006-01-02 15:04:05"),
-						"",
-						msg.Tasklab.Taskid,
-						msg.Tasklab.Taskname,
-						sdkinfo.SdkName,
-						"",
-						"",
-						face.Result.Gender,
-						face.Result.Age,
-						"",
-						face.Result.Race,
-						face.Result.Smile,
-						face.Result.Beauty,
-						"涓嶆槸姣忎釜浜鸿劯绠楁硶閮芥湁",
-						"---",
-						"",
-						"",
-						"",
-						"",
-						"",
-						"",
-						"",
-						"",
-						0,
-						Base{
-							"鏄瘡涓汉鑴哥畻娉曢兘鏈夊悧",
-							"",
-							0,
-							"",
-							"",
-							"",
-							"",
-							"",
-							"",
-							"",
-							"",
-						},
-					}
-					requstbody, err := json.Marshal(pervideo)
-
-					if err != nil {
-						log.Println("json parse error ", err)
-						return
-
-					}
-					err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
-					if err != nil {
-						log.Println("es can not execute right.")
-					}
-				}
-			} else {
-				continue
+	// 鐩存帴浠庤鍒欑殑鏍囩鏁版嵁閲屾嬁绗﹀悎瑙勫垯鐨勪汉鑴哥粨鏋�
+	for _, result := range msg.RuleResult {
+		if !result.IsYolo {
+			for _,face1 := range result.Faces{
+				println(face1)
 			}
 		}
+	}
+	for _, sdkinfo := range msg.Tasklab.Sdkinfos {
+		//if sdkinfo.Sdktype == "FaceDetect" {
+		//	if len(sdkinfo.Sdkdata) > 1 {
+		//
+		//		faceParam := protomsg.ParamFacePos{}
+		//		err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
+		//		if err1 != nil {
+		//			log.Println("瑙f瀽FACE sdk鏈夎", err1)
+		//			continue
+		//		}
+		//		for _, face := range faceParam.Faces {
+		//			pervideo := PerVideoPicture{
+		//				uuid.NewV4().String(),
+		//				msg.Cid,
+		//				msg.Caddr,
+		//				time.Now().Format("2006-01-02 15:04:05"),
+		//				"",
+		//				msg.Tasklab.Taskid,
+		//				msg.Tasklab.Taskname,
+		//				sdkinfo.SdkName,
+		//				"",
+		//				"",
+		//				face.Result.Gender,
+		//				face.Result.Age,
+		//				"",
+		//				face.Result.Race,
+		//				face.Result.Smile,
+		//				face.Result.Beauty,
+		//				"涓嶆槸姣忎釜浜鸿劯绠楁硶閮芥湁",
+		//				"---",
+		//				"",
+		//				"",
+		//				"",
+		//				"",
+		//				"",
+		//				"",
+		//				"",
+		//				"",
+		//				0,
+		//				Base{ // 鍙湁浜鸿劯姣斿鏈夎繖涓俊鎭紝浜鸿劯妫�娴嬪苟娌℃湁
+		//					"鏄瘡涓汉鑴哥畻娉曢兘鏈夊悧",
+		//					"",
+		//					0,
+		//					"",
+		//					"",
+		//					"",
+		//					"",
+		//					"",
+		//					"",
+		//					"",
+		//					"",
+		//				},
+		//			}
+		//			requstbody, err := json.Marshal(pervideo)
+		//
+		//			if err != nil {
+		//				log.Println("json parse error ", err)
+		//				return
+		//
+		//			}
+		//			err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
+		//			if err != nil {
+		//				log.Println("es can not execute right.")
+		//			}
+		//		}
+		//
+		//	} else {
+		//		continue
+		//	}
+		//}
 		if sdkinfo.Sdktype == "Yolo" {
 			if len(sdkinfo.Sdkdata) > 1 {
 

--
Gitblit v1.8.0