From 10503ddbe9f65f03dca53d1dd4d4b3a72fc7e6af Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 13 十一月 2019 14:38:26 +0800
Subject: [PATCH] 删除islink

---
 insertdata/insertDataToEs.go |  340 ++++++++++++++++++++++++++++++++++---------------------
 1 files changed, 209 insertions(+), 131 deletions(-)

diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index a558f45..b363f36 100644
--- a/insertdata/insertDataToEs.go
+++ b/insertdata/insertDataToEs.go
@@ -40,6 +40,7 @@
 	Id              string                `json:"id"`
 	CameraId        string                `json:"cameraId"`
 	CameraAddr      string                `json:"cameraAddr"`
+	CameraName      string                `json:"cameraName"`
 	PicDate         string                `json:"picDate"`
 	PicMaxUrl       string                `json:"picMaxUrl"`
 	TaskId          string                `json:"taskId"`
@@ -61,24 +62,25 @@
 	AnalyServerName string                `json:"analyServerName"`
 	AnalyServerIp   string                `json:"analyServerIp"`
 	ClusterId       string                `json:"clusterId"`
-	LinkId          string                `json:"linkId"`
 	DetectScore     float64               `json:"detectScore"`
 	IsAlarm         int                   `json:"isAlarm"`
 	IsAckAlarm      int                   `json:"isAckAlarm"`
 	IsCollect       int                   `json:"isCollect"`
 	IsDelete        int                   `json:"isDelete"`
 	BaseInfo        []*structure.BaseInfo `json:"baseInfo"`
-	TargetInfo      Target
+	TargetInfo      []Target
+	LinkTag         string          `json:"linkTag"`
+	LinkTagInfo     []*PerVideoPicture `json:"linkTagInfo"`
 }
 
 type Target struct {
-	TargetId       string `json:"targetId"`
+	TargetId       string  `json:"targetId"`
 	TargetScore    float64 `json:"targetScore"`
 	TargetLocation Points
 }
 
 type Points struct {
-	TopLeft  Point `json:"topLeft"`
+	TopLeft     Point `json:"topLeft"`
 	BottomRight Point `json:"bottomRight"`
 }
 
@@ -86,6 +88,7 @@
 	Lat float64 `json:"lat"`
 	Lon float64 `json:"lon"`
 }
+
 //  yolo琛屼负鐨勬暟鎹粨鏋�
 type Personaction struct {
 	Id              string      `json:"id"`
@@ -103,13 +106,14 @@
 	ClusterId       string      `json:"clusterId"`
 	PicSmUrl        []string    `json:"picSmUrl"`
 	PicDate         string      `json:"picDate"`
-	LinkId          string      `json:"linkId"`
 	VideoUrl        string      `json:"videoUrl"`
 	IsAlarm         int         `json:"isAlarm"`
 	IsAckAlarm      int         `json:"isAckAlarm"`
 	IsCollect       int         `json:"isCollect"`
 	IsDelete        int         `json:"isDelete"`
-	TargetInfo		[]Target
+	TargetInfo      []Target
+	LinkTag         string          `json:"linkTag"`
+	LinkTagInfo     []*PerVideoPicture `json:"linkTagInfo"`
 }
 
 type AlarmRule struct {
@@ -117,7 +121,6 @@
 	AlarmLevel   string `json:"alarmLevel"`
 	RuleText     string `json:"ruleText"`
 	DefenceState bool   `json:"defenceState"`
-	IsLink       bool   `json:"isLink"`
 	LinkInfo     string `json:"linkInfo"`
 }
 
@@ -127,30 +130,6 @@
 	rules []structure.Result
 }
 
