From 05eb392fbf5c8e6b5cfd615e1dc56bf1ff7cdc64 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期四, 19 十二月 2019 14:04:17 +0800
Subject: [PATCH] add targetlocation

---
 EsClient.go |   77 +++++++++++++++++++++++++++++++++++---
 1 files changed, 70 insertions(+), 7 deletions(-)

diff --git a/EsClient.go b/EsClient.go
index 7ea8c54..0d608a9 100644
--- a/EsClient.go
+++ b/EsClient.go
@@ -57,9 +57,10 @@
 	}
 }
 
-func Parsesources(sources []map[string]interface{}) (esinfos []*protomsg.Esinfo) {
+func Parsesources(sources []map[string]interface{}) (multiInfos []*protomsg.MultiFeaCache) {
 	var ok bool
 	for _, source := range sources {
+		var multiInfo protomsg.MultiFeaCache
 		var tmpinfo protomsg.Esinfo
 		tmpinfo.Id, ok = source["id"].(string)
 		if !ok {
@@ -72,14 +73,74 @@
 				if !ok {
 					continue
 				}
+				if m["attachTarget"] != nil {
+					multiInfo.AttachFeature, ok = m["attachTarget"].(map[string]interface{})["feature"].(string)
+				}
+				if m["targetLocation"] != nil {
+					targetLocation := m["targetLocation"].(map[string]interface{})
+					topLeft := targetLocation["topLeft"].(map[string]interface{})
+					tx := topLeft["x"].(float64)
+					ty := topLeft["y"].(float64)
+					bottomRight := targetLocation["bottomRight"].(map[string]interface{})
+					bx := bottomRight["x"].(float64)
+					by := bottomRight["y"].(float64)
+					multiInfo.TargetLocation = &protomsg.TargetLocation{
+						TopLeft: &protomsg.Location{
+							X: tx,
+							Y: ty,
+						},
+						BottomRight: &protomsg.Location{
+							X: bx,
+							Y: by,
+						},
+					}
+				}
 			}
 		}
 		tmpinfo.AnalyServerId, ok = source["analyServerId"].(string)
 		if !ok {
 			continue
 		}
-		esinfos = append(esinfos, &tmpinfo)
-
+		multiInfo.EsInfo = &tmpinfo
+		multiInfo.CameraId, ok = source["cameraId"].(string)
+		//fmt.Println(source["cameraId"])
+		leis := make([]*protomsg.LinkEsInfo, 0)
+		if source["linkTagInfo"] != nil {
+			for _, v := range source["linkTagInfo"].([]interface{}) {
+				var lei protomsg.LinkEsInfo
+				mi := v.(map[string]interface{})
+				lei.CameraId, ok = mi["cameraId"].(string)
+				if mi["targetInfo"] != nil {
+					for _, val := range mi["targetInfo"].([]interface{}) {
+						fmi := val.(map[string]interface{})
+						lei.Feature, ok = fmi["feature"].(string)
+						if fmi["targetLocation"] != nil {
+							targetLocation := fmi["targetLocation"].(map[string]interface{})
+							topLeft := targetLocation["topLeft"].(map[string]interface{})
+							tx := topLeft["x"].(float64)
+							ty := topLeft["y"].(float64)
+							bottomRight := targetLocation["bottomRight"].(map[string]interface{})
+							bx := bottomRight["x"].(float64)
+							by := bottomRight["y"].(float64)
+							multiInfo.TargetLocation = &protomsg.TargetLocation{
+								TopLeft: &protomsg.Location{
+									X: tx,
+									Y: ty,
+								},
+								BottomRight: &protomsg.Location{
+									X: bx,
+									Y: by,
+								},
+							}
+						}
+						lei.AttachFeature, ok = fmi["attachTarget"].(map[string]interface{})["feature"].(string)
+					}
+				}
+				leis = append(leis, &lei)
+			}
+		}
+		multiInfo.LinkEsInfo = leis
+		multiInfos = append(multiInfos, &multiInfo)
 	}
 	return
 }
@@ -110,7 +171,7 @@
 		tmpinfo.SdkName, ok = source["sdkName"].(string)
 
 		if source["picMaxUrl"] != nil {
-			for _,v := range source["picMaxUrl"].([]interface{}) {
+			for _, v := range source["picMaxUrl"].([]interface{}) {
 				tmpinfo.PicMaxUrl = append(tmpinfo.PicMaxUrl, v.(string))
 			}
 		}
@@ -121,6 +182,7 @@
 				m := v.(map[string]interface{})
 				var ti protomsg.TargetInfo
 				ti.TargetId, ok = m["targetId"].(string)
+				ti.TargetType, ok = m["targetType"].(string)
 				ti.TargetScore, ok = m["targetScore"].(float64)
 				ti.Feature, ok = m["feature"].(string)
 				ti.PicSmUrl, ok = m["picSmUrl"].(string)
@@ -157,7 +219,7 @@
 				bi.CompareScore, ok = m["compareScore"].(float64)
 				bi.BwType, ok = m["bwType"].(string)
 				bi.TargetId, ok = m["targetId"].(string)
-				bi.TableName, ok = m["tableName"].(string)
+				bi.TargetName, ok = m["targetName"].(string)
 				bi.TargetPicUrl, ok = m["targetPicUrl"].(string)
 				bi.MonitorLevel, ok = m["monitorLevel"].(string)
 				bi.Labels, ok = m["labels"].(string)
@@ -208,7 +270,7 @@
 				lti.TaskName, ok = lm["taskName"].(string)
 				lti.SdkName, ok = lm["sdkName"].(string)
 				if lm["picMaxUrl"] != nil {
-					for _,v := range lm["picMaxUrl"].([]interface{}) {
+					for _, v := range lm["picMaxUrl"].([]interface{}) {
 						lti.PicMaxUrl = append(lti.PicMaxUrl, v.(string))
 					}
 				}
@@ -219,6 +281,7 @@
 						m := v.(map[string]interface{})
 						var ti protomsg.TargetInfo
 						ti.TargetId, ok = m["targetId"].(string)
+						ti.TargetType, ok = m["targetType"].(string)
 						ti.TargetScore, ok = m["targetScore"].(float64)
 						ti.Feature, ok = m["feature"].(string)
 						ti.PicSmUrl, ok = m["picSmUrl"].(string)
@@ -255,7 +318,7 @@
 						bi.CompareScore, ok = m["compareScore"].(float64)
 						bi.BwType, ok = m["bwType"].(string)
 						bi.TargetId, ok = m["targetId"].(string)
-						bi.TableName, ok = m["tableName"].(string)
+						bi.TargetName, ok = m["targetName"].(string)
 						bi.TargetPicUrl, ok = m["targetPicUrl"].(string)
 						bi.MonitorLevel, ok = m["monitorLevel"].(string)
 						bi.Labels, ok = m["labels"].(string)

--
Gitblit v1.8.0