From 22955cd5a948ace0f497c9d618914607f86dba5c Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期四, 06 六月 2019 15:41:19 +0800 Subject: [PATCH] add hub flag --- esutil/EsClient.go | 77 ++++++++++++++++++++++++++------------ 1 files changed, 52 insertions(+), 25 deletions(-) diff --git a/esutil/EsClient.go b/esutil/EsClient.go index 539d072..6b7c6d3 100644 --- a/esutil/EsClient.go +++ b/esutil/EsClient.go @@ -99,13 +99,18 @@ if exist { source["race"] = prace } - fmt.Println("race: ", source["race"]) + + pGender, exist := source["Gender"] + if exist { + source["gender"] = pGender + } + source["ageDescription"] = getAgeDesc(source["Age"]) source["videoNum"] = getVideoUrl(source) - timeFormat := "2019-05-28 15:04:05" - picDateStr, err := time.Parse(timeFormat, source["picDate"].(string)) - fmt.Println("picDate: ", picDateStr) + picDate := source["picDate"].(string) + lastIdx := strings.LastIndex(picDate,":") + picDateStr := picDate[:lastIdx] if err == nil { source["picDate"] = picDateStr } @@ -124,7 +129,7 @@ func getAgeDesc(age interface{})(ageDesc string) { if age !=nil { - ageInt := age.(int32) + ageInt := age.(float64) if ageInt >0 && ageInt<7 { ageDesc = "绔ュ勾" } else if ageInt >=7 && ageInt<18 { @@ -139,7 +144,6 @@ ageDesc = "" } } - fmt.Println("ageDescription: ", ageDesc) return ageDesc } @@ -159,7 +163,7 @@ Content string `json:"content"` } -func getSourceBaseInfo(source map[string]interface{}) (baseInfoJson string) { +func getSourceBaseInfo(source map[string]interface{}) []BaseInfo { sdkType := source["sdkType"].(string) baseInfoArr := make([]BaseInfo,0) if sdkType == "浜鸿劯" { @@ -202,18 +206,17 @@ } baseInfoArr = append(baseInfoArr, baseInfo) - bytes, err := json.Marshal(baseInfoArr) - if err !=nil { - return "" - } - return string(bytes) + //bytes, err := json.Marshal(baseInfoArr) + //if err !=nil { + // return "" + //} } - return "" + return baseInfoArr } func getVideoUrl(source map[string]interface{}) (videoUrl string){ imgKey := source["imgKey"].(string) - picDte := source["picDate"].(string)//鎶撴媿鏃ユ湡 + picDate := source["picDate"].(string)//鎶撴媿鏃ユ湡 cameraId := source["videoReqNum"].(string)//鎽勫儚鏈篿d indeviceId := source["indeviceid"].(string)//鍒嗘瀽璁惧id deviceMap := make(map[string]string,0) @@ -232,31 +235,41 @@ reqUrl := deviceMap[indeviceId] paramMap := make(map[string]interface{},0) paramMap["imgKey"] = imgKey - paramMap["picDate"] = picDte + paramMap["picDate"] = picDate 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 + filePath := resp.FilePath + videoUrl = "" if !strings.Contains(filePath, "/cut"){ - return "" + videoUrl = "" + fmt.Printf("videoReqUrl:%s ,imgKey:%s ,picDate:%s ,cameraId:%s ,filePath:%s \n ",reqUrl,imgKey,picDate,cameraId,filePath) } else { strArr := strings.Split(filePath, "/cut") ngxUrl := ngxMap[indeviceId] if ngxUrl !="" && len(strArr) >0 { - return ngxUrl + strArr[1] + videoUrl = ngxUrl + strArr[1] } } - - return "" + if videoUrl == ""{ + respMap := make(map[string]interface{},0) + err := json.Unmarshal(respBytes, &respMap) + if err !=nil { + fmt.Println("resp UnmarshalToMap err: ",err) + } + fmt.Println("resp: ",respMap) + } + return videoUrl } type RespVideo struct{ @@ -269,11 +282,13 @@ var err error bodyJson, err = json.Marshal(body) if err != nil { + log.Log.Errorln("doPostRequestMarshal err:",err) return resultBytes, err } } request, err := http.NewRequest("POST", url, bytes.NewBuffer(bodyJson)) if err != nil { + log.Log.Errorln("NewRequest ERR:",err) return resultBytes, err } request.Header.Set("Content-type", contentType) @@ -291,14 +306,19 @@ request.Header.Add(key, val) } } - client := &http.Client{} + timeOut:= time.Duration(8*time.Second)//set request timeout + client := &http.Client{ + Timeout:timeOut, + } resp, err := client.Do(request) if err != nil { + log.Log.Errorln("DoRequest ERR:",err) return resultBytes, err } defer resp.Body.Close() resultBytes, err = ioutil.ReadAll(resp.Body) if err != nil { + log.Log.Errorln("ReadAll ERR:",err) return resultBytes, err } return resultBytes, nil @@ -311,14 +331,21 @@ return value[i-1] } -func PostAction(sec int, Eurl string, picurl string) []byte { +func PostAction(sec int, Eurl string, picurl string, ishub string) []byte { index := "videopersons,personaction" url := fmt.Sprintf("%s%s%s", Eurl, index, "/_search") seccond := strconv.Itoa(sec) - - prama := "{\"query\":{\"bool\":{\"filter\":[{\"range\":{\"picDate\":{\"gte\":\"now+8h-" + seccond + "s\",\"lt\":\"now+8h\"}}}]}},\"size\":\"1000\",\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," + - "\"_source\":[\"baseInfo\",\"Gender\",\"personId\",\"personPicUrl\",\"indeviceName\",\"imgKey\",\"sdkType\",\"ageDescription\",\"indeviceid\",\"content\",\"Id\",\"picAddress\",\"picMaxUrl\",\"picDate\",\"Race\",\"videoNum\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"idcard\",\"videoIp\",\"videoReqNum\"]" + + preSec := "" + if sec > 60 { + preSec = "-60s" + } + var ishubReq ="" + if ishub == "hub" { + ishubReq = "{\"term\":{\"personIsHub\":\"1\"}}," + } + prama := "{\"query\":{\"bool\":{\"filter\":["+ishubReq+"{\"range\":{\"picDate\":{\"gte\":\"now+8h-" + seccond + "s\",\"lt\":\"now+8h"+preSec+"\"}}}]}},\"size\":\"1000\",\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," + + "\"_source\":[\"baseInfo\",\"Gender\",\"BaseName\",\"Age\",\"personId\",\"personPicUrl\",\"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) -- Gitblit v1.8.0