package esutil
|
|
import (
|
"encoding/json"
|
"errors"
|
"fmt"
|
"io"
|
"io/ioutil"
|
"net/http"
|
"strconv"
|
"strings"
|
"time"
|
|
log "github.com/long/test/log"
|
)
|
|
func GetEsDataReq(url string, parama 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)
|
}
|
sources = append(sources, source)
|
}
|
data["datalist"] = sources
|
return nil, data
|
} else {
|
return nil, dat
|
}
|
}
|
|
//sdk类型
|
func sdkTypeToValue(i int) string {
|
value := []string{"人脸", "车辆", "人体", "入侵", "拥挤", "靠右行", "人员异常", "个体静止"}
|
|
return value[i-1]
|
}
|
|
func PostAction(sec int, Eurl 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\"]" +
|
"}"
|
err, tokenRes := GetEsDataReq(url, prama, true)
|
if err != nil {
|
log.Log.Errorln(err)
|
return nil
|
}
|
jsonstring, _ := json.Marshal(tokenRes)
|
if len(jsonstring) <= 26 {
|
return nil
|
}
|
return jsonstring
|
}
|
