From c7cfafb13d36512f3e73457e94c26e31dedf5f7a Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 31 五月 2019 11:29:12 +0800
Subject: [PATCH] conn close log

---
 esutil/EsClient.go |  127 +++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 123 insertions(+), 4 deletions(-)

diff --git a/esutil/EsClient.go b/esutil/EsClient.go
index fdf50d6..5fdcfa1 100644
--- a/esutil/EsClient.go
+++ b/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,29 @@
 
 				source["sdkType"] = sdkTypeToValue(sdk)
 			}
+
+			pmax, exist := source["picMaxUrl"].(string)
+			//fmt.Println("picMaxUrl: ",pmax)
+			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
+			}
+			//fmt.Println("picSmUrl: ",psm)
+			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 +106,99 @@
 	}
 }
 
+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
+	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绫诲瀷
 func sdkTypeToValue(i int) string {
 	value := []string{"浜鸿劯", "杞﹁締", "浜轰綋", "鍏ヤ镜", "鎷ユ尋", "闈犲彸琛�", "浜哄憳寮傚父", "涓綋闈欐"}
@@ -88,16 +206,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\",\"imgKey\",\"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

--
Gitblit v1.8.0