From d0e6e8c6ef16afbc276fc13dece6239476f8d4e3 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期五, 28 八月 2020 16:46:49 +0800
Subject: [PATCH] add out

---
 EsClient.go |  110 +++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 106 insertions(+), 4 deletions(-)

diff --git a/EsClient.go b/EsClient.go
index 79c7c0d..8e08e76 100644
--- a/EsClient.go
+++ b/EsClient.go
@@ -476,6 +476,112 @@
 	return tmpinfos
 }
 
+func FaceSourceAggregations(buf [] byte, thresholdTime float64, queryUseTime float64) (sources map[string]interface{}, err error) {
+	s := make(map[string]interface{})
+	loc, err := time.LoadLocation("Asia/Shanghai")
+	if err != nil {
+		return nil, errors.New("鏃跺尯璁剧疆閿欒")
+	}
+	var info interface{}
+	json.Unmarshal(buf, &info)
+	out, ok := info.(map[string]interface{})
+	if !ok {
+		return nil, errors.New("http response interface can not change map[string]interface{}")
+	}
+	middle, ok := out["aggregations"].(map[string]interface{})
+	if !ok {
+		return nil, errors.New("first hits change error!")
+	}
+	bucketsAggs := middle["buckets_aggs"].(map[string]interface{})
+	buckets := bucketsAggs["buckets"].([]interface{})
+	if len(buckets) == 0 {
+		return nil, nil
+	}
+	allSource := make([]map[string]interface{}, 0)
+	for _, inf := range buckets {
+		hitsSources := make([]map[string]interface{}, 0)
+		topAttentionHits := inf.(map[string]interface{})["top_attention_hits"].(map[string]interface{})
+		middleHits := topAttentionHits["hits"].(map[string]interface{})
+		finalHits := middleHits["hits"].([]interface{})
+		startTime := ""
+		indexLength := len(finalHits)
+		point := 0
+		for _, in := range finalHits {
+			point = point+1
+			tmpHitSource := make(map[string]interface{})
+			tmpbuf, ok := in.(map[string]interface{})
+			if !ok {
+				fmt.Println("change to source error!")
+				continue
+			}
+			source, ok := tmpbuf["_source"].(map[string]interface{})
+			if !ok {
+				fmt.Println("change _source error!")
+				continue
+			}
+			baseInfo := source["baseInfo"].([]interface{})[0].(map[string]interface{})
+			targetInfo := source["targetInfo"].([]interface{})[0].(map[string]interface{})
+			tmpTime := source["picDate"].(string)
+			mTime, err := time.ParseInLocation("2006-01-02 15:04:05", tmpTime, loc)
+			if err != nil {
+				return nil, errors.New("鏃堕棿瑙f瀽閿欒")
+			}
+
+			sTime := tmpTime
+			eTime := mTime.Add(time.Second*1).Format("2006-01-02 15:04:05")
+			stayTime := 1.0
+			if startTime != "" && point < indexLength{
+				sinTime, _ := time.ParseInLocation("2006-01-02 15:04:05", startTime, loc)
+				passTime := math.Abs(mTime.Sub(sinTime).Seconds())
+				hitsSources[len(hitsSources)-1]["stayTime"] = stayTime
+				//fmt.Println("passTime:   ", passTime)
+				if passTime <= thresholdTime || point == indexLength{
+					startTime = tmpTime
+					hitsSources[len(hitsSources)-1]["endTime"] = tmpTime
+					if point == indexLength{
+						hitStartTime := hitsSources[len(hitsSources)-1]["startTime"].(string)
+						realStartTime, _ := time.ParseInLocation("2006-01-02 15:04:05", hitStartTime, loc)
+						stayTime = math.Abs(mTime.Sub(realStartTime).Seconds())
+						hitsSources[len(hitsSources)-1]["stayTime"] = stayTime
+					}
+					continue
+				} else {
+					hitStartTime := hitsSources[len(hitsSources)-1]["startTime"].(string)
+					hitEndTime := hitsSources[len(hitsSources)-1]["endTime"].(string)
+					realStartTime, _ := time.ParseInLocation("2006-01-02 15:04:05", hitStartTime, loc)
+					realEndTime, _ := time.ParseInLocation("2006-01-02 15:04:05", hitEndTime, loc)
+					stayTime = math.Abs(realEndTime.Sub(realStartTime).Seconds())
+					if sinTime.Sub(mTime).Seconds() == 0 {
+						sinTime.Add(time.Second * 1)
+						sinTime.Format("2006-01-02 15:04:05")
+						hitsSources[len(hitsSources)-1]["endTime"] = tmpTime
+						stayTime =  1
+					}
+					hitsSources[len(hitsSources)-1]["stayTime"] = stayTime
+					startTime = ""
+					continue
+					//fmt.Println(hitsSources[len(hitsSources)-1])
+				}
+			}
+			//fmt.Println("========================================================")
+			startTime = tmpTime
+			tmpHitSource["personId"] = baseInfo["targetId"].(string)
+			tmpHitSource["startTime"] = sTime
+			tmpHitSource["startFacePicUrl"] = targetInfo["picSmUrl"].(string)
+			tmpHitSource["endTime"] = eTime
+			tmpHitSource["stayTime"] = stayTime
+			hitsSources = append(hitsSources, tmpHitSource)
+		}
+		allSource = append(allSource, hitsSources...)
+	}
+	count := len(allSource)
+	//fmt.Println(count)
+	s["count"] = count
+	s["allSource"] = allSource
+	s["queryUseTime"] = queryUseTime
+	return s, nil
+}
+
 func SourceAggregations(buf [] byte, thresholdTime float64, queryUseTime float64) (sources map[string]interface{}, err error) {
 	s := make(map[string]interface{})
 	loc, err := time.LoadLocation("Asia/Shanghai")
@@ -592,10 +698,6 @@
 			hitsSources = append(hitsSources, tmpHitSource)
 		}
 		allSource = append(allSource, hitsSources...)
-		//	tmpSources["groupKey"] = groupKey
-		//	tmpSources["doc_count"] = docCount
-		//	tmpSources["hits_sources"] = hitsSources
-		//	sources = append(sources, tmpSources)
 	}
 	count := len(allSource)
 	//fmt.Println(count)

--
Gitblit v1.8.0