tcp server 用于给andriod 客户端定时发送消息
liuxiaolong
2019-05-31 c7cfafb13d36512f3e73457e94c26e31dedf5f7a
esutil/EsClient.go
@@ -1,6 +1,7 @@
package esutil
import (
   "bytes"
   "encoding/json"
   "errors"
   "fmt"
@@ -75,6 +76,7 @@
         }
         pmax, exist := source["picMaxUrl"].(string)
         //fmt.Println("picMaxUrl: ",pmax)
         if !exist {
            return errors.New("picMaxurl is not string"), nil
         }
@@ -88,10 +90,12 @@
         if !exist {
            return errors.New("picSmUrl is not string"), nil
         }
         //fmt.Println("picSmUrl: ",psm)
         if !strings.HasPrefix(psm, "http") {
            source["picSmUrl"] = picurl + psm
         }
         //videoUrl := getVideoUrl(source)
         source["videoNum"] = getVideoUrl(source)
         sources = append(sources, source)
      }
@@ -100,6 +104,99 @@
   } else {
      return nil, dat
   }
}
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
   deviceMap := make(map[string]string,0)
   deviceMap["DSVAD010120181119"] = "http://172.17.50.241:11111/getRecordVideoPath"
   deviceMap["DSVAD010220181119"] = "http://172.17.50.242:11111/getRecordVideoPath"
   deviceMap["DSVAD010320181119"] = "http://172.17.50.243:11111/getRecordVideoPath"
   deviceMap["DSVAD010420181119"] = "http://172.17.50.244:11111/getRecordVideoPath"
   ngxMap := make(map[string]string,0)
   ngxMap["DSVAD010120181119"] = "http://58.118.225.79:44180/videosource"
   ngxMap["DSVAD010220181119"] = "http://58.118.225.79:44280/videosource"
   ngxMap["DSVAD010320181119"] = "http://58.118.225.79:44380/videosource"
   ngxMap["DSVAD010420181119"] = "http://58.118.225.79:44480/videosource"
   reqUrl := deviceMap[indeviceId]
   paramMap := make(map[string]interface{},0)
   paramMap["imgKey"] = imgKey
   paramMap["picDate"] = picDte
   paramMap["videoNum"] = cameraId
   respBytes, err := doPostRequest(reqUrl, "application/json", paramMap, nil, nil)
   if err !=nil{
      return ""
   }
   var resp RespVideo
   err = json.Unmarshal(respBytes, &resp)
   if err !=nil {
      return ""
   }
   filePath := resp.FilePath
   if !strings.Contains(filePath, "/cut"){
      return ""
   } else {
      strArr := strings.Split(filePath, "/cut")
      ngxUrl := ngxMap[indeviceId]
      if ngxUrl !="" && len(strArr) >0 {
         return ngxUrl + strArr[1]
      }
   }
   return ""
}
type RespVideo struct{
   FilePath string `json:"file_path"`
}
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类型
@@ -116,9 +213,10 @@
   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\",\"imageKey\",\"sdkType\",\"ageDescription\",\"indeviceid\",\"content\",\"ID\",\"picAddress\",\"picMaxUrl\",\"picDate\",\"race\",\"videoNum\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"IDCard\",\"videoIp\",\"videoReqNum\"]" +
      "\"_source\":[\"baseInfo\",\"gender\",\"indeviceName\",\"imgKey\",\"sdkType\",\"ageDescription\",\"indeviceid\",\"content\",\"ID\",\"picAddress\",\"picMaxUrl\",\"picDate\",\"race\",\"videoNum\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"IDCard\",\"videoIp\",\"videoReqNum\"]" +
      "}"
   err, tokenRes := GetEsDataReq(url, prama, picurl, true)
   if err != nil {
      log.Log.Errorln(err)
      return nil