From f4e8f206a6760bdc31734dfcb1c65916b5b76311 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期四, 22 八月 2019 12:05:58 +0800
Subject: [PATCH] add cluster

---
 controllers/esSearch.go |   55 +++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 51 insertions(+), 4 deletions(-)

diff --git a/controllers/esSearch.go b/controllers/esSearch.go
index 1c20631..c9e5ec6 100644
--- a/controllers/esSearch.go
+++ b/controllers/esSearch.go
@@ -1,4 +1,4 @@
-package controllers
+	package controllers
 
 import (
 	"basic.com/dbapi.git"
@@ -81,9 +81,10 @@
 	queryBody := searchBody["inputValue"].(string)
 	//妫�绱㈡
 	if queryBody != "" {
-		queryStr = "\"must\":[{\"multi_match\":{\"query\":\"" + queryBody + "\",\"fields\":[\"baseInfo.sex\",\"baseInfo.phoneNum.raw^1.5\",\"baseInfo.personName.raw^1.5\",\"alarmRules.alarmLevel^1.5\",\"ageDescription^1.5\",\"taskName^1.5\",\"baseInfo.tableName^1.5\",\"sex^2.0\",\"race^2.0\",\"content^1.0\",\"baseInfo.idCard^1.8\",\"cameraAddr^1.0\"]," +
+		queryStr = "\"must\":[{\"multi_match\":{\"query\":\"" + queryBody + "\",\"fields\":[\"baseInfo.sex\",\"baseInfo.phoneNum.raw^1.5\",\"baseInfo.personName.raw^1.5\",\"alarmRules.alarmLevel^1.5\",\"alarmRules.linkInfo^1.5\",\"ageDescription^1.5\",\"taskName^1.5\",\"baseInfo.tableName^1.5\",\"sex^2.0\",\"race^2.0\",\"content^1.0\",\"baseInfo.idCard^1.8\",\"cameraAddr^1.0\"]," +
 			"\"type\":\"cross_fields\",\"operator\":\"OR\",\"slop\":0,\"prefix_length\":0,\"max_expansions\":50,\"zero_terms_query\":\"NONE\",\"auto_generate_synonyms_phrase_query\":true,\"fuzzy_transpositions\":true,\"boost\":1}}],"
 	}
+
 	searchTime := searchBody["searchTime"].([]interface{})
 	if searchBody["searchTime"] == nil || len(searchTime) < 2 {
 		info := make(map[string]interface{}, 0)
@@ -152,12 +153,58 @@
 		analyServerFilterStr +
 		"{\"range\":{\"picDate\":{\"from\":\"" + gteDate + "\",\"to\":\"" + lteDate + "\",\"include_lower\":true,\"include_upper\":true,\"boost\":1}}}]}}," +
 		"\"sort\":[{\"_score\":{\"order\":\"desc\"}},{\"picDate\":{\"order\":\"desc\"}}]," +
-		"\"_source\":[\"baseInfo\",\"alarmRules\",\"sex\",\"analyServerName\",\"sdkName\",\"ageDescription\",\"content\",\"id\",\"cameraAddr\",\"picMaxUrl\",\"picDate\",\"race\",\"videoUrl\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"isAlarm\",\"analyServerIp\",\"cameraId\"]" +
+		"\"_source\":[\"baseInfo\",\"alarmRules\",\"sex\",\"analyServerName\",\"sdkName\",\"ageDescription\",\"content\",\"id\",\"cameraAddr\",\"picMaxUrl\",\"picDate\",\"race\",\"videoUrl\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"isAlarm\",\"analyServerIp\",\"cameraId\",\"linkId\"]" +
 		"}"
 	//logger.Debug("findEsData.param:",prama)
 	//鏁版嵁瑙f瀽
 	tokenRes := esutil.GetEsDataReq(url, prama, true)
-	return tokenRes
+	//fmt.Println(tokenRes)
+	tmpAllDate := make(map[string]interface{})
+	tmpDate := make(map[string][]interface{})
+	tmpAllDate["total"] = tokenRes["total"]
+	data := tokenRes["datalist"]
+	for _, masterInfoValues := range data.([]interface{}) {
+		masterInfo := masterInfoValues.(map[string]interface{})
+		sources := make(map[string]interface{}, 0)
+		sources["activeObject"] = masterInfo
+		source := make([]map[string]interface{}, 0)
+		source = append(source, masterInfo)
+		linkId := ""
+		if masterInfo["linkId"] != nil {
+			linkId = masterInfo["linkId"].(string)
+		}
+		id := masterInfo["id"].(string)
+		if linkId != "" {
+			linkInfo := GetLinkInfo(linkId, id)
+			for _, slaveInfoValues := range linkInfo {
+				slaveInfo := slaveInfoValues
+				source = append(source, slaveInfo)
+			}
+
+		}
+		sources["list"] = source
+		tmpDate["datalist"] = append(tmpDate["datalist"], sources)
+	}
+	tmpAllDate["datalist"] = tmpDate["datalist"]
+	return tmpAllDate
+}
+
+func GetLinkInfo(linkId string, id string) []map[string]interface{} {
+	url := "http://" + config.EsInfo.Masterip + ":" + config.EsInfo.Httpport +
+		"/" + config.EsInfo.EsIndex.VideoPersons.IndexName + "," + config.EsInfo.EsIndex.Personaction.IndexName + "/_search?search_type=dfs_query_then_fetch"
+	jsonDSL := "{\"query\":{\"bool\":{\"filter\":[{\"term\":{\"linkId.keyword\":\"" + linkId + "\"}}],\"must_not\":[{\"term\":{\"id\":\"" + id + "\"}}]}},\"size\":100}"
+
+	buf, err := esutil.EsReq("POST", url, []byte(jsonDSL))
+	if err != nil {
+		logger.Debug(err)
+	}
+
+	sources, err := esutil.Sourcelist(buf)
+	if err != nil {
+		logger.Debug(err)
+	}
+	return sources
+
 }
 
 //sdkTYype瀛楀吀"\"_source\":[\"BaseName\",\"Gender\",\"Race\",\"content\",\"idcard\",\"picAddress\",\"picDate\",\"sdkType\",\"Age\",\"personId\",\"personIsHub\",\"personPicUrl\",\"picLocalUrl\",\"picSmUrl\",\"videoIp\",\"videoNum\",\"cameraId\",\"ageDescription\",\"likePer\"]" +

--
Gitblit v1.8.0