-// 寰�ES鎻掓暟鎹�
-//func InsertToEs(msg ruleserver.ResultMsg) {
-//	var timeLabel string
-//	// 鐩存帴浠庤鍒欑殑鏍囩鏁版嵁閲屾嬁绗﹀悎瑙勫垯鐨勪汉鑴哥粨鏋�
-//	if msg.RuleResult["timeLabel"] != nil {
-//		timeLabel = msg.RuleResult["timeLabel"].(string)
-//	}
-//	logger.Debug("鎻掑叆鏁版嵁鍓嶇湅鐪嬫姤璀︽爣蹇椾綅锛�", timeLabel)
-//	if timeLabel == "01" { // 鏃犲畾鏃跺櫒鐘舵�佽鎻掑叆鐨勬姤璀︽暟鎹�
-//		InsertFace(msg)
-//		flag := ruleserver.BodyIsSame(msg.SdkMessage)
-//		if !flag {
-//			InsertYolo(msg)
-//		}
-//	}
-//	if timeLabel == "10" { // 瀹氭椂鍣ㄧ姸鎬佽鎻掑叆鐨勯甯ф姤璀︽暟鎹�傝繛甯︾潃瀹氭椂鍣ㄥ紑鍚椂鐨勯偅甯�
-//		InsertFace(msg)
-//		InsertYolo(msg)
-//	}
-//	//if timeLabel == "12" { // 骞堕潪鎶ヨ鏁版嵁锛屽彧鏄姸鎬佹敼鍙樼殑鏁版嵁
-//	//	//ChangeStatusFace(msg)
-//	//	ChangeStatusYolo(msg)
-//	//}
-//}
 func InsertToEs(msg structure.ResultMsg) {
 	defer func() {
 		if err := recover(); err != nil {
@@ -164,25 +143,12 @@
 	weedfsUrl = "http://" + localConfig1.WebPicIp + ":" + strconv.Itoa(int(localConfig1.WebPicPort)) + "/submit"
 	videoPersonUrl = "http://" + localConfig1.AlarmIp + ":" + strconv.Itoa(int(localConfig1.AlarmPort)) + "/" + EsInfo.EsIndex.VideoPersons.IndexName + "/" + EsInfo.EsIndex.VideoPersons.IndexType
 	personAction = "http://" + localConfig1.AlarmIp + ":" + strconv.Itoa(int(localConfig1.AlarmPort)) + "/" + EsInfo.EsIndex.Personaction.IndexName + "/" + EsInfo.EsIndex.Personaction.IndexType
-	InsertFace(msg, "")
-	// 濡傛灉鏍囩涓惈鏈夋寔缁椂闂撮娆℃姤璀︾殑timeLabel鐨勮瘽鍒欎笉闇�瑕佽繃浜轰綋杩借釜锛屼笉鐒跺氨娌$殑鎻掑叆浜�
-	fk := ruleserver.TrackOrNot(msg.RuleResult)
-	if fk {
-		InsertYolo(msg, "")
-		//if msg.Cid == "61de081a-7ed9-4970-8432-41d642c35456" {
-		//	logger.Warn("鎹曟崏188鎽勫儚鏈虹殑鎸佺画鏃堕棿浠诲姟.....")
-		//	os.Exit(1)
-		//}
-	} else {
-		flag := ruleserver.BodyIsSame(msg.SdkMessage)
-		if !flag {
-			InsertYolo(msg, "")
-		}
-	}
+	InsertFace(msg)
+	InsertYolo(msg)
 }
 
 // 寰�es涓彃鍏ヤ汉鑴告暟鎹�
-func InsertFace(msg structure.ResultMsg, linkId string) {
+func InsertFace(msg structure.ResultMsg) {
 	if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]structure.FaceResult)) > 0 {
 		logger.Info("寰�ES鎻掍汉鑴告暟鎹�")
 		faces := []*FaceAndRules{}
@@ -199,14 +165,20 @@
 					panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
 				}
 				alarmRules := []AlarmRule{}
-				logger.Warn("浜鸿劯id涓猴細", face.Id, "浜鸿劯鐨勮鍒欓暱搴︿负锛�", len(face.rules))
 				//os.Exit(1)
 				for _, faceResult := range face.rules {
 					alarm := ChangeToString(faceResult.DefenceState, faceResult.AlarmLevel)
-					alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText, faceResult.DefenceState, faceResult.IsLink, ""})
+					alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText, faceResult.DefenceState, ""})
 				}
 				i := protomsg.Image{}
 				err = proto.Unmarshal(bdata, &i)
