From 7ce84b3d1e45d35c7c9f73561893039a05c16995 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 20 十二月 2019 19:38:18 +0800
Subject: [PATCH] fix unTargz
---
controllers/esSearch.go | 237 ++++++++++++++++++++++++++++++++++++++++-------------------
1 files changed, 160 insertions(+), 77 deletions(-)
diff --git a/controllers/esSearch.go b/controllers/esSearch.go
index 0a9f139..161e1ac 100644
--- a/controllers/esSearch.go
+++ b/controllers/esSearch.go
@@ -2,19 +2,22 @@
import (
"basic.com/dbapi.git"
+ "basic.com/valib/logger.git"
"fmt"
+ "github.com/gin-gonic/gin"
"strconv"
"strings"
-
- "github.com/gin-gonic/gin"
+ "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
@@ -25,142 +28,222 @@
// @Failure 500 {string} json "{"code":500, msg:"杩斿洖閿欒淇℃伅", success:false}"
// @Router /data/api-v/es/esSearch [POST]
func (sc *EsSearchController) PostEsSearch(c *gin.Context) {
- searchBody := make(map[string]interface{}, 0)
+ //searchBody := make(map[string]interface{}, 0)
+ //c.BindJSON(&searchBody)
+ //data := findEsData(searchBody)
- c.BindJSON(&searchBody)
+ var arg models.EsSearch
+ err := c.BindJSON(&arg)
+ if err !=nil || arg.Page <=0 && arg.Size<=0 {
+ util.ResponseFormat(c,code.RequestParamError,"鍙傛暟鏈夎")
+ return
+ }
+ data := findEsData(&arg)
- data := findEsData(searchBody)
util.ResponseFormat(c, code.Success, data)
}
+
//鑾峰彇鐩戞帶绛夌骇
-func getAlarmLevel(alarmlevel []int) ([]string) {
+func getAlarmLevel(alarmlevel []int32) []string {
d := dbapi.DicApi{}
- res,data := d.FindByType("ALARMLEVEL")
- if data == nil {
- fmt.Println("dsadasd")
- }
- fmt.Println(res)
- // fmt.Println(data)
- alarmLevel := make(map[int]string)
-
+ res, data := d.FindByType("ALARMLEVEL")
+ if data == nil {
+ logger.Debug("data is nil")
+ }
+ logger.Debug(res)
+ // logger.Debug(data)
+ alarmLevel := make(map[int32]string)
tmp := data.(map[string]interface{})
- for _,value := range tmp["ALARMLEVEL"].([]interface{}) {
- // fmt.Println(value.(map[string]interface{})["value"]," ",value.(map[string]interface{})["name"])
- val := value.(map[string]interface{})["value"].(string)
- name := value.(map[string]interface{})["name"].(string)
- vl,err := strconv.Atoi(val)
- if err != nil {
- fmt.Println("dsadas")
- }
- alarmLevel[vl] = name
+ for _, value := range tmp["ALARMLEVEL"].([]interface{}) {
+ // logger.Debug(value.(map[string]interface{})["value"]," ",value.(map[string]interface{})["name"])
+ val := value.(map[string]interface{})["value"].(string)
+ name := value.(map[string]interface{})["name"].(string)
+ vl, err := strconv.Atoi(val)
+ if err != nil {
+ logger.Debug(err)
}
- alarmLevelRes := make([]string,len(alarmlevel))
- for i,v := range alarmlevel {
- alarmLevelRes[i] = alarmLevel[v]
+ alarmLevel[int32(vl)] = name
+ }
+ alarmLevelRes := make([]string, len(alarmlevel))
+ for i, v := range alarmlevel {
+ if v < 0 {
+ alarmLevelRes[i] = "鎾ら槻"
+ } else {
+ alarmLevelRes[i] = alarmLevel[v]
}
- return alarmLevelRes
+ }
+ return alarmLevelRes
}
-func findEsData(searchBody map[string]interface{}) map[string]interface{} {
- webPage := int(searchBody["page"].(float64))
+func findEsData(searchBody *models.EsSearch) map[string]interface{} {
+ webPage := searchBody.Page
- webSize := int(searchBody["size"].(float64))
+ webSize := searchBody.Size
from := strconv.Itoa((webPage - 1) * webSize)
//esFrom := strconv.Itoa(from)
//esSize := strconv.Itoa(webSize)
- size := strconv.Itoa(webPage * webSize)
+ size := strconv.Itoa(webSize)
//璇锋眰绱㈠紩
- index := config.EsInfo.EsIndex.VideoPersons.IndexName + "," + config.EsInfo.EsIndex.Personaction.IndexName
+ index := config.EsInfo.EsIndex.AiOcean.IndexName
queryStr := ""
- queryBody := searchBody["inputValue"].(string)
+ queryBody := searchBody.InputValue
//妫�绱㈡
if queryBody != "" {
- queryStr = "\"must\":[{\"multi_match\":{\"query\":\"" + queryBody + "\",\"fields\":[\"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\":[\"cameraAddr^1.5\",\"taskName^1.5\",\"sdkName^1.5\",\"showLabels^3.0\",\"baseInfo.tableName^1.5\",\"baseInfo.targetName^1.5\",\"baseInfo.labels^1.5\",\"alarmRules.alarmLevel^1.5\",\"linkTag^1.5\",\"linkTagInfo.cameraAddr\",\"linkTagInfo.taskName^1.5\",\"linkTagInfo.sdkName^1.5\",\"linkTagInfo.showLabels^1.5\",\"linkTagInfo.baseInfo.tableName^1.5\",\"linkTagInfo.baseInfo.targetName^1.5\",\"linkTagInfo.baseInfo.labels^1.5\",\"linkTagInfo.alarmRules.alarmLevel^1.5\"]," +
"\"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}}],"
}
- gteDate := searchBody["searchTime"].([]interface{})[0].(string)
- lteDate := searchBody["searchTime"].([]interface{})[1].(string)
+
+ searchTime := searchBody.SearchTime
+ if searchTime == nil || len(searchTime) < 2 {
+ info := make(map[string]interface{}, 0)
+ info["err"] = "璇疯緭鍏ユ椂闂磋寖鍥�"
+ return info
+ }
+
+ gteDate := searchTime[0]
+ lteDate := searchTime[1]
+
+ //鍒ゆ柇鏁版嵁ID
+ idStr := ""
+ linkTagInfoIdStr :=""
+ //personId := searchBody.Id
+ id := []string{}
+ if id != nil && len(id) > 0 {
+ esId := strings.Replace(strings.Trim(fmt.Sprint(id), "[]"), " ", "\",\"", -1)
+ idStr = "{\"terms\":{\"id\":[\"" + esId + "\"]}},"
+ linkTagInfoIdStr = "{\"terms\":{\"linkTagInfo.id\":[\"" + esId + "\"]}},"
+ }
+
//鍒ゆ柇浠诲姟ID
taskIdStr := ""
- taskId := searchBody["tasks"].([]interface{})
+ linkTagInfoTaskIdStr := ""
+ taskId := searchBody.Tasks
if taskId != nil && len(taskId) > 0 {
esTaskId := strings.Replace(strings.Trim(fmt.Sprint(taskId), "[]"), " ", "\",\"", -1)
taskIdStr = "{\"terms\":{\"taskId\":[\"" + esTaskId + "\"]}},"
+ linkTagInfoTaskIdStr = "{\"terms\":{\"linkTagInfo.taskId\":[\"" + esTaskId + "\"]}},"
}
+
//鍒ゆ柇鎽勫儚鏈篒D
cameraIdStr := ""
- cameraId := searchBody["treeNodes"].([]interface{})
+ linkTagInfoCameraIdStr := ""
+ cameraId := searchBody.TreeNodes
if cameraId != nil && len(cameraId) > 0 {
esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1)
cameraIdStr = "{\"terms\":{\"cameraId\":[\"" + esCameraId + "\"]}},"
+ linkTagInfoCameraIdStr = "{\"terms\":{\"linkTagInfo.cameraId\":[\"" + esCameraId + "\"]}},"
}
//鍒ゆ柇搴撹〃ID
- tableId := searchBody["tabs"].([]interface{})
+ tableId := searchBody.Tabs
esTableId := ""
esTableIdStr := ""
+ linkTagInfoEsTableIdStr := ""
if tableId != nil && len(tableId) > 0 {
esTableId = strings.Replace(strings.Trim(fmt.Sprint(tableId), "[]"), " ", "\",\"", -1)
- index = config.EsInfo.EsIndex.VideoPersons.IndexName
esTableIdStr = "{\"terms\":{\"baseInfo.tableId\":[\"" + esTableId + "\"]}},"
+ linkTagInfoEsTableIdStr = "{\"terms\":{\"linkTagInfo.baseInfo.tableId\":[\"" + esTableId + "\"]}},"
}
//鍒ゆ柇鏀惰棌鐘舵��
isCollectStr := ""
- isCollect := searchBody["collection"].(string)
+ linkTagInfoIsCollectStr := ""
+ isCollect := searchBody.Collection
if isCollect != "" {
- isCollectStr = "{\"term\":{\"isCollect\":\"" + isCollect + "\"}},"
+ isCollectStr = "{\"term\":{\"isCollect\":" + isCollect + "}},"
+ linkTagInfoIsCollectStr = "{\"term\":{\"linkTagInfo.isCollect\":" + isCollect + "}},"
}
//鍒ゆ柇甯冮槻绛夌骇id
- // alarmLevelId :=searchBody["alarmlevel"].([]int)
- // alarmLevelStr := ""
- // defenceStateStr := ""
- // if len(alarmLevelId) > 0 {
- // flag := false
- // for i,v := range alarmLevelId {
- // if v == -1 {
- // alarmLevelId = append(alarmLevelId[:i], alarmLevelId[i+1:]...)
- // defenceStateStr = "{\"term\":{\"alarmRules.defenceState\":false}},"
- // flag = true
- // break
- // }
- // }
- // if len(alarmLevelId) > 0{
- // if flag == false {
- // defenceStateStr = "{\"term\":{\"alarmRules.defenceState\":true}},"
- // }
- // id := strings.Replace(strings.Trim(fmt.Sprint(getAlarmLevel(alarmLevelId)), "[]"), " ", "\",\"", -1)
- // alarmLevelStr = "{\"terms\":{\"alarmRules.alarmLevel\":[\"" + id + "\"]}},"
- // }
- // }
-
+ alarmLevelId := searchBody.AlarmLevel
+ alarmLevelStr := ""
+ linkTagInfoAlarmLevelStr := ""
+ if len(alarmLevelId) > 0 {
+ alarmLevelTypes := strings.Replace(strings.Trim(fmt.Sprint(getAlarmLevel(alarmLevelId)), "[]"), " ", "\",\"", -1)
+ alarmLevelStr = "{\"terms\":{\"alarmRules.alarmLevel.raw\":[\"" + alarmLevelTypes + "\"]}},"
+ linkTagInfoAlarmLevelStr = "{\"terms\":{\"linkTagInfo.alarmRules.alarmLevel.raw\":[\"" + alarmLevelTypes + "\"]}},"
+ }
+
//浣跨敤es搴曞眰鏈哄埗澶勭悊鍒嗛〉
//璇锋眰澶�
- url := "http://" + config.EsInfo.Masterip + ":" + config.EsInfo.Httpport +
+ localConf, err2 := cache.GetServerInfo()
+ 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 + "\"}},"
+ linkTagInfoAnalyServerFilterStr := "{\"term\":{\"linkTagInfo.analyServerId\":\"" + localConf.ServerId + "\"}},"
//璇锋眰浣�
prama := "{\"from\":\"" + from + "\"," +
"\"size\":\"" + size + "\"," +
"\"query\":{\"bool\":{" + queryStr +
- "\"filter\":[" +
+ "\"should\":[" +
+ "{\"bool\":{\"filter\":[" +
cameraIdStr +
- // alarmLevelStr +
- // defenceStateStr +
+ alarmLevelStr +
+ idStr +
taskIdStr +
isCollectStr +
esTableIdStr +
analyServerFilterStr +
- "{\"range\":{\"picDate\":{\"from\":\"" + gteDate + "\",\"to\":\"" + lteDate + "\",\"include_lower\":true,\"include_upper\":true,\"boost\":1}}}]}}," +
+ "{\"range\":{\"picDate\":{\"from\":\"" + gteDate + "\",\"to\":\"" + lteDate + "\",\"include_lower\":true,\"include_upper\":true,\"boost\":1}}}"+
+ "]}}," +
+ "{\"bool\":{\"filter\":[" +
+ linkTagInfoCameraIdStr +
+ linkTagInfoAlarmLevelStr +
+ linkTagInfoIdStr +
+ linkTagInfoTaskIdStr +
+ linkTagInfoIsCollectStr +
+ linkTagInfoEsTableIdStr +
+ linkTagInfoAnalyServerFilterStr +
+ "{\"range\":{\"linkTagInfo.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\":{\"includes\":[],\"excludes\":[\"*.feature\"]}" +
"}"
- fmt.Println(prama)
+ logger.Debug("findEsData.param:",prama)
//鏁版嵁瑙ｆ瀽
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)
+
+ if masterInfo["linkTagInfo"] != nil {
+ linkTagInfo := masterInfo["linkTagInfo"].([]interface{})
+ delete(masterInfo, "linkTagInfo")
+ if len(linkTagInfo) >0 {
+ slaveList := make([]interface{}, 0)
+ slaveList = append(slaveList, masterInfo)
+ slaveList = append(slaveList, linkTagInfo...)
+ sources["list"] = slaveList
+ } else {
+ sources["list"] = []interface{}{
+ masterInfo,
+ }
+ }
+ } else {
+ if _,ok := masterInfo["linkTagInfo"];ok {
+ delete(masterInfo, "linkTagInfo")
+ }
+ sources["list"] = []interface{}{
+ masterInfo,
+ }
+ }
+
+ sources["activeObject"] = masterInfo
+
+ tmpDate["datalist"] = append(tmpDate["datalist"], sources)
+ }
+ tmpAllDate["datalist"] = tmpDate["datalist"]
+ return tmpAllDate
}
//sdkTYype瀛楀吀"\"_source\":[\"BaseName\",\"Gender\",\"Race\",\"content\",\"idcard\",\"picAddress\",\"picDate\",\"sdkType\",\"Age\",\"personId\",\"personIsHub\",\"personPicUrl\",\"picLocalUrl\",\"picSmUrl\",\"videoIp\",\"videoNum\",\"cameraId\",\"ageDescription\",\"likePer\"]" +
@@ -178,7 +261,7 @@
tokenRes := esutil.GetEsDataReq(url, prama, true)
personRes := make(map[string]string)
- //fmt.Println(tokenRes["datalist"].([]interface{}))
+ //logger.Debug(tokenRes["datalist"].([]interface{}))
result := tokenRes["datalist"].([]interface{})
for _, value := range result {
userMap := value.(map[string]interface{})
@@ -200,7 +283,7 @@
// personRes["phoneNum"] = value.(map[string]interface{})["phoneNum"].(string)
// personRes["sex"] = value.(map[string]interface{})["sex"].(string)
}
- //fmt.Println(personRes)
+ //logger.Debug(personRes)
if len(personRes) < 1 {
personRes["personName"] = ""
personRes["phoneNum"] = ""
--
Gitblit v1.8.0