From 712ef81c6b8813a498a66cf61ffcd37e5c502c8c Mon Sep 17 00:00:00 2001 From: sunty <1172534965@qq.com> Date: 星期五, 01 十一月 2019 13:45:30 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- controllers/esSearch.go | 56 ++++++++++++++++++++++++++++++++++---------------------- 1 files changed, 34 insertions(+), 22 deletions(-) diff --git a/controllers/esSearch.go b/controllers/esSearch.go index 33a6bca..bd08e48 100644 --- a/controllers/esSearch.go +++ b/controllers/esSearch.go @@ -1,22 +1,23 @@ - package controllers +package controllers - import ( - "basic.com/dbapi.git" - "fmt" - "github.com/gin-gonic/gin" - "strconv" - "strings" - "webserver/cache" - "webserver/extend/code" - "webserver/extend/config" - "webserver/extend/esutil" - "basic.com/valib/logger.git" - "webserver/extend/util" - "webserver/models" - ) +import ( + "basic.com/dbapi.git" + "basic.com/valib/logger.git" + "fmt" + "github.com/gin-gonic/gin" + "strconv" + "strings" + "webserver/cache" + "webserver/extend/code" + "webserver/extend/config" + "webserver/extend/esutil" + "webserver/extend/util" + "webserver/models" +) type EsSearchController struct{} +// @Security ApiKeyAuth // @Summary 妫�绱� // @Description 淇℃伅妫�绱㈠拰姣斿妫�绱� // @Accept json @@ -88,7 +89,7 @@ queryBody := searchBody.InputValue //妫�绱㈡ 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\",\"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\"]," + + 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^3.0\",\"ageDescription^1.5\",\"taskName^1.5\",\"baseInfo.tableName^1.0\",\"sex^2.0\",\"race^2.0\",\"content^1.0\",\"baseInfo.idCard^1.8\",\"cameraAddr^2.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}}]," } @@ -101,6 +102,16 @@ gteDate := searchTime[0] lteDate := searchTime[1] + + //鍒ゆ柇浜哄憳ID + personIdStr := "" + //personId := searchBody.Id + personId := []string{} + if personId != nil && len(personId) > 0 { + esPersonId := strings.Replace(strings.Trim(fmt.Sprint(personId), "[]"), " ", "\",\"", -1) + personIdStr = "{\"terms\":{\"id\":[\"" + esPersonId + "\"]}}," + } + //鍒ゆ柇浠诲姟ID taskIdStr := "" taskId := searchBody.Tasks @@ -108,6 +119,7 @@ esTaskId := strings.Replace(strings.Trim(fmt.Sprint(taskId), "[]"), " ", "\",\"", -1) taskIdStr = "{\"terms\":{\"taskId\":[\"" + esTaskId + "\"]}}," } + //鍒ゆ柇鎽勫儚鏈篒D cameraIdStr := "" cameraId := searchBody.TreeNodes @@ -142,15 +154,14 @@ //浣跨敤es搴曞眰鏈哄埗澶勭悊鍒嗛〉 //璇锋眰澶� localConf, err2 := cache.GetServerInfo() - if err2 !=nil || localConf.AlarmIp == "" { + if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" { logger.Debug("localConfig is wrong!!!") return nil } url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) + "/" + index + "/_search?search_type=dfs_query_then_fetch" - var setApi dbapi.SysSetApi - _, sysconf := setApi.GetServerInfo() - analyServerFilterStr := "{\"term\":{\"analyServerId\":\"" + sysconf.ServerId + "\"}}," + + analyServerFilterStr := "{\"term\":{\"analyServerId\":\"" + localConf.ServerId + "\"}}," //璇锋眰浣� prama := "{\"from\":\"" + from + "\"," + @@ -159,6 +170,7 @@ "\"filter\":[" + cameraIdStr + alarmLevelStr + + personIdStr + taskIdStr + isCollectStr + esTableIdStr + @@ -203,13 +215,13 @@ func GetLinkInfo(linkId string, id string) []map[string]interface{} { localConf, err2 := cache.GetServerInfo() - if err2 !=nil || localConf.AlarmIp == "" { + if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" { logger.Debug("localConfig is wrong!!!") return nil } url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) + "/" + 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}" + jsonDSL := "{\"query\":{\"bool\":{\"filter\":[{\"term\":{\"linkId\":\"" + linkId + "\"}}],\"must_not\":[{\"term\":{\"id\":\"" + id + "\"}}]}},\"size\":100}" buf, err := esutil.EsReq("POST", url, []byte(jsonDSL)) if err != nil { -- Gitblit v1.8.0