+				// 鍏堜紶灏忓浘锛屽啀浼犲ぇ鍥撅紝闃叉鑴镐笂鏈夌嚎
+				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 {
+					logger.Error("涓婁紶灏忓浘鍑洪敊")
+				}
+				// 涓婁紶澶у浘
 				if imgMaxUrl == "" {
 					bigPhotoUrl := make(map[string]interface{})
 					bigPhotoUrl, err = util.DrawPolygonOnImageForFace(msg.Cid, i, msg.RuleResult["face"].([]structure.FaceResult), weedfsUrl)
@@ -214,23 +186,6 @@
 					imgMaxUrl = bigPhotoUrl["fileUrl"].(string)
 					picTime = i.Timestamp
 				}
-				// 浜鸿劯妫�娴嬶紝娌℃湁鐩镐技鐨勫簳搴撲汉鍛�
-				localConfig, err := cache.GetServerInfo()
-				if err != nil {
-					logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�")
-				}
-				serverIp, err := GetLocalIP()
-				// 鏌ヨcameraName
-				camera, err := cache.GetCameraById(msg.Cid)
-				if err != nil {
-					logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�")
-				}
-				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 {
-					logger.Error("涓婁紶灏忓浘鍑洪敊")
-				}
-				//logger.Info("================灏忓浘鍦板潃:", resp["fileUrl"].(string))
 				sex := ""
 				logger.Info(sex)
 				if face.ThftRes.Gender == 1 {
@@ -241,21 +196,18 @@
 				race := getRaceString(face.ThftRes.Race)
 				ageDescription := getDescription(face.ThftRes.Age)
 				logger.Info(ageDescription)
-				esDataId := uuid.NewV4().String()
-				linksId := ""
-				if linkId != "" {
-					linksId = linkId
-				}
 				var target = new(Target)
 				target.TargetId = strconv.FormatUint(face.Id, 10)
 				target.TargetScore = face.Score
-				target.TargetLocation = Points{TopLeft:Point{face.Location.X,face.Location.Y},BottomRight:Point{face.Location.X+face.Location.Width,face.Location.Y+face.Location.Height}}
+				target.TargetLocation = Points{TopLeft: Point{face.Location.X, face.Location.Y}, BottomRight: Point{face.Location.X + face.Location.Width, face.Location.Y + face.Location.Height}}
 				//logger.Info("浜鸿劯鐨刬d:",strconv.FormatUint(face.Id, 10))
-				logger.Info("%%%%%%%%%%%%%%%%%%%%%鐩爣淇℃伅锛�",*target)
+				var targetInfos []Target
+				targetInfos = append(targetInfos,*target)
 				pervideo := PerVideoPicture{
-					esDataId,
+					face.Uuid,
 					msg.Cid,
-					camera.Addr,
+					msg.Push.Cam.Addr,
+					msg.Push.Cam.Name,
 					picTime,
 					imgMaxUrl,
 					msg.Tasklab.Taskid,
@@ -272,19 +224,20 @@
 					face.ThftRes.Beauty,
 					base64.StdEncoding.EncodeToString(face.Feature),
 					[]string{resp["fileUrl"].(string)},
-					"鏆傛棤闆嗙兢",
-					localConfig.ServerId,
-					localConfig.ServerName,
-					serverIp,
 					"",
-					linksId,
+					msg.Push.ServerId,
+					msg.Push.ServerName,
+					msg.Push.LocalIp,
+					"",
 					face.Score,
 					1,
 					0,
 					0,
 					0,
 					face.Liker,
-					*target,
+					targetInfos,
+					"",
+					nil,
 				}
 				requstbody, err := json.Marshal(pervideo)
 
@@ -298,7 +251,7 @@
 				} else {
 					logger.Info("鎻掑叆es杩斿洖鐨勪俊鎭細", resp1)
 					// 鍙戝嚭褰曞儚淇″彿
-					ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 1})
+					ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: msg.Push.PushId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{}, Type: 1})
 				}
 			}
 		}
