From c9296a4e0869aa9b4a98df7d0433ef0d6b9a0b10 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期四, 21 十一月 2019 16:11:23 +0800
Subject: [PATCH] fix init

---
 EsApi.go |   55 ++++++++++++++++++++++++++-----------------------------
 1 files changed, 26 insertions(+), 29 deletions(-)

diff --git a/EsApi.go b/EsApi.go
index 6061321..a5651cb 100644
--- a/EsApi.go
+++ b/EsApi.go
@@ -294,34 +294,39 @@
 }
 
 //鍒濆鍖栧疄鏃舵姄鎷�
-func InitRealTimeCapture(serverIp string, serverPort string, indexName string, isAlarm bool, quantity int) ([]protomsg.AIOcean, error) {
+func InitRealTimeCapture(serverIp string, serverPort string, indexName string, isAlarm bool, category string, quantity int) ([]protomsg.AIOcean, error) {
 	var aIOceanInfo []protomsg.AIOcean
 	url := "http://" + serverIp + ":" + serverPort +
 		"/" + indexName + "/_search"
-	queryStr := ""
-	if isAlarm == true {
-		queryStr = `"query":{
-		"match_all":{}
-	},`
-	} else {
-		queryStr = `"query":{
+	categoryStr := ""
+	if category != "" {
+		categoryStr = `				{
+					"term":{
+						"targetInfo.targetType":"` + category + `"
+					}
+				},`
+
+	}
+	queryStr := `"query":{
 		"bool":{
 			"filter":[
+				` + categoryStr + `
 				{
 					"term":{
-						"isAlarm":true
+						"isAlarm":` + strconv.FormatBool(isAlarm) + `
 					}
 				}
 			]
 		}
 	},`
-	}
+
 	DSLJson := `{
 	"size":` + strconv.Itoa(quantity) + `,
 	` + queryStr + `
 	"sort":[{"picDate":{"order":"desc"}}],
 	"_source": {"includes":[],"excludes":["*.feature"]}
 	}`
+	fmt.Println(DSLJson)
 	buf, err := EsReq("POST", url, []byte(DSLJson))
 	if err != nil {
 		return aIOceanInfo, err
@@ -342,18 +347,6 @@
 	var aIOceanInfo []protomsg.AIOcean
 	url := "http://" + serverIp + ":" + serverPort +
 		"/" + indexName + "/_search"
-	queryStr := ""
-	if isAlarm == true {
-		fmt.Println("continue")
-	} else {
-		queryStr = `
-				{
-					"term":{
-						"isAlarm":1
-					}
-				}
-					`
-	}
 	DSLJson := `{
 	"size":20,
 	"query":{
@@ -367,11 +360,15 @@
 						}		
 					}
 				},
-			` + queryStr + `
+				{
+					"term":{
+						"isAlarm":` + strconv.FormatBool(isAlarm) + `
+					}
+				}
 			]
 		}
 	},
-	"_source": ["baseInfo", "alarmRules", "sex", "analyServerName", "sdkName", "ageDescription", "content", "id", "cameraAddr", "picMaxUrl", "picDate", "race", "videoUrl", "picSmUrl", "taskName", "personIsHub", "isAlarm", "analyServerIp", "cameraId", "isAckAlarm"]
+	"_source": {"includes":[],"excludes":["*.feature"]}
 	}`
 
 	buf, err := EsReq("POST", url, []byte(DSLJson))
@@ -391,10 +388,6 @@
 
 //缁煎悎缁熻
 func StatisticsComprehensive(serverIp string, serverPort string, indexName string, isAlarm bool) (total int, err error) {
-	isAlarmStr := ""
-	if isAlarm == true {
-		isAlarmStr = `,{"term":{"isAlarm":true}}`
-	}
 	url := "http://" + serverIp + ":" + serverPort +
 		"/" + indexName + "/_search"
 	DSLJson := `{
@@ -407,8 +400,12 @@
 						"gte":"now+8h/d"
 						}
 					}
+				},
+				{
+					"term":{
+						"isAlarm":` + strconv.FormatBool(isAlarm) + `
+					}
 				}
-				` + isAlarmStr + `
 				]
 			}
 		}

--
Gitblit v1.8.0