From 82d1feccae00f5863dea6d0a85911cf92771f90b Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期六, 22 六月 2019 12:27:15 +0800
Subject: [PATCH] insert data

---
 main.go |   24 ++++++++++++++++++++++--
 1 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 6b5506e..333df4b 100644
--- a/main.go
+++ b/main.go
@@ -3,8 +3,9 @@
 import (
 	"fmt"
 	"ruleprocess/ruleserver"
+	"ruleprocess/insertdata"
+
 	"sync"
-	"analysis/work"
 	"basic.com/pubsub/protomsg.git"
 	"basic.com/valib/deliver.git"
 	"github.com/golang/protobuf/proto"
@@ -44,6 +45,9 @@
 				// 鎶奱rg閲岀殑鎵撶殑鏍囩鎷垮嚭鏉ョ粰m鍐嶅皝瑁呬竴灞�
 				resultMag := ruleserver.ResultMsg{SdkMessage: m, RuleResult: arg.RuleResult}
 				fmt.Println("鎵撳畬鏍囩鍚庣殑缁撴灉锛�",resultMag)
+
+				// 灏嗘墦瀹屾爣绛剧殑鏁版嵁鎻掑叆鍒癊S
+				insertdata.InsertToEs(resultMag)
 			}
 		}
 	}
@@ -77,7 +81,7 @@
 	arg.KeepRight = false
 	arg.IsStatic = false
 	fmt.Println("浠巑ongos涓嬁鍒扮殑鏁版嵁鍖呴暱搴︿负锛�", len(msg))
-	for _, sdkinfo := range m.Tasklab.Sdkinfos {
+	for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo绠楁硶
 		if sdkinfo.Sdktype == "Yolo" {
 			yoloParam := protomsg.ParamYoloObj{}
 			err = proto.Unmarshal(sdkinfo.Sdkdata, &yoloParam)
@@ -91,6 +95,22 @@
 			}
 
 		}
+		if sdkinfo.Sdktype == "FaceDetect" { // 浜鸿劯妫�娴�
+			faceParam := protomsg.ParamFacePos{}
+			err = proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
+			if err != nil {
+				fmt.Println("瑙f瀽sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err)
+				continue
+			}
+			for _, info := range faceParam.Faces {
+				photoMap := ruleserver.PhotoMap{Rects:nil, Score: float64(info.Pos.Quality)}
+				arg.Photo = append(arg.Photo, photoMap)
+			}
+		}
+		//if sdkinfo.Sdktype == "FaceExtract" { // 浜鸿劯鎻愬彇
+
+		//}
+
 	}
 	return m
 }

--
Gitblit v1.8.0