@@ -319,12 +272,12 @@
 func hebingFace(faces []*FaceAndRules, faceResult structure.FaceResult) []*FaceAndRules {
 	for _, arg := range faceResult.Args {
 		// 鎷垮埌姣忎竴寮犱汉鑴�
-		logger.Info("褰掔疆浜鸿劯鏃剁浉浼艰�呯殑鏁伴噺锛�", len(arg.Liker))
+		//logger.Info("褰掔疆浜鸿劯鏃剁浉浼艰�呯殑鏁伴噺锛�", len(arg.Liker))
 		flag := false
 		for _, face := range faces {
-			for _, lik := range face.Liker {
-				logger.Warn("--------鍚堝苟浜鸿劯鏃剁浉浼艰�咃細", lik.PersonId, lik.TableName)
-			}
+			//for _, lik := range face.Liker {
+			//	//logger.Warn("--------鍚堝苟浜鸿劯鏃剁浉浼艰�咃細", lik.PersonId, lik.TableName)
+			//}
 			if arg.Id == face.Id {
 				flag = true
 				face.rules = append(face.rules, faceResult.Result)
@@ -351,7 +304,7 @@
 }
 
 // 寰�es涓彃鍏olo鏁版嵁
-func InsertYolo(msg structure.ResultMsg, linkId string) {
+func InsertYolo(msg structure.ResultMsg) {
 	if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]structure.Result)) > 0 {
 		// 鍏堝垽鏂竴涓嬫暟鎹甫鐨勮鍒欐爣绛炬槸鍚︽湁鍙互鎻掑叆鐨�
 		flag := false
@@ -377,7 +330,7 @@
 					if yoloResult.IsLink {
 						linkInfo = "鑱斿姩浠诲姟"
 					}
-					alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, yoloResult.IsLink, linkInfo})
+					alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, linkInfo})
 					// 涓婁紶缂撳瓨鏁版嵁鐨勫浘鐗囨嬁鍒皍rl
 					if yoloResult.Others.CacheData != nil {
 						//InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
@@ -403,10 +356,10 @@
 						}
 					}
 					// 瑁呴厤鐩爣淇℃伅鏁版嵁
-					for _,target := range yoloResult.Location  {
+					for _, target := range yoloResult.Location {
 						// 鍘婚噸娣诲姞
 						var flag = true
-						for _,selectTarget := range targetInfos  {
+						for _, selectTarget := range targetInfos {
 							if strconv.FormatUint(target.TargetId, 10) == selectTarget.TargetId {
 								flag = false
 								break
@@ -416,30 +369,26 @@
 							var target1 = new(Target)
 							target1.TargetId = strconv.FormatUint(target.TargetId, 10)
 							target1.TargetScore = target.TargetScore
-							target1.TargetLocation = Points{TopLeft:Point{target.X,target.Y},BottomRight:Point{target.X+target.Width,target.Y+target.Height}}
-							targetInfos = append(targetInfos,*target1)
+							target1.TargetLocation = Points{TopLeft: Point{target.X, target.Y}, BottomRight: Point{target.X + target.Width, target.Y + target.Height}}
+							targetInfos = append(targetInfos, *target1)
 						}
 					}
 				}
 			}
-			linkFlag := false
+
+			linkTagInfos := []*PerVideoPicture{}
 			for _, yoloResult := range msg.RuleResult["yolo"].([]structure.Result) {
 				if (yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10") && yoloResult.Others.LinkCache != nil && len(yoloResult.Others.LinkCache) > 1 {
-					linkId := uuid.NewV4().String()
 					for _, msg2 := range yoloResult.Others.LinkCache {
-						logger.Warn("鎻掑叆鑱斿姩鏁版嵁", "姝ゅ抚鏁版嵁鐨刬d涓�", msg2.Cid)
-						InsertYolo(msg2, linkId)
-						linkFlag = true
+						// 鎶妋sg2鐨勬暟鎹閰嶆垚涓�涓狿erVideoPicture缁撴瀯浣�
+						linkTagInfos = append(linkTagInfos,msg2PersonVideo(msg2))
 					}
 				}
 			}
-			if linkFlag {
-				// 鑱斿姩鏁版嵁涓寘鍚湰甯ф暟鎹紝鎻掑叆鑱斿姩鏁版嵁鍚庡氨涓嶉渶瑕佹彃鍏ユ湰甯ф暟鎹簡
-				logger.Warn("鎴愬姛鎻掑叆涓や釜鑱斿姩鍥剧墖")
-				//os.Exit(1)
-				return
+			linkTag := ""
+			if len(linkTagInfos) > 0 {
+				linkTag = "鑱斿姩浠诲姟"
 			}
-			//logger.Info("--------璧板埌杩欏効灏变笉涓�鏍�")
 			isAlarm := 0
 			resp := make(map[string]interface{})
 			// 瑙e帇缂╁苟涓婁紶鍥剧墖
@@ -464,57 +413,39 @@
 				// 涓嶆槸鎶ヨ鏁版嵁涓嶅瓨
 				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()
-			logger.Info("%%%%%%%%%%%%%%%%%yolo鐨則arget淇℃伅锛�",targetInfos)
 			if resp["fileUrl"] != nil {
 				url = append(url, resp["fileUrl"].(string))
-				esDataId := uuid.NewV4().String()
-				linksId := ""
-				if linkId != "" {
-					linksId = linkId
-				}
+				//esDataId := uuid.NewV4().String()
 				peraction := Personaction{
-					esDataId,
+					msg.Push.PushId,
 					msg.Cid,
-					camera.Name,
-					camera.Addr,
+					msg.Push.Cam.Name,
+					msg.Push.Cam.Addr,
 					msg.Tasklab.Taskid,
 					msg.Tasklab.Taskname,
 					sdkNames,
 					"",
 					alarmRules,
-					localConfig.ServerId,
-					localConfig.ServerName,
-					serverIp,
+					msg.Push.ServerId,
+					msg.Push.ServerName,
+					msg.Push.LocalIp,
 					"",
 					url,
 					i.Timestamp,
-					linksId,
 					"",
 					isAlarm,
 					0,
 					0,
 					0,
 					targetInfos,
+					linkTag,
+					linkTagInfos,
 				}
 				requstbody, err := json.Marshal(peraction)
 
 				if err != nil {
 					logger.Info("json parse error ", err)
 					return
-
 				}
 				resp1, err2 := EsReq("POST", personAction, requstbody)
 				if err2 != nil {
@@ -522,7 +453,7 @@
 				} else {
 					logger.Debug("鎻掑叆es杩斿洖鐨勬暟鎹俊鎭槸锛�", resp1)
 					// 鍙戝嚭褰曞儚淇″彿
-					ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 2})
+					ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: msg.Push.PushId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{}, Type: 2})
 					logger.Warn("__________________________________________寰�ES鎻掑叆yolo鏁版嵁鎴愬姛")
 					//os.Exit(1)
 				}
