package controllers
|
|
import (
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/bhomedbapi.git"
|
"basic.com/valib/logger.git"
|
"fmt"
|
"strconv"
|
"strings"
|
"vamicro/config"
|
"vamicro/search-service/esutil"
|
"vamicro/search-service/models"
|
)
|
|
// @Security ApiKeyAuth
|
// @Summary 检索
|
// @Description 信息检索和比对检索
|
// @Accept json
|
// @Produce json
|
// @Tags es
|
// @Param obj body models.EsSearch true "底库数据"
|
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"返回错误信息", success:false}"
|
// @Router /data/api-v/es/esSearch [POST]
|
func (sc *EsSearchController) PostEsSearchAggs(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var arg models.EsSearch
|
err := c.BindJSON(&arg)
|
if err != nil || arg.Page <= 0 && arg.Size <= 0 {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
data := findEsDataAggs(&arg)
|
|
return &bhomeclient.Reply{Success: true, Data: data}
|
}
|
|
//// 获取监控等级
|
//func getAlarmLevel(alarmlevel []int32) []string {
|
// d := bhomedbapi.DicApi{}
|
// 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{}) {
|
// // 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)
|
// }
|
// 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
|
//}
|
|
func findEsDataAggs(searchBody *models.EsSearch) map[string]interface{} {
|
//webPage := searchBody.Page
|
|
//webSize := searchBody.Size
|
//from := strconv.Itoa((webPage - 1) * webSize)
|
//esFrom := strconv.Itoa(from)
|
//esSize := strconv.Itoa(webSize)
|
//size := strconv.Itoa(webSize)
|
//请求索引
|
index := config.EsInfo.EsIndex.AiOcean.IndexName
|
queryStr := ""
|
queryBody := searchBody.InputValue
|
//检索框
|
if queryBody != "" {
|
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}}],"
|
}
|
|
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 := ""
|
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 + "\"]}},"
|
}
|
//判断数据来源
|
dataSourceStr := ""
|
dataSource := searchBody.DataSource
|
if dataSource != "" {
|
dataSourceStr = "{\"term\":{\"dataSource\":\"" + dataSource + "\"}},"
|
}
|
//判断摄像机ID
|
cameraIdStr := ""
|
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
|
esTableId := ""
|
esTableIdStr := ""
|
linkTagInfoEsTableIdStr := ""
|
if tableId != nil && len(tableId) > 0 {
|
esTableId = strings.Replace(strings.Trim(fmt.Sprint(tableId), "[]"), " ", "\",\"", -1)
|
esTableIdStr = "{\"terms\":{\"baseInfo.tableId\":[\"" + esTableId + "\"]}},"
|
linkTagInfoEsTableIdStr = "{\"terms\":{\"linkTagInfo.baseInfo.tableId\":[\"" + esTableId + "\"]}},"
|
}
|
//判断收藏状态
|
isCollectStr := ""
|
linkTagInfoIsCollectStr := ""
|
isCollect := searchBody.Collection
|
if isCollect != "" {
|
if isCollect == "1" {
|
isCollectStr = "{\"term\":{\"isCollect\":true}},"
|
linkTagInfoIsCollectStr = "{\"term\":{\"linkTagInfo.isCollect\":true}},"
|
} else if isCollect == "0" {
|
isCollectStr = "{\"term\":{\"isCollect\":false}},"
|
linkTagInfoIsCollectStr = "{\"term\":{\"linkTagInfo.isCollect\":false}},"
|
}
|
}
|
//判断布防等级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底层机制处理分页
|
//请求头
|
serverId := config.Server.AnalyServerId
|
var sApi bhomedbapi.SysSetApi
|
flag, localConf := sApi.GetServerInfo()
|
if !flag || localConf.AlarmIp == "" || localConf.AlarmPort == 0 {
|
logger.Debug("localConf err")
|
return nil
|
}
|
url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) +
|
"/" + index + "/_search?search_type=dfs_query_then_fetch"
|
|
analyServerFilterStr := ""
|
if !searchBody.IsAll {
|
analyServerFilterStr = "{\"term\":{\"analyServerId\":\"" + serverId + "\"}},"
|
}
|
linkTagInfoAnalyServerFilterStr := "{\"term\":{\"linkTagInfo.analyServerId\":\"" + serverId + "\"}},"
|
|
//请求体
|
prama := "{\"size\": 0," +
|
"\"query\":{\"bool\":{" + queryStr +
|
"\"should\":[" +
|
"{\"bool\":{\"filter\":[" +
|
dataSourceStr +
|
cameraIdStr +
|
alarmLevelStr +
|
idStr +
|
taskIdStr +
|
isCollectStr +
|
esTableIdStr +
|
analyServerFilterStr +
|
"{\"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}}}" +
|
"]}}" +
|
"],\"minimum_should_match\":1}}," +
|
"\"sort\":[{\"_score\":{\"order\":\"desc\"}},{\"updateTime\":{\"order\":\"desc\"}}]," +
|
"\"_source\":{\"includes\":[],\"excludes\":[\"*.feature\"]}," +
|
"\"aggs\": {\"group_by_taskId\": {\"terms\": {\"field\": \"taskId\",\"size\": 1000000},\"aggs\": {\"task_info\": {\"top_hits\": {\"size\": 1,\"_source\": [\"taskName\"]}}}}}" +
|
"}"
|
logger.Debug("url: ", url)
|
logger.Debug("findEsData.param:", prama)
|
//数据解析
|
tokenResBuf, errEsReq := esutil.EsReq("POST", url, []byte(prama))
|
if errEsReq != nil {
|
return nil
|
}
|
tmpAllData, errSourceAggregationList := esutil.SourceAggregationList(tokenResBuf)
|
if errSourceAggregationList != nil {
|
return nil
|
}
|
dataAggs := make(map[string]interface{})
|
dataAggs["buckets"] = tmpAllData
|
return dataAggs
|
}
|
