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