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