From 93262f2823b1f1e2a780172fe5669d59853fdaf1 Mon Sep 17 00:00:00 2001
From: 龙赣华 <slongertian@gmail.com>
Date: 星期二, 28 五月 2019 12:01:42 +0800
Subject: [PATCH] merge esclient.go

---
 esutil/EsClient.go |   85 ++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 85 insertions(+), 0 deletions(-)

diff --git a/esutil/EsClient.go b/esutil/EsClient.go
index fa64d44..ef5c658 100644
--- a/esutil/EsClient.go
+++ b/esutil/EsClient.go
@@ -1,6 +1,7 @@
 package esutil
 
 import (
+	"bytes"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -92,6 +93,8 @@
 			if !strings.HasPrefix(psm, "http") {
 				source["picSmUrl"] = picurl + psm
 			}
+			//videoUrl := getVideoUrl(source)
+			source["videoNum"] = getVideoUrl(source)
 
 			sources = append(sources, source)
 		}
@@ -100,6 +103,87 @@
 	} else {
 		return nil, dat
 	}
+}
+
+func getVideoUrl(source map[string]interface{}) (videoUrl string) {
+	imgKey := source["imgKey"].(string)
+	picDte := source["picDate"].(string)        //鎶撴媿鏃ユ湡
+	cameraId := source["videoReqNum"].(string)  //鎽勫儚鏈篿d
+	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绫诲瀷
@@ -119,6 +203,7 @@
 		"\"_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, picurl, true)
+
 	if err != nil {
 		log.Log.Errorln(err)
 		return nil

--
Gitblit v1.8.0