@@ -615,3 +546,150 @@
 	}
 	return race
 }
+
+func msg2PersonVideo (msg structure.ResultMsg) *PerVideoPicture{
+	if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]structure.Result)) > 0 {
+		// 鍏堝垽鏂竴涓嬫暟鎹甫鐨勮鍒欐爣绛炬槸鍚︽湁鍙互鎻掑叆鐨�
+		flag := false
+		for _, res := range msg.RuleResult["yolo"].([]structure.Result) {
+			//logger.Info("瀹氭椂鍣ㄦ墦鐨勬暟瀛楁爣绛撅細",res.Others.TimeLabel)
+			if res.Others.TimeLabel == "01" || res.Others.TimeLabel == "10" {
+				flag = true
+			}
+		}
+		if flag {
+			logger.Info("鎻掑叆Yolo缁勮鏁版嵁")
+			var sdkNames string = ""
+			alarmRules := []AlarmRule{}
+			var targetInfos []Target
+			url := []string{}
+			for _, yoloResult := range msg.RuleResult["yolo"].([]structure.Result) {
+				if yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10" {
+					// 鎷煎嚭sdkname
+					//logger.Info("搴旇杩涙潵鎵嶅鐨�")
+					sdkNames = sdkNames + yoloResult.SdkName
+					alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
+					linkInfo := ""
+					if yoloResult.IsLink {
+						linkInfo = "鑱斿姩浠诲姟"
+					}
+					alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, yoloResult.IsLink, linkInfo})
+					// 涓婁紶缂撳瓨鏁版嵁鐨勫浘鐗囨嬁鍒皍rl
+					if yoloResult.Others.CacheData != nil {
+						//InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
+						// 鎶婄紦瀛樼殑鏁版嵁涓婁紶鍚庡緱鍒板湴鍧�瀛樿繘鍘�
+						// 瑙e帇缂╁苟涓婁紶鍥剧墖
+						msgs := yoloResult.Others.CacheData
+						for _, msg1 := range msgs {
+							bdata, err := util.UnCompress(msg1.Data)
+							if err != nil {
+								panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
+							}
+							i := protomsg.Image{}
+							err = proto.Unmarshal(bdata, &i)
+							resp1, err1 := util.DrawPolygonOnImageForYolo(msg1.Cid, i, msg1.RuleResult["yolo"].([]structure.Result), weedfsUrl)
+							if err1 != nil {
+								logger.Error("缂撳瓨鏁版嵁鐢绘鎴栦笂浼犲浘鐗囨湇鍔″櫒鍑洪敊", err)
+							} else {
+								logger.Info("涓婁紶鐨勫浘鐗囦俊鎭細", resp1)
+							}
+							if resp1["fileUrl"] != nil {
+								url = append(url, resp1["fileUrl"].(string))
+							}
+						}
+					}
+					// 瑁呴厤鐩爣淇℃伅鏁版嵁
+					for _, target := range yoloResult.Location {
+						// 鍘婚噸娣诲姞
+						var flag = true
+						for _, selectTarget := range targetInfos {
+							if strconv.FormatUint(target.TargetId, 10) == selectTarget.TargetId {
+								flag = false
+								break
+							}
+						}
+						if flag {
+							var target1 = new(Target)
+							target1.TargetId = strconv.FormatUint(target.TargetId, 10)
+							target1.TargetScore = target.TargetScore
+							target1.TargetLocation = Points{TopLeft: Point{target.X, target.Y}, BottomRight: Point{target.X + target.Width, target.Y + target.Height}}
+							targetInfos = append(targetInfos, *target1)
+						}
+					}
+				}
+			}
+			isAlarm := 0
+			resp := make(map[string]interface{})
+			// 瑙e帇缂╁苟涓婁紶鍥剧墖
+			bdata, err := util.UnCompress(msg.Data)
+			if err != nil {
+				panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
+			}
+			i := protomsg.Image{}
+			err = proto.Unmarshal(bdata, &i)
+			if len(alarmRules) > 0 {
+				isAlarm = 1
+				//resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
+				resp, err = util.DrawPolygonOnImageForYolo(msg.Cid, i, msg.RuleResult["yolo"].([]structure.Result), weedfsUrl)
+				if err != nil {
+					logger.Error("鐢绘鎴栦笂浼犲浘鐗囨湇鍔″櫒鍑洪敊", err)
+					return nil
+				} else {
+					logger.Info("涓婁紶鐨勫浘鐗囦俊鎭細", resp)
+				}
+			} else {
+				isAlarm = 0
+				// 涓嶆槸鎶ヨ鏁版嵁涓嶅瓨
+				return nil
+			}
+			if resp["fileUrl"] != nil {
+				url = append(url, resp["fileUrl"].(string))
+				//esDataId := uuid.NewV4().String()
+				pervideo := &PerVideoPicture{
+					msg.Push.PushId + "-" + uuid.NewV4().String(),
+					msg.Cid,
+					msg.Push.Cam.Addr,
+					msg.Push.Cam.Name,
+					i.Timestamp,
+					"",
+					msg.Tasklab.Taskid,
+					msg.Tasklab.Taskname,
+					"浜鸿劯",
+					"",
+					alarmRules,
+					time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿
+					"",                                      //鏆傛敼涓轰汉鑴竔d strconv.FormatUint(face.Id, 10)
+					0,
+					"", // 鏆傛敼涓哄垎鍊糵mt.Sprintf("%.2f",face.Score)
+					"",
+					0,
+					0,
+					"",
+					[]string{resp["fileUrl"].(string)},
+					"",
+					msg.Push.ServerId,
+					msg.Push.ServerName,
+					msg.Push.LocalIp,
+					"",
+					0,
+					isAlarm,
+					0,
+					0,
+					0,
+					nil,
+					targetInfos,
+					"",
+					nil,
+				}
+				return pervideo
+			} else {
+				return nil
+			}
+		} else {
+			logger.Debug("timeLabel鏉′欢閮戒笉绗﹀悎锛�")
+			return nil
+		}
+	} else {
+		return nil
+	}
+}
\ No newline at end of file

--
Gitblit v1.8.0