merge esclient.go

龙赣华

2019-05-28 93262f2823b1f1e2a780172fe5669d59853fdaf1

 esutil/EsClient.go

@@ -1,6 +1,7 @@
package esutil

import (
   "bytes"
   "encoding/json"
   "errors"
   "fmt"
@@ -14,7 +15,7 @@
   log "github.com/long/test/log"
)

func GetEsDataReq(url string, parama string, isSource bool) (error, map[string]interface{}) {
func GetEsDataReq(url string, parama string, picurl string, isSource bool) (error, map[string]interface{}) {
   //log.Log.Infoln("es 查询请求路径" + url) //  配置信息 获取
   req, err := http.NewRequest("POST", url, strings.NewReader(parama))

@@ -63,6 +64,7 @@
         }

         source["id"] = value.(map[string]interface{})["_id"]

         sdkType := source["sdkType"]
         if sdkType != nil {
            sdk, err := strconv.Atoi(sdkType.(string))
@@ -72,6 +74,28 @@

            source["sdkType"] = sdkTypeToValue(sdk)
         }

         pmax, exist := source["picMaxUrl"].(string)
         if !exist {
            return errors.New("picMaxurl is not string"), nil
         }

         if !strings.HasPrefix(pmax, "http") {
            source["picMaxUrl"] = picurl + pmax

         }

         psm, exist := source["picSmUrl"].(string)
         if !exist {
            return errors.New("picSmUrl is not string"), nil
         }

         if !strings.HasPrefix(psm, "http") {
            source["picSmUrl"] = picurl + psm
         }
         //videoUrl := getVideoUrl(source)
         source["videoNum"] = getVideoUrl(source)

         sources = append(sources, source)
      }
      data["datalist"] = sources
@@ -81,6 +105,87 @@
   }
}

func getVideoUrl(source map[string]interface{}) (videoUrl string) {
   imgKey := source["imgKey"].(string)
   picDte := source["picDate"].(string)        //抓拍日期
   cameraId := source["videoReqNum"].(string)  //摄像机id
   indeviceId := source["indeviceid"].(string) //分析设备id

   var deviceMap map[string]string
   deviceMap["DSVAD010120181119"] = "http://58.118.225.79:41241/ManCarAnaly/perVideoPic/uploadRtnVideo"
   deviceMap["DSVAD010220181119"] = "http://58.118.225.79:41242/ManCarAnaly/perVideoPic/uploadRtnVideo"
   deviceMap["DSVAD010320181119"] = "http://58.118.225.79:41243/ManCarAnaly/perVideoPic/uploadRtnVideo"
   deviceMap["DSVAD010420181119"] = "http://58.118.225.79:41244/ManCarAnaly/perVideoPic/uploadRtnVideo"

   reqUrl := deviceMap[indeviceId]
   var paramMap map[string]interface{}
   paramMap["imgKey"] = imgKey
   paramMap["picDate"] = picDte
   paramMap["videoPath"] = cameraId

   respBytes, err := doPostRequest(reqUrl, "application/x-www-form-urlencoded", paramMap, nil, nil)
   if err != nil {
      return ""
   }
   var resp RespVideo
   err = json.Unmarshal(respBytes, &resp)
   if err != nil {
      return ""
   }
   return resp.VideoPath
}

type RespVideo struct {
   Code      int32  `json:"code"`
   ErrorStr  string `json:"errorStr"`
   Msg       string `json:"msg"`
   Resp      interface{}
   Status    string `json:"status"`
   VideoPath string `json:"videoPath"`
}

func doPostRequest(url string, contentType string, body map[string]interface{}, params map[string]string, headers map[string]string) ([]byte, error) {
   var resultBytes []byte
   var bodyJson []byte
   if body != nil {
      var err error
      bodyJson, err = json.Marshal(body)
      if err != nil {
         return resultBytes, err
      }
   }
   request, err := http.NewRequest("POST", url, bytes.NewBuffer(bodyJson))
   if err != nil {
      return resultBytes, err
   }
   request.Header.Set("Content-type", contentType)
   //add params
   q := request.URL.Query()
   if params != nil {
      for key, val := range params {
         q.Add(key, val)
      }
      request.URL.RawQuery = q.Encode()
   }
   // add headers
   if headers != nil {
      for key, val := range headers {
         request.Header.Add(key, val)
      }
   }
   client := &http.Client{}
   resp, err := client.Do(request)
   if err != nil {
      return resultBytes, err
   }
   defer resp.Body.Close()
   resultBytes, err = ioutil.ReadAll(resp.Body)
   if err != nil {
      return resultBytes, err
   }
   return resultBytes, nil
}

//sdk类型
func sdkTypeToValue(i int) string {
   value := []string{"人脸", "车辆", "人体", "入侵", "拥挤", "靠右行", "人员异常", "个体静止"}
@@ -88,16 +193,17 @@
   return value[i-1]
}

func PostAction(sec int, Eurl string) []byte {
func PostAction(sec int, Eurl string, picurl string) []byte {
   index := "videopersons,personaction"
   url := fmt.Sprintf("%s%s%s", Eurl, index, "/_search")

   seccond := strconv.Itoa(sec)

   prama := "{\"query\":{\"bool\":{\"filter\":[{\"term\":{\"personIsHub\":\"1\"}},{\"range\":{\"picDate\":{\"gte\":\"now+8h-" + seccond + "s\",\"lt\":\"now+8h\"}}}]}},\"size\":\"1000\",\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," +
      "\"_source\":[\"baseInfo\",\"gender\",\"indeviceName\",\"sdkType\",\"ageDescription\",\"content\",\"ID\",\"picAddress\",\"picMaxUrl\",\"picDate\",\"race\",\"videoNum\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"IDCard\",\"videoIp\",\"videoReqNum\"]" +
      "\"_source\":[\"baseInfo\",\"gender\",\"indeviceName\",\"imageKey\",\"sdkType\",\"ageDescription\",\"indeviceid\",\"content\",\"ID\",\"picAddress\",\"picMaxUrl\",\"picDate\",\"race\",\"videoNum\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"IDCard\",\"videoIp\",\"videoReqNum\"]" +
      "}"
   err, tokenRes := GetEsDataReq(url, prama, true)
   err, tokenRes := GetEsDataReq(url, prama, picurl, true)

   if err != nil {
      log.Log.Errorln(err)
      return nil