Merge remote-tracking branch 'origin/master'

sunty

2019-10-30 a1ef2d42d320f075c75e8cef52a879114441c103

 controllers/esSearch.go

@@ -1,18 +1,19 @@
   package controllers
package controllers

   import (
      "basic.com/dbapi.git"
      "fmt"
      "github.com/gin-gonic/gin"
      "strconv"
      "strings"
      "webserver/extend/code"
      "webserver/extend/config"
      "webserver/extend/esutil"
      "webserver/extend/logger"
      "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{}

@@ -42,7 +43,7 @@
}

//获取监控等级
func getAlarmLevel(alarmlevel []string) []string {
func getAlarmLevel(alarmlevel []int32) []string {
   d := dbapi.DicApi{}
   res, data := d.FindByType("ALARMLEVEL")
   if data == nil {
@@ -50,7 +51,7 @@
   }
   logger.Debug(res)
   // logger.Debug(data)
   alarmLevel := make(map[int]string)
   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"])
@@ -60,15 +61,14 @@
      if err != nil {
         logger.Debug(err)
      }
      alarmLevel[vl] = name
      alarmLevel[int32(vl)] = name
   }
   alarmLevelRes := make([]string, len(alarmlevel))
   for i, v := range alarmlevel {
      value,_ := strconv.Atoi(v)
      if value < 0 {
      if v < 0 {
         alarmLevelRes[i] = "撤防"
      } else {
         alarmLevelRes[i] = alarmLevel[value]
         alarmLevelRes[i] = alarmLevel[v]
      }
   }
   return alarmLevelRes
@@ -88,7 +88,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 +101,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 +118,7 @@
      esTaskId := strings.Replace(strings.Trim(fmt.Sprint(taskId), "[]"), " ", "\",\"", -1)
      taskIdStr = "{\"terms\":{\"taskId\":[\"" + esTaskId + "\"]}},"
   }

   //判断摄像机ID
   cameraIdStr := ""
   cameraId := searchBody.TreeNodes
@@ -141,11 +152,15 @@

   //使用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 + "\"}},"

   //请求体
   prama := "{\"from\":\"" + from + "\"," +
@@ -154,6 +169,7 @@
      "\"filter\":[" +
      cameraIdStr +
      alarmLevelStr +
      personIdStr +
      taskIdStr +
      isCollectStr +
      esTableIdStr +
@@ -197,9 +213,14 @@
}

func GetLinkInfo(linkId string, id string) []map[string]interface{} {
   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)) +
      "/" + 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 {