From efbf83e4778cb61e7da23c8a4a62b352929b40ea Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期四, 21 十一月 2019 17:56:38 +0800
Subject: [PATCH] fix

---
 EsApi.go |   58 ++++++++++++++++++++++++++++++++++------------------------
 1 files changed, 34 insertions(+), 24 deletions(-)

diff --git a/EsApi.go b/EsApi.go
index a5651cb..f5cd55c 100644
--- a/EsApi.go
+++ b/EsApi.go
@@ -294,28 +294,32 @@
 }
 
 //鍒濆鍖栧疄鏃舵姄鎷�
-func InitRealTimeCapture(serverIp string, serverPort string, indexName string, isAlarm bool, category string, quantity int) ([]protomsg.AIOcean, error) {
+func InitRealTimeCapture(serverIp string, serverPort string, indexName string, isAlarm string, category string, quantity int) ([]protomsg.AIOcean, error) {
 	var aIOceanInfo []protomsg.AIOcean
 	url := "http://" + serverIp + ":" + serverPort +
 		"/" + indexName + "/_search"
-	categoryStr := ""
-	if category != "" {
-		categoryStr = `				{
+	var filterArr []string
+	if isAlarm != "all" {
+		filterArr = append(filterArr,`				{
+					"term":{
+						"isAlarm":"` + isAlarm + `"
+					}
+				}`)
+	}
+
+	if category != "all" {
+		filterArr = append(filterArr,`				{
 					"term":{
 						"targetInfo.targetType":"` + category + `"
 					}
-				},`
+				}`)
 
 	}
-	queryStr := `"query":{
+
+ 	queryStr := `"query":{
 		"bool":{
 			"filter":[
-				` + categoryStr + `
-				{
-					"term":{
-						"isAlarm":` + strconv.FormatBool(isAlarm) + `
-					}
-				}
+				`+strings.Join(filterArr,",")+`
 			]
 		}
 	},`
@@ -387,23 +391,29 @@
 }
 
 //缁煎悎缁熻
-func StatisticsComprehensive(serverIp string, serverPort string, indexName string, isAlarm bool) (total int, err error) {
+func StatisticsComprehensive(serverIp string, serverPort string, indexName string, isAlarm string) (total int, err error) {
 	url := "http://" + serverIp + ":" + serverPort +
 		"/" + indexName + "/_search"
+	isAlarmStr := ""
+	if isAlarm != "all" {
+		isAlarmStr = `				{
+					"term":{
+						"isAlarm":"` + isAlarm + `"
+					}
+				},`
+
+	}
 	DSLJson := `{
 	"size":0,
 	"query":{
 		"bool":{
-			"filter":[{
+			"filter":[
+				` + isAlarmStr + `
+				{
 				"range":{
 					"picDate":{
 						"gte":"now+8h/d"
 						}
-					}
-				},
-				{
-					"term":{
-						"isAlarm":` + strconv.FormatBool(isAlarm) + `
 					}
 				}
 				]
@@ -448,9 +458,9 @@
 		}
 	},
 	"aggs":{
-		"sdkId_status":{
+		"sdkName_status":{
 			"terms":{
-				"field":"taskId"
+				"field":"sdkName.raw"
 			}
 		}
 	}
@@ -469,7 +479,7 @@
 	if !ok {
 		return nil, errors.New("first hits change error!")
 	}
-	sdkName_status, ok := middle["sdkId_status"].(map[string]interface{})
+	sdkName_status, ok := middle["sdkName_status"].(map[string]interface{})
 	if !ok {
 		return nil, errors.New("first hits change error!")
 	}
@@ -481,9 +491,9 @@
 			fmt.Println("change to source error!")
 			continue
 		}
-		sdkId := tmpbuf["key"].(string)
+		sdkName := tmpbuf["key"].(string)
 		count := int(tmpbuf["doc_count"].(float64))
-		source["id"] = sdkId
+		source["name"] = sdkName
 		source["value"] = count
 		sources = append(sources, source)
 	}

--
Gitblit v1.8.0