| | |
| | | package esutil |
| | | |
| | | import ( |
| | | "bytes" |
| | | "encoding/json" |
| | | "errors" |
| | | "fmt" |
| | | "io" |
| | | "io/ioutil" |
| | |
| | | log "github.com/long/test/log" |
| | | ) |
| | | |
| | | func GetEsDataReq(url string, parama string, isSource bool) (error, map[string]interface{}) { |
| | | log.Log.Infoln("es 查询请求路径" + url) // 配置信息 获取 |
| | | 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)) |
| | | |
| | | if err != nil { |
| | | log.Log.Errorln("build request error! ") |
| | | return err, nil |
| | | } |
| | | |
| | |
| | | dec := json.NewDecoder(strings.NewReader(jsonStr)) |
| | | |
| | | if err := dec.Decode(&dat); err == io.EOF { |
| | | log.Log.Infoln(err.Error()) |
| | | return err, nil |
| | | } else if err != nil { |
| | | log.Log.Errorln(err.Error()) |
| | | return err, nil |
| | | } |
| | | // 是否需要 解析 es 返回的 source |
| | | if isSource { |
| | | dat = dat["hits"].(map[string]interface{}) |
| | | dat, ok := dat["hits"].(map[string]interface{}) |
| | | if !ok { |
| | | return errors.New("data is not type of map[string]interface{}"), nil |
| | | } |
| | | |
| | | var data = make(map[string]interface{}, 2) |
| | | data["total"] = dat["total"] |
| | | sources := []interface{}{} |
| | | for _, value := range dat["hits"].([]interface{}) { |
| | | source := value.(map[string]interface{})["_source"].(map[string]interface{}) |
| | | source, ok := value.(map[string]interface{})["_source"].(map[string]interface{}) |
| | | if !ok { |
| | | return errors.New("value is not type of map[string]interface{}"), nil |
| | | } |
| | | |
| | | source["id"] = value.(map[string]interface{})["_id"] |
| | | |
| | | sdkType := source["sdkType"] |
| | | if sdkType != nil { |
| | | sdk, _ := strconv.Atoi(sdkType.(string)) |
| | | sdk, err := strconv.Atoi(sdkType.(string)) |
| | | if err != nil { |
| | | return err, nil |
| | | } |
| | | |
| | | 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 |
| | |
| | | } |
| | | } |
| | | |
| | | 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类型 |
| | | func sdkTypeToValue(i int) string { |
| | | value := []string{"人脸", "车辆", "人体", "入侵", "拥挤", "靠右行", "人员异常", "个体静止"} |
| | |
| | | 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\":[\"BaseName\",\"Gender\",\"Race\",\"content\",\"idcard\",\"picAddress\",\"picDate\",\"sdkType\",\"Age\",\"personId\",\"personIsHub\",\"personPicUrl\",\"picLocalUrl\",\"picSmUrl\",\"videoIp\",\"videoNum\",\"videoReqNum\",\"ageDescription\",\"likePer\",\"picMaxUrl\"]" + "}" |
| | | "\"_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) |
| | | |
| | | err, tokenRes := GetEsDataReq(url, prama, true) |
| | | if err != nil { |
| | | log.Log.Errorln(err) |
| | | return nil |