package esutil
|
|
import (
|
"bytes"
|
"encoding/json"
|
"errors"
|
"fmt"
|
"io"
|
"io/ioutil"
|
"net/http"
|
"strconv"
|
"strings"
|
"time"
|
|
log "github.com/long/test/log"
|
)
|
|
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 {
|
return err, nil
|
}
|
|
req.Header.Add("Content-Type", "application/json")
|
timeout := time.Duration(10 * time.Second) //超时时间50ms
|
client := &http.Client{Timeout: timeout}
|
resp, err := client.Do(req)
|
|
if err != nil {
|
return err, nil
|
}
|
|
defer resp.Body.Close()
|
body, err := ioutil.ReadAll(resp.Body)
|
if err != nil {
|
return err, nil
|
}
|
|
jsonStr := string(body)
|
var dat map[string]interface{}
|
dec := json.NewDecoder(strings.NewReader(jsonStr))
|
|
if err := dec.Decode(&dat); err == io.EOF {
|
return err, nil
|
} else if err != nil {
|
return err, nil
|
}
|
// 是否需要 解析 es 返回的 source
|
if isSource {
|
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, 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, 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
|
}
|
|
if !strings.HasPrefix(psm, "http") {
|
source["picSmUrl"] = picurl + psm
|
}
|
|
prace, exist := source["Race"]
|
if exist {
|
source["race"] = prace
|
}
|
|
pGender, exist := source["Gender"]
|
if exist {
|
source["gender"] = pGender
|
}
|
|
source["ageDescription"] = getAgeDesc(source["Age"])
|
|
source["videoNum"] = getVideoUrl(source)
|
picDate := source["picDate"].(string)
|
lastIdx := strings.LastIndex(picDate,":")
|
picDateStr := picDate[:lastIdx]
|
if err == nil {
|
source["picDate"] = picDateStr
|
}
|
|
baseInfo := getSourceBaseInfo(source)
|
source["baseInfo"] = baseInfo
|
|
sources = append(sources, source)
|
}
|
data["datalist"] = sources
|
return nil, data
|
} else {
|
return nil, dat
|
}
|
}
|
|
func getAgeDesc(age interface{})(ageDesc string) {
|
if age !=nil {
|
ageInt := age.(float64)
|
if ageInt >0 && ageInt<7 {
|
ageDesc = "童年"
|
} else if ageInt >=7 && ageInt<18 {
|
ageDesc = "少年"
|
} else if ageInt >=18 && ageInt<40 {
|
ageDesc = "青年"
|
} else if ageInt >=40 && ageInt<65 {
|
ageDesc = "中年"
|
} else if ageInt >=65 {
|
ageDesc = "老年"
|
} else {
|
ageDesc = ""
|
}
|
}
|
return ageDesc
|
}
|
|
type BaseInfo struct {
|
TaskId string `json:"taskId"`
|
TaskName string `json:"taskName"`
|
LikePer string `json:"likePer"`
|
TableId string `json:"tableId"`
|
TableName string `json:"tableName"`
|
PersonId string `json:"personId"`
|
PersonPicUrl string `json:"personPicUrl"`
|
PersonName string `json:"personName"`
|
Gender string `json:"gender"`
|
PhoneNum string `json:"phoneNum"`
|
IDCard string `json:"IDCard"`
|
MonitorLevel string `json:"monitorLevel"`
|
Content string `json:"content"`
|
}
|
|
func getSourceBaseInfo(source map[string]interface{}) []BaseInfo {
|
sdkType := source["sdkType"].(string)
|
baseInfoArr := make([]BaseInfo,0)
|
if sdkType == "人脸" {
|
likePer,baseName,personId,idCard,personPicUrl,gender,content :="","","","","","",""
|
if source["likePer"] !=nil {
|
likePer = source["likePer"].(string)
|
}
|
if source["BaseName"] !=nil {
|
baseName = source["BaseName"].(string)
|
}
|
if source["personId"] !=nil {
|
personId = source["personId"].(string)
|
}
|
if source["idcard"] !=nil {
|
idCard = source["idcard"].(string)
|
}
|
if source["personPicUrl"] !=nil {
|
personPicUrl = source["personPicUrl"].(string)
|
}
|
if source["Gender"] !=nil {
|
gender = source["Gender"].(string)
|
}
|
if source["content"] !=nil {
|
content = source["content"].(string)
|
}
|
var baseInfo = BaseInfo{
|
TaskId:"",//2.0新字段
|
TaskName:"",//2.0新字段
|
LikePer:likePer,
|
TableId:"",//2.0新字段
|
TableName:baseName,
|
PersonId:personId,
|
PersonName:idCard,//人员姓名，从管理平台获取
|
PersonPicUrl:personPicUrl,
|
Gender:gender,
|
PhoneNum:"",//手机号，从管理平台获取
|
IDCard:idCard,
|
MonitorLevel:"",//2.0新字段
|
Content:content,
|
}
|
|
baseInfoArr = append(baseInfoArr, baseInfo)
|
//bytes, err := json.Marshal(baseInfoArr)
|
//if err !=nil {
|
// return ""
|
//}
|
}
|
return baseInfoArr
|
}
|
|
func getVideoUrl(source map[string]interface{}) (videoUrl string){
|
imgKey := source["imgKey"].(string)
|
picDate := 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"] = 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
|
videoUrl = ""
|
if !strings.Contains(filePath, "/cut"){
|
videoUrl = ""
|
} else {
|
strArr := strings.Split(filePath, "/cut")
|
ngxUrl := ngxMap[indeviceId]
|
if ngxUrl !="" && len(strArr) >0 {
|
videoUrl = ngxUrl + strArr[1]
|
}
|
}
|
if videoUrl == "" {
|
fmt.Println("videoReqUrl: ",reqUrl)
|
fmt.Printf("imgKey:%s ,picDate:%s ,cameraId:%s ",imgKey,picDate,cameraId)
|
}
|
|
return videoUrl
|
}
|
|
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)
|
}
|
}
|
timeOut:= time.Duration(8*time.Second)//set request timeout
|
client := &http.Client{
|
Timeout:timeOut,
|
}
|
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, picurl 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\",\"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)
|
|
if err != nil {
|
log.Log.Errorln(err)
|
return nil
|
}
|
jsonstring, _ := json.Marshal(tokenRes)
|
if len(jsonstring) <= 26 {
|
return nil
|
}
|
return jsonstring
|
}
|
