From f6ca7bb43270474fa876ff6ba62c6b2113b045ad Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期五, 31 五月 2024 14:03:29 +0800
Subject: [PATCH] Optimize day and night algorithm, format date and time, batch process by date.
---
EsApi.go | 2746 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 2,585 insertions(+), 161 deletions(-)
diff --git a/EsApi.go b/EsApi.go
index 4aff237..f27b95c 100644
--- a/EsApi.go
+++ b/EsApi.go
@@ -1,219 +1,2643 @@
package esutil
import (
- "encoding/json"
- "errors"
- "fmt"
- "strconv"
- "strings"
+ "basic.com/pubsub/protomsg.git"
+ "encoding/json"
+ "errors"
+ "fmt"
+ "sort"
+ "strconv"
+ "strings"
+ "sync"
+ "time"
+)
- "basic.com/pubsub/protomsg.git"
- )
-
-// 鏌ヨ搴曞簱浜哄憳淇℃伅
-func Personinfos( queryIndex int, queryNums int, indexName string, serverIp string, serverPort string, analyServerId string) ([]*protomsg.Esinfo, error){
- var dbinfos []*protomsg.Esinfo
- point := strconv.Itoa(queryIndex)
- number := strconv.Itoa(queryNums)
- JsonDSL := ""
- if indexName == "videopersons" {
- JsonDSL = ` {
- "from": ` + point + `,
- "query": {
- "bool": {
- "filter": [
- {
- "term": {
- "analyServerId": "` + analyServerId + `"
- }
- }
- ]
- }
- },
- "size":`+ number +`,
- "_source": [
- "id",
- "faceFeature"
- ]
- }`
- }else {
- JsonDSL = ` {
- "from": ` + point + `,
- "query": {
- "match_all": {}
- },
- "size":`+ number +`,
- "_source": [
- "id",
- "tableId",
- "faceFeature"
- ]
- }`
- }
- //fmt.Println("url: "+"http://"+serverIp+":"+serverPort+"/"+indexName+"/_search","body: ",JsonDSL)
- buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_search",[]byte(JsonDSL))
- if err != nil {
- return dbinfos ,errors.New("http request dbtablename info is err!")
- }
-
- // 杩斿洖 _source 鏁扮粍
- sources, err := Sourcelist(buf)
- if err != nil {
- return dbinfos,err
- }
-
- // 杩斿洖鎵�鏈夋煡璇㈢殑鏁版嵁
- dbpersoninfos := Parsesources(sources)
- return dbpersoninfos, nil
+var logPrint = func(i ...interface{}) {
+ fmt.Println(i)
}
-// 鏍规嵁搴曞簱id鏌ヨ搴曞簱淇℃伅
-func Dbtablefosbyid (id []string, indexName string, serverIp string, serverPort string) ([]protomsg.Dbtable, error) {
- var dbinfo []protomsg.Dbtable
- dbtableId := strings.Replace(strings.Trim(fmt.Sprint(id), "[]"), " ", "\",\"", -1)
- var dbinfoRequest = `
+func InitLog(fn func(i ...interface{})) {
+ if fn != nil {
+ logPrint = fn
+ }
+}
+
+//***********************閲嶅簡Start**********************************//
+
+//鍒ゆ柇鏃堕棿鏄惁鍐嶈寖鍥翠箣鍐�
+func isTimeInRange(timeStr, startStr, endStr string) bool {
+ layout := "15:04:05"
+
+ timeStamp, err := time.Parse(layout, timeStr)
+ if err != nil {
+ fmt.Println("Error parsing timestamp:", err)
+ return false
+ }
+
+ startTime, err := time.Parse(layout, startStr)
+ if err != nil {
+ fmt.Println("Error parsing start time:", err)
+ return false
+ }
+
+ endTime, err := time.Parse(layout, endStr)
+ if err != nil {
+ fmt.Println("Error parsing end time:", err)
+ return false
+ }
+
+ if startTime.After(endTime) {
+ // 璺ㄨ秺鏃ユ湡鐨勬儏鍐�
+ //fmt.Println("璺ㄦ棩鏈�",timeStamp, timeStamp.After(startTime), timeStamp.Before(endTime))
+ return timeStamp.After(startTime) || timeStamp.Before(endTime)
+ } else {
+ // 涓嶈法瓒婃棩鏈熺殑鎯呭喌
+ //fmt.Println("涓嶈法鏃ユ湡",timeStamp, timeStamp.After(startTime), timeStamp.Before(endTime))
+ return timeStamp.After(startTime) && timeStamp.Before(endTime)
+ }
+}
+
+//鍒ゆ柇涓や釜鏃堕棿鍏堝悗
+func compareTimes(time1Str, time2Str string) int {
+ layout := "15:04:05"
+
+ time1, err := time.Parse(layout, time1Str)
+ if err != nil {
+ fmt.Println("Error parsing time 1:", err)
+ return 0
+ }
+
+ time2, err := time.Parse(layout, time2Str)
+ if err != nil {
+ fmt.Println("Error parsing time 2:", err)
+ return 0
+ }
+
+ if time1.Before(time2) {
+ return -1 // time1 鍦� time2 涔嬪墠
+ } else if time1.After(time2) {
+ return 1 // time1 鍦� time2 涔嬪悗
+ } else {
+ return 0 // time1 鍜� time2 鐩哥瓑
+ }
+}
+
+//鍒ゆ柇鏃ユ湡鐩稿樊鍑犲ぉ
+func daysBetweenDates(date1Str, date2Str string) int {
+ layout := "2006-01-02"
+
+ date1, err := time.Parse(layout, date1Str)
+ if err != nil {
+ fmt.Println("Error parsing date 1:", err)
+ return 0
+ }
+
+ date2, err := time.Parse(layout, date2Str)
+ if err != nil {
+ fmt.Println("Error parsing date 2:", err)
+ return 0
+ }
+
+ duration := date2.Sub(date1)
+ days := int(duration.Hours() / 24)
+
+ return days
+}
+
+//璁＄畻鏃堕棿闃堝��
+func checkTimeDifference(timestampStr1 string, timestampStr2 string, intervalInMinutes int) bool {
+ layout := "2006-01-02 15:04:05"
+ timestampStr1 = timestampStr1[:19]
+ timestampStr2 = timestampStr2[:19]
+ // 灏嗗瓧绗︿覆瑙ｆ瀽涓烘椂闂�
+ time1, err := time.Parse(layout, timestampStr1)
+ if err != nil {
+ fmt.Println("鏃堕棿瑙ｆ瀽澶辫触:", err)
+ return false
+ }
+ time2, err := time.Parse(layout, timestampStr2)
+ if err != nil {
+ fmt.Println("鏃堕棿瑙ｆ瀽澶辫触:", err)
+ return false
+ }
+
+ // 璁＄畻鏃堕棿宸�
+ diff := time2.Sub(time1)
+
+ // 妫�鏌ユ椂闂村樊鏄惁灏忎簬绛変簬鎸囧畾鐨勯棿闅�
+ if diff.Minutes() <= float64(intervalInMinutes) {
+ return true
+ } else {
+ return false
+ }
+}
+
+////鏍煎紡鍖栨椂闂磆h:mm:ss:zzz to hh:mm:ss
+//func formatTime(inputTime string) (string, error) {
+// parsedTime, err := time.Parse("15:04:05:000", inputTime)
+// if err != nil {
+// return "", err
+// }
+//
+// formattedTime := parsedTime.Format("15:04:05")
+// return formattedTime, nil
+//}
+
+func decodeActivityId(frequency int, intervalInMinutes int, source []map[string]interface{}) ([]map[string]interface{}, error) {
+ docInfo := make([]map[string]interface{}, 0)
+ for _, info := range source {
+ tmpInfo := make(map[string]interface{})
+ documentNumber := info["key"].(string)
+ tmpInfo["documentNumber"] = documentNumber
+ //fmt.Println("documentNumber: ", documentNumber)
+ topHits := info["top_hits"].(map[string]interface{})
+ hits := topHits["hits"].(map[string]interface{})
+ hitsResult := hits["hits"].([]interface{})
+ dataId := make([]string, 0)
+ picUrl := ""
+ if hitsResult[0].(map[string]interface{})["_source"].(map[string]interface{})["baseInfo"] != nil {
+ baseInfo := hitsResult[0].(map[string]interface{})["_source"].(map[string]interface{})["baseInfo"]
+ if v, ok := baseInfo.([]interface{}); ok {
+ picUrl = v[0].(map[string]interface{})["targetPicUrl"].(string)
+ } else if v, ok := baseInfo.(map[string]interface{}); ok {
+ picUrl = v["targetPicUrl"].(string)
+ }
+ } else {
+ if hitsResult[0].(map[string]interface{})["_source"].(map[string]interface{})["targetInfo"] != nil {
+ picUrl = hitsResult[0].(map[string]interface{})["_source"].(map[string]interface{})["targetInfo"].([]interface{})[0].(map[string]interface{})["picSmUrl"].(string)
+ }
+ }
+ tmpInfo["picUrl"] = picUrl
+ lastCaptureTime := ""
+ for _, sourceInfo := range hitsResult {
+ rSourceInfo := sourceInfo.(map[string]interface{})
+ source := rSourceInfo["_source"].(map[string]interface{})
+ captureTime := source["picDate"].(string)
+ id := source["id"].(string)
+ if lastCaptureTime != "" {
+ if checkTimeDifference(lastCaptureTime[:19], captureTime[:19], intervalInMinutes) {
+ //fmt.Println("鏃堕棿灏忎簬杩炵画闃堝��", lastCaptureTime, captureTime)
+ continue
+ }
+ }
+ lastCaptureTime = captureTime
+ dataId = append(dataId, id)
+ }
+
+ if len(dataId) > frequency {
+ tmpInfo["id"] = dataId
+ docInfo = append(docInfo, tmpInfo)
+ }
+ }
+ return docInfo, nil
+}
+
+func DayNightActivityQuery(comIds []string, cameraIds []string, docNumber string, startDateStr string, endDateStr string, activeHour string, frequency int,
+ intervalInMinutes int, indexName string, serverIp string, serverPort string) (map[string]interface{}, error) {
+ esURL := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ hours := strings.Split(activeHour, "-")
+ startTimeStr := hours[0]
+ endTimeStr := hours[1]
+ layoutTime := "15:04:05"
+ startTime, err := time.Parse(layoutTime, startTimeStr)
+ if err != nil {
+ return nil, err
+ }
+ endTime, err := time.Parse(layoutTime, endTimeStr)
+ if err != nil {
+ return nil, err
+ }
+ layoutDate := "2006-01-02"
+ startDate, err := time.Parse(layoutDate, startDateStr)
+ if err != nil {
+ fmt.Println("Error parsing start time:", err)
+ return nil, err
+ }
+
+ endDate, err := time.Parse(layoutDate, endDateStr)
+ if err != nil {
+ fmt.Println("Error parsing end time:", err)
+ return nil, err
+ }
+ rangePicDate := make([]string, 0)
+ if startTime.Before(endTime) {
+ for !startDate.After(endDate) {
+ sDateTime := startDate.Format(layoutDate) + " " + startTimeStr
+ eDateTime := startDate.Format(layoutDate) + " " + endTimeStr
+ rangePicDate = append(rangePicDate, `
+ {
+ "range": {
+ "picDate": {
+ "gte": "`+sDateTime+`",
+ "lt": "`+eDateTime+`"
+ }
+ }
+ }`)
+ startDate = startDate.AddDate(0, 0, 1)
+ }
+ } else if startTime.After(endTime) {
+ if startDate.Year() == endDate.Year() && startDate.Month() == endDate.Month() && startDate.Day() == endDate.Day() {
+ fmt.Println("璺ㄦ棩鏈熶笉鏀寔鍗曟棩鏌ヨ")
+ return nil, errors.New("璺ㄦ棩鏈熶笉鏀寔鍗曟棩鏌ヨ锛�")
+ }
+ for !startDate.After(endDate) {
+ sDateTime := startDate.Format(layoutDate) + " " + startTimeStr
+ startDate = startDate.AddDate(0, 0, 1)
+ //fendDate := startDate.AddDate(0, 0, 1)
+ if startDate.Year() == endDate.Year() && startDate.Month() == endDate.Month() && startDate.Day() == endDate.Day() {
+ eDateTime := startDate.Format(layoutDate) + " " + endTimeStr
+ rangePicDate = append(rangePicDate, `
+ {
+ "range": {
+ "picDate": {
+ "gte": "`+sDateTime+`",
+ "lt": "`+eDateTime+`"
+ }
+ }
+ }`)
+ //fmt.Println("startDate: ", startDate, "/t endDate: ", endDate)
+ break
+ }
+ eDateTime := startDate.Format(layoutDate) + " " + endTimeStr
+ rangePicDate = append(rangePicDate, `
+ {
+ "range": {
+ "picDate": {
+ "gte": "`+sDateTime+`",
+ "lt": "`+eDateTime+`"
+ }
+ }
+ }`)
+ }
+ }
+ //fmt.Println(rangePicDate)
+ //return nil, nil
+ filterDocIdAttr := ""
+ if docNumber != "" {
+ filterDocIdAttr = "{\"term\": {\"documentNumber\": \"" + docNumber + "\"}},"
+ }
+ comIdsStr := ""
+ if comIds != nil && len(comIds) > 0 {
+ esComIds := strings.Replace(strings.Trim(fmt.Sprint(comIds), "[]"), " ", "\",\"", -1)
+ comIdsStr = "{\"terms\":{\"communityId\":[\"" + esComIds + "\"]}},"
+ }
+ cameraIdsStr := ""
+ if comIds != nil && len(cameraIds) > 0 {
+ esCameraIds := strings.Replace(strings.Trim(fmt.Sprint(cameraIds), "[]"), " ", "\",\"", -1)
+ cameraIdsStr = "{\"terms\":{\"cameraId\":[\"" + esCameraIds + "\"]}},"
+ }
+
+ var result = make(map[string]interface{})
+ docInfos := make([]map[string]interface{}, 0)
+ for _, rpd := range rangePicDate {
+ //if comIdsStr != "" || filterDocIdAttr != "" || cameraIdsStr != "" {
+ // rpd += ","
+ //}
+ queryDSL := `
+ {
+ "size": 0,
+ "query": {
+ "bool": {
+ "filter": [
+ ` + comIdsStr + `
+ ` + cameraIdsStr + `
+ ` + filterDocIdAttr + `
+ ` + rpd + `
+ ],
+ "must_not": [
+ {
+ "term": {
+ "documentNumber": ""
+ }
+ }
+ ]
+ }
+ },
+ "aggs": {
+ "group_by_documentnumber": {
+ "terms": {
+ "field": "documentNumber",
+ "size": 100000
+ },
+ "aggs": {
+ "top_hits": {
+ "top_hits": {
+ "_source": [
+ "id",
+ "picDate",
+ "baseInfo.targetPicUrl",
+ "targetInfo.picSmUrl"
+ ],
+ "size": 100000,
+ "sort": [
+ {
+ "picDate": {
+ "order": "asc"
+ }
+ }
+ ]
+ }
+ }
+ }
+ }
+ }
+ }`
+ fmt.Println(esURL)
+ fmt.Println(queryDSL)
+ buf, err := EsReq("POST", esURL, []byte(queryDSL))
+ if err != nil {
+ return nil, err
+ }
+ source, err := SourceAggregationList(buf)
+ if err != nil {
+ return nil, err
+ }
+ //fmt.Println(source)
+ //docResult 鏈鏌ヨ缁撴灉
+ docResult, err := decodeActivityId(frequency, intervalInMinutes, source)
+ if err != nil {
+ return nil, err
+ }
+ if len(docResult) == 0 {
+ continue
+ }
+ if len(docInfos) == 0 {
+ docInfos = append(docInfos, docResult...)
+ continue
+ }
+ for _, dr := range docResult {
+ var found bool
+ for _, di := range docInfos {
+ if dr["documentNumber"].(string) == di["documentNumber"].(string) {
+ // 鎵惧埌浜嗙浉鍚岀殑documentNumber锛屽皢id鍚堝苟鍒癲i["id"]涓�
+ ids, ok := di["id"].([]string)
+ if !ok {
+ // 濡傛灉id涓嶆槸瀛楃涓插垏鐗囩被鍨嬶紝鍒欒繘琛屽垵濮嬪寲
+ ids = []string{}
+ }
+ ids = append(ids, dr["id"].([]string)...)
+ di["id"] = ids
+ //docInfos[diIndex] = di // 鏇存柊鍘熷docInfos涓殑鍊�
+ found = true
+ break
+ }
+ }
+ if !found {
+ // 娌℃湁鎵惧埌鐩稿悓鐨刣ocumentNumber锛屽皢dr娣诲姞鍒癲ocInfos涓�
+ docInfos = append(docInfos, dr)
+ }
+ }
+ }
+ if len(docInfos) == 0 {
+ return nil, nil
+ }
+ DataInfos, err := GetInfosByIds(docInfos[0]["id"].([]string), indexName, serverIp, serverPort)
+ result["documentNumbers"] = docInfos
+ result["datalist"] = DataInfos
+ return result, nil
+}
+
+type acmInfo struct {
+ documentNumber string
+ camerasInfos []camerasInfo
+}
+
+type camerasInfo struct {
+ cameraId string
+ captureInfos []captureInfo
+}
+
+type captureInfo struct {
+ id string
+ picDate string
+}
+
+func addSecondsToTimestamp(timestamp string, seconds int) (string, error) {
+ parsedTime, err := time.Parse("2006-01-02 15:04:05", timestamp)
+ if err != nil {
+ return "", err
+ }
+ newTime := parsedTime.Add(time.Second * time.Duration(seconds))
+ newTimestamp := newTime.Format("2006-01-02 15:04:05")
+ return newTimestamp, nil
+}
+
+func decodeInfo(intervalInMinutes int, source []map[string]interface{}) ([]acmInfo, error) {
+ acmInfos := make([]acmInfo, 0)
+ for _, info := range source {
+ var aInfo acmInfo
+ documentNumber := info["key"].(string)
+ aInfo.documentNumber = documentNumber
+ groupByCameraId := info["group_by_cameraId"].(map[string]interface{})
+ cameraBuckets := groupByCameraId["buckets"].([]interface{})
+ for _, cameraInfo := range cameraBuckets {
+ var camsInfo camerasInfo
+ cInfo := cameraInfo.(map[string]interface{})
+ cameraId := cInfo["key"].(string)
+ camsInfo.cameraId = cameraId
+ dataBuckets := cInfo["top_hits"].(map[string]interface{})["hits"].(map[string]interface{})["hits"].([]interface{})
+ markTime := ""
+ for _, dataInfo := range dataBuckets {
+ var capInfo captureInfo
+ dInfo := dataInfo.(map[string]interface{})
+ dSource := dInfo["_source"].(map[string]interface{})
+ id := dSource["id"].(string)
+ picDate := dSource["picDate"].(string)
+ //addFlag := false
+ if markTime == "" {
+ //addFlag = true
+ markTime = picDate
+ } else {
+ if checkTimeDifference(markTime, picDate, intervalInMinutes) {
+ //fmt.Println(markTime, picDate)
+ markTime = picDate
+ continue
+ }
+ markTime = picDate
+ }
+ capInfo.id = id
+ capInfo.picDate = picDate
+ camsInfo.captureInfos = append(camsInfo.captureInfos, capInfo)
+ }
+ aInfo.camerasInfos = append(aInfo.camerasInfos, camsInfo)
+ }
+ acmInfos = append(acmInfos, aInfo)
+ }
+ return acmInfos, nil
+}
+
+type addResultIds struct {
+ documentNumber string
+ unionIds []unionId
+}
+
+type unionId struct {
+ baseId string
+ targetId string
+}
+
+func addResultInfo(source []map[string]interface{}, targetAddResultIds *[]addResultIds, bId string) {
+ found := false
+ for _, info := range source {
+ documentNumber := info["key"].(string)
+ dataBuckets := info["top_hits"].(map[string]interface{})["hits"].(map[string]interface{})["hits"].([]interface{})
+ id := dataBuckets[0].(map[string]interface{})["_source"].(map[string]interface{})["id"].(string)
+ //fmt.Println("documentNumber: ", documentNumber, "\tid: ", id)
+ for i, docInfo := range *targetAddResultIds {
+ if documentNumber == docInfo.documentNumber {
+ //fmt.Println("鏂版洿鏂�")
+ (*targetAddResultIds)[i].unionIds = append((*targetAddResultIds)[i].unionIds, unionId{baseId: bId, targetId: id})
+ found = true
+ break
+ }
+ }
+ if !found {
+ //fmt.Println("鏂版坊鍔�")
+ var targetAddResultId addResultIds
+ targetAddResultId.documentNumber = documentNumber
+ targetAddResultId.unionIds = append(targetAddResultId.unionIds, unionId{baseId: bId, targetId: id})
+ *targetAddResultIds = append(*targetAddResultIds, targetAddResultId)
+ }
+
+ }
+}
+
+func removeDuplicates(nums []string) []string {
+ result := make([]string, 0)
+ seen := make(map[string]bool)
+
+ for _, num := range nums {
+ if !seen[num] {
+ result = append(result, num)
+ seen[num] = true
+ }
+ }
+
+ return result
+}
+
+func findAnalyzeCoordinatedMovementsInfos(infos []acmInfo, docNumber string, beforeTime int, afterTime int, frequency int,
+ indexName string, serverIp string, serverPort string) (map[string]interface{}, error) {
+ //baseAddResultIds := make([]addResultIds, 0)
+ targetAddResultIds := make([]addResultIds, 0)
+ esURL := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ for _, info := range infos {
+ for _, cInfo := range info.camerasInfos {
+ for _, pInfo := range cInfo.captureInfos {
+ gteDate, err := addSecondsToTimestamp(pInfo.picDate[:19], beforeTime)
+ if err != nil {
+ fmt.Println(err)
+ }
+ lteDate, err := addSecondsToTimestamp(pInfo.picDate[:19], afterTime)
+ if err != nil {
+ fmt.Println(err)
+ }
+ queryDSL := `
+ {
+ "size": 0,
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "range": {
+ "picDate": {
+ "gte": "` + gteDate + `",
+ "lte": "` + lteDate + `"
+ }
+ }
+ },
+ {
+ "term": {
+ "cameraId": "` + cInfo.cameraId + `"
+ }
+ }
+ ],
+ "must_not": [
+ {
+ "term": {
+ "documentNumber": "` + docNumber + `"
+ }
+ }
+ ]
+ }
+ },
+ "aggs": {
+ "group_by_documentnumber": {
+ "terms": {
+ "field": "documentNumber",
+ "size": 100000
+ },
+ "aggs": {
+ "top_hits": {
+ "top_hits": {
+ "_source": [
+ "id",
+ "cameraId",
+ "picDate"
+ ],
+ "size": 10000,
+ "sort": [
+ {
+ "picDate": {
+ "order": "asc"
+ }
+ }
+ ]
+ }
+ }
+ }
+ }
+ }
+ }`
+ //fmt.Println(esURL, queryDSL)
+ buf, err := EsReq("POST", esURL, []byte(queryDSL))
+ if err != nil {
+ return nil, err
+ }
+ source, err := SourceAggregationList(buf)
+ if err != nil {
+ return nil, err
+ }
+ //fmt.Println("pInfo.id: ", pInfo.id)
+ addResultInfo(source, &targetAddResultIds, pInfo.id)
+ //fmt.Println("targetAddResultIds: ", targetAddResultIds)
+ if err != nil {
+ return nil, err
+ }
+ //fmt.Println(source)
+ }
+ }
+ }
+ //fmt.Println("targetAddResultIds: ", targetAddResultIds)
+ baseIds := make([]string, 0)
+ targetIds := make([]string, 0)
+ for _, tAIdInfo := range targetAddResultIds {
+ if len(tAIdInfo.unionIds) >= frequency {
+ for _, unionId := range tAIdInfo.unionIds {
+ baseIds = append(baseIds, unionId.baseId)
+ targetIds = append(targetIds, unionId.targetId)
+ }
+ }
+ }
+
+ rdbaseIds := removeDuplicates(baseIds)
+ rdtargetIds := removeDuplicates(targetIds)
+ baseInfos, err := GetInfosByIds(rdbaseIds, indexName, serverIp, serverPort)
+ if err != nil {
+ return nil, err
+ }
+ targetInfos, err := GetInfosByIds(rdtargetIds, indexName, serverIp, serverPort)
+ if err != nil {
+ return nil, err
+ }
+ docNumberMap := make(map[string][]interface{})
+ for _, tinfo := range targetInfos {
+ docNumber := tinfo["documentNumber"].(string)
+ docNumberMap[docNumber] = append(docNumberMap[docNumber], tinfo)
+ }
+ targetRecordInfos := make([]map[string]interface{}, 0)
+ for docNumber, infos := range docNumberMap {
+ ifs := make(map[string]interface{})
+ ifs["documentNumber"] = docNumber
+ ifs["recordInfos"] = infos
+ targetRecordInfos = append(targetRecordInfos, ifs)
+ }
+ result := make(map[string]interface{})
+ result["baseRecordInfo"] = baseInfos
+ result["targetRecordInfo"] = targetRecordInfos
+ return result, nil
+}
+
+func AnalyzeCoordinatedMovements(comIds []string, docNumber string, startDate string, endDate string, beforeTime int, afterTime int, frequency int,
+ intervalInMinutes int, indexName string, serverIp string, serverPort string) (map[string]interface{}, error) {
+ esURL := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ //鍒ゆ柇绀惧尯IDs
+ comIdsStr := ""
+ if comIds == nil || len(comIds) > 0 {
+ esComIds := strings.Replace(strings.Trim(fmt.Sprint(comIds), "[]"), " ", "\",\"", -1)
+ comIdsStr = "{\"terms\":{\"communityId\":[\"" + esComIds + "\"]}},"
+ }
+ queryDSL := `
+ {
+ "size": 0,
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "range": {
+ "picDate": {
+ "gte": "` + startDate + `",
+ "lte": "` + endDate + `"
+ }
+ }
+ },
+ ` + comIdsStr + `
+ {
+ "term": {
+ "documentNumber": "` + docNumber + `"
+ }
+ }
+ ]
+ }
+ },
+ "aggs": {
+ "group_by_documentnumber": {
+ "terms": {
+ "field": "documentNumber",
+ "size": 100000
+ },
+ "aggs": {
+ "group_by_cameraId": {
+ "terms": {
+ "field": "cameraId",
+ "size": 10000
+ },
+ "aggs": {
+ "top_hits": {
+ "top_hits": {
+ "_source": [
+ "id",
+ "cameraId",
+ "picDate"
+ ],
+ "size": 10000,
+ "sort": [
+ {
+ "picDate": {
+ "order": "asc"
+ }
+ }
+ ]
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+ }`
+ //fmt.Println(esURL)
+ //fmt.Println(queryDSL)
+ buf, err := EsReq("POST", esURL, []byte(queryDSL))
+ if err != nil {
+ return nil, err
+ }
+ source, err := SourceAggregationList(buf)
+ if err != nil {
+ return nil, err
+ }
+ docResult, err := decodeInfo(intervalInMinutes, source)
+ if err != nil {
+ return nil, err
+ }
+ //fmt.Println(docResult)
+ result, err := findAnalyzeCoordinatedMovementsInfos(docResult, docNumber, beforeTime, afterTime, frequency, indexName, serverIp, serverPort)
+ if err != nil {
+ return nil, err
+ }
+ //fmt.Println(result)
+ return result, nil
+}
+
+func GetInfosByIds(ids []string, indexName string, serverIp string, serverPort string) ([]map[string]interface{}, error) {
+ captureIds := strings.Replace(strings.Trim(fmt.Sprint(ids), "[]"), " ", "\",\"", -1)
+ esURL := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ queryDSL := `
{
"query": {
"bool": {
"filter": [{
"terms": {
"id": [
- "`+ dbtableId +`"
+ "` + captureIds + `"
]
}
}]
}
- }
+ },
+ "size":1000000,
+ "sort":[{"picDate":{"order":"desc"}}],
+ "_source": {"includes":[],"excludes":["*.feature"]}
}
- `
- buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_search",[]byte(dbinfoRequest))
- if err != nil {
- return dbinfo, err
- }
+`
+ buf, err := EsReq("POST", esURL, []byte(queryDSL))
+ if err != nil {
+ return nil, err
+ }
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return nil, err
+ }
- sources, err := Sourcelist(buf)
- if err != nil {
- return dbinfo , err
- }
-
- dbtable := Dbtablebyid(sources)
- return dbtable, nil
+ return sources, nil
}
-// 鏍规嵁搴曞簱浜哄憳id鏌ヨ搴曞簱浜哄憳淇℃伅
-func Dbpersoninfosbyid (id []string, indexName string, serverIp string, serverPort string) ([]protomsg.Dbperson, error) {
- var dbinfo []protomsg.Dbperson
- dbtablePersonId := strings.Replace(strings.Trim(fmt.Sprint(id), "[]"), " ", "\",\"", -1)
- var dbinfoRequest = `
+// ***********************閲嶅簡End************************************//
+// 鏍规嵁鎶撴媿浜哄憳id鏌ヨ鎶撴媿浜哄憳淇℃伅
+func AIOceaninfosbyid(id []string, indexName string, serverIp string, serverPort string) ([]protomsg.AIOcean, error) {
+ var aIOceanInfo []protomsg.AIOcean
+ videopersonsPersonId := strings.Replace(strings.Trim(fmt.Sprint(id), "[]"), " ", "\",\"", -1)
+ var dbinfoRequest = `
{
"query": {
"bool": {
"filter": [{
"terms": {
"id": [
- "`+ dbtablePersonId +`"
+ "` + videopersonsPersonId + `"
]
}
}]
}
- }
+ },
+ "size":1000000
}
`
- buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_search",[]byte(dbinfoRequest))
- if err != nil {
- return dbinfo, err
- }
+ buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_search", []byte(dbinfoRequest))
+ if err != nil {
+ return aIOceanInfo, err
+ }
- sources, err := Sourcelist(buf)
- if err != nil {
- return dbinfo , err
- }
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return aIOceanInfo, err
+ }
- dbperson := Dbpersonbyid(sources)
- println(dbperson)
- return dbperson,nil
+ aIOcean := AIOceanAnalysis(sources)
+ return aIOcean, nil
}
-// 鏍规嵁tableid 鏌ヨtablename
-func Dbtablename(tableid string, indexName string, serverIp string, serverPort string) (tablename string, err error) {
- var dbinfotable =` {
- "query": {
- "bool": {
- "filter": [
+// 鏍规嵁鎶撴媿浜哄憳id鏌ヨ瑙嗛鍦板潃
+func AIOceanVideoUrlbyid(id string, indexName string, serverIp string, serverPort string) (string, error) {
+ //var aIOceanInfo []protomsg.AIOcean
+ //videopersonsPersonId := strings.Replace(strings.Trim(fmt.Sprint(id), "[]"), " ", "\",\"", -1)
+ var dbinfoRequest = `
+ {
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "term": {
+ "id": "` + id + `"
+ }
+ }
+ ]
+ }
+ },
+ "_source": [
+ "videoUrl"
+ ]
+ }
+ `
+ buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_search", []byte(dbinfoRequest))
+ if err != nil {
+ return "", err
+ }
+
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return "", err
+ }
+ videoUrl := sources[0]["videoUrl"].(string)
+ //aIOcean := AIOceanAnalysis(sources)
+ return videoUrl, nil
+}
+
+// 鏍规嵁鎶撴媿搴撲汉鍛榠d鏌ヨ鐗瑰緛鍊�
+func GetVideoPersonFaceFeatureById(id string, indexName string, serverIp string, serverPort string) (string, error) {
+ var jsonDSL = `
+ {
+ "query": {
+ "bool": {
+ "filter": [{
+ "term": {
+ "id":"` + id + `"
+ }
+ }]
+ }
+ },
+ "_source":["targetInfo.feature"]
+ }
+ `
+ buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_search", []byte(jsonDSL))
+ if err != nil {
+ return "", err
+ }
+
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return "", err
+ }
+
+ feature := sources[0]["targetInfo"].([]interface{})[0].(map[string]interface{})["feature"].(string)
+ return feature, nil
+}
+
+// 鏍规嵁鐩爣id鏌ヨ宸茶拷鍔犳潯鏁�
+func GetLinkTagInfoSize(id string, indexName string, serverIp string, serverPort string) (size int, err error) {
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ queryDSL := `{
+ "query": {
+ "term":{
+ "id":"` + id + `"
+ }
+ }
+ }`
+ buf, err := EsReq("POST", url, []byte(queryDSL))
+ if err != nil {
+ return -1, err
+ }
+ source, err := Sourcelist(buf)
+ if err != nil {
+ return -1, err
+ }
+ if source[0]["linkTagInfo"] != nil {
+ size = len(source[0]["linkTagInfo"].([]interface{}))
+ } else {
+ return -1, errors.New("璇ユ暟缁勪笉瀛樺湪")
+ }
+ return size, nil
+}
+
+// 鏍规嵁鐩爣id杩藉姞璺熻釜淇℃伅
+func AppendTargetInfo(id string, targetInfo string, indexName string, serverIp string, serverPort string, updateTime string) (string, error) {
+ if targetInfo == "" {
+ return "", errors.New("append data is nil")
+ }
+ var info interface{}
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_update_by_query?refresh=true"
+ jsonDSL := `{
+ "query": {
+ "term":{
+ "id":"` + id + `"
+ }
+ },
+ "script": {
+ "lang": "painless",
+ "inline": "ctx._source.linkTagInfo.add(params.newparam);ctx._source.updateTime='` + updateTime + `'",
+ "params": {
+ "newparam": ` + targetInfo + `
+ }
+ }
+}`
+ logPrint(jsonDSL)
+ buf, err := EsReq("POST", url, []byte(jsonDSL))
+ if err != nil {
+ return "", err
+ }
+ json.Unmarshal(buf, &info)
+ out, ok := info.(map[string]interface{})
+ logPrint(out)
+ if !ok {
+ return "", errors.New("http response interface can not change map[string]interface{}")
+ }
+ middle, ok := out["updated"].(float64)
+ if !ok {
+ return "", errors.New("first updated change error!")
+ }
+ mes := ""
+ if middle == 1 {
+ mes = "杩藉姞鎴愬姛"
+ }
+ if middle == 0 {
+ mes = "宸茬粡杩藉姞"
+ }
+ return mes, nil
+
+}
+
+/**************************************customer analysis util start**************************************/
+/*******************sort []map util*******************/
+type MapsSort struct {
+ Key string
+ MapList []map[string]interface{}
+}
+
+func (m *MapsSort) Len() int {
+ return len(m.MapList)
+}
+
+func (m *MapsSort) Less(i, j int) bool {
+ return m.MapList[i][m.Key].(string) > m.MapList[j][m.Key].(string)
+}
+
+func (m *MapsSort) Swap(i, j int) {
+ m.MapList[i], m.MapList[j] = m.MapList[j], m.MapList[i]
+}
+
+/*******************sort []map util*******************/
+//鏍规嵁鏃堕棿鑼冨洿鑱氬悎鎵�鏈夊尯鍩熶汉淇℃伅锛岃繑鍥炲浐瀹氭潯鏁�
+func GetFaceDataByTimeAndTotal(startTime string, endTime string, total int, thresholdTime int, thresholdStayTime int, serverIp string, serverPort string, indexName string) (resData []map[string]interface{}, err error) {
+ var requestUrl = "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ var requestBody = `{
+ "query": {
+ "bool": {
+ "filter": [
{
- "term": {
- "id":"`+tableid+`"
+ "range": {
+ "picDate": {
+ "gte": "` + startTime + `",
+ "lte": "` + endTime + `"
+ }
+ }
+ },
+ {
+ "term":{
+ "targetInfo.targetType.raw": "FaceDetect"
}
}
- ]
+ ]
+ }
+ },
+ "size": 0,
+ "aggs": {
+ "buckets_aggs": {
+ "composite": {
+ "sources": [
+ {
+ "faceId": {
+ "terms": {
+ "field": "baseInfo.targetId"
+ }
+ }
+ },
+ {
+ "areaId": {
+ "terms": {
+ "field": "targetInfo.areaId"
+ }
+ }
+ }
+ ],
+ "size": 10000000
+ },
+ "aggs": {
+ "top_attention_hits": {
+ "top_hits": {
+ "size": 1000000,
+ "sort": [
+ {
+ "picDate": {
+ "order": "asc"
+ }
+ }
+ ],
+ "_source": {
+ "includes": [
+ "baseInfo.targetId",
+ "targetInfo.picSmUrl",
+ "targetInfo.areaId",
+ "picDate"
+ ]
+ }
+ }
+ }
}
- },
- "_source": [
- "tableName"
- ]
- }
- `
- buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_search",[]byte(dbinfotable))
- if err != nil {
- return "" ,errors.New("http request dbtablename info is err!")
- }
- sources, err := Sourcelist(buf)
- if err != nil {
- return "",err
- }
-
- for _, source := range sources {
- if name, ok := source["tableName"].(string); ok {
- tablename = name
- break
}
}
- return tablename, nil
+}`
+ buf, err := EsReq("POST", requestUrl, []byte(requestBody))
+ if err != nil {
+ return nil, err
+ }
+ source, err := FaceSourceAggregations(buf, thresholdTime, thresholdStayTime)
+ if err != nil {
+ return nil, err
+ }
+ if len(source) == 0 {
+ return source, nil
+ }
+ faceSource := make([]map[string]interface{}, 0)
+ for index, info := range source {
+ if int(info["stayTime"].(float64)) > thresholdStayTime {
+ faceSource = append(faceSource, source[index])
+ }
+ }
+ mapsSort := MapsSort{}
+ mapsSort.Key = "endTime"
+ mapsSort.MapList = faceSource
+ sort.Sort(&mapsSort)
+ if len(faceSource) > total {
+ return mapsSort.MapList[:total], nil
+ }
+ return mapsSort.MapList, nil
}
+func GetFaceDataByTimeAndId(startTime string, endTime string, id string, thresholdTime int, thresholdStayTime int, serverIp string, serverPort string, indexName string) (resData []map[string]interface{}, err error) {
+ var requestUrl = "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ var requestBody = `{
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "range": {
+ "picDate": {
+ "gte": "` + startTime + `",
+ "lte": "` + endTime + `"
+ }
+ }
+ },
+ {
+ "term":{
+ "targetInfo.targetType.raw": "FaceDetect"
+ }
+ },
+ {
+ "term":{
+ "baseInfo.targetId": "` + id + `"
+ }
+ }
+ ]
+ }
+ },
+ "size": 0,
+ "aggs": {
+ "buckets_aggs": {
+ "composite": {
+ "sources": [
+ {
+ "faceId": {
+ "terms": {
+ "field": "baseInfo.targetId"
+ }
+ }
+ },
+ {
+ "areaId": {
+ "terms": {
+ "field": "targetInfo.areaId"
+ }
+ }
+ }
+ ],
+ "size": 10000000
+ },
+ "aggs": {
+ "top_attention_hits": {
+ "top_hits": {
+ "size": 1000000,
+ "sort": [
+ {
+ "picDate": {
+ "order": "asc"
+ }
+ }
+ ],
+ "_source": {
+ "includes": [
+ "baseInfo.targetId",
+ "targetInfo.picSmUrl",
+ "targetInfo.areaId",
+ "picDate"
+ ]
+ }
+ }
+ }
+ }
+ }
+ }
+}`
+ buf, err := EsReq("POST", requestUrl, []byte(requestBody))
+ if err != nil {
+ return nil, err
+ }
+ source, err := FaceSourceAggregations(buf, thresholdTime, thresholdStayTime)
+ if err != nil {
+ return nil, err
+ }
+ if len(source) == 0 {
+ return source, nil
+ }
+ faceSource := make([]map[string]interface{}, 0)
+ for index, info := range source {
+ if int(info["stayTime"].(float64)) > thresholdStayTime {
+ faceSource = append(faceSource, source[index])
+ }
+ }
+ mapsSort := MapsSort{}
+ mapsSort.Key = "startTime"
+ mapsSort.MapList = faceSource
+ sort.Sort(&mapsSort)
+ return mapsSort.MapList, nil
+}
-//鏍规嵁鎶撴媿浜哄憳id鏇存柊锛坴ideourl锛夋憚鍍忔満鍦板潃
-func UpdateVideourlById(id string, videoUrl string, indexName string, serverIp string, serverPort string)(statu int){
- var info interface{}
- var videoUrlInfo = `
+func GetFaceIdDeduplication(startTime string, endTime string, serverIp string, serverPort string, indexName string) (ids []map[string]interface{}, err error) {
+ var requestUrl = "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ var requestBody = `{
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "range": {
+ "picDate": {
+ "gte": "` + startTime + `",
+ "lte": "` + endTime + `"
+ }
+ }
+ },
+ {
+ "term": {
+ "targetInfo.targetType.raw": "FaceDetect"
+ }
+ }
+ ]
+ }
+ },
+ "size": 0,
+ "aggs": {
+ "buckets_aggs": {
+ "composite": {
+ "sources": [
+ {
+ "faceId": {
+ "terms": {
+ "field": "baseInfo.targetId"
+ }
+ }
+ }
+ ],
+ "size": 10000000
+ },
+ "aggs": {
+ "top_attention_hits": {
+ "top_hits": {
+ "size": 1,
+ "sort": [
+ {
+ "picDate": {
+ "order": "desc"
+ }
+ }
+ ],
+ "_source": {
+ "includes": [
+ "picDate"
+ ]
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+}`
+ //fmt.Println(requestUrl)
+ //fmt.Println(requestBody)
+ buf, err := EsReq("POST", requestUrl, []byte(requestBody))
+ if err != nil {
+ return nil, err
+ }
+ ids, err1 := SourceDeduplication(buf)
+ if err1 != nil {
+ return nil, err1
+ }
+ if len(ids) > 1 {
+ mapsSort := MapsSort{}
+ mapsSort.Key = "lastTime"
+ mapsSort.MapList = ids
+ sort.Sort(&mapsSort)
+ return mapsSort.MapList, nil
+ }
+ return ids, nil
+}
+
+// 缁熻鍚勪釜鍖哄煙浜烘暟
+func StatisticsEveryAreaPersonsNumber(startTime string, endTime string, serverIp string, serverPort string, indexName string) ([]map[string]interface{}, error) {
+ var requestUrl = "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ var requestBody = `{
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "range": {
+ "picDate": {
+ "gte": "` + startTime + `",
+ "lte": "` + endTime + `"
+ }
+ }
+ },
+ {
+ "term": {
+ "targetInfo.targetType.raw": "Yolo"
+ }
+ }
+ ]
+ }
+ },
+ "size": 0,
+ "aggs": {
+ "buckets_aggs": {
+ "composite": {
+ "sources": [
+ {
+ "areaId": {
+ "terms": {
+ "field": "targetInfo.areaId"
+ }
+ }
+ }
+ ],
+ "size": 10000000
+ }
+ }
+ }
+}`
+ buf, err := EsReq("POST", requestUrl, []byte(requestBody))
+ if err != nil {
+ return nil, err
+ }
+ result, err := SourceStatistics(buf)
+ if err != nil {
+ return nil, err
+ }
+ return result, nil
+}
+
+/**************************************customer analysis util end**************************************/
+//鏍规嵁鎽勫儚鏈哄垪琛ㄥ拰鏃堕棿鏌ヨ浜哄憳娴忚杞ㄨ抗
+func GetPersonDataByCameraIdAndTime(cameraId []string, startTime string, endTime string, serverIp string, serverPort string, indexName string) (map[string]interface{}, error) {
+
+ var filterArr []string
+ if cameraId != nil && len(cameraId) > 0 {
+ esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1)
+ filterArr = append(filterArr, `{
+ "terms": {
+ "cameraId": ["`+esCameraId+`"]
+ }
+ }`)
+ }
+ filterArr = append(filterArr, `{
+ "range": {
+ "picDate": {
+ "gte": "`+startTime+`",
+ "lte": "`+endTime+`"
+ }
+ }
+ }`)
+ filterArr = append(filterArr, ` {
+ "term": {
+ "targetInfo.targetType.raw": "Yolo"
+ }
+ }`)
+ queryStr := strings.Join(filterArr, ",")
+
+ personUrl := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ personBody := `{
+ "query": {
+ "bool": {
+ "filter": [
+ ` + queryStr + `
+ ]
+ }
+ },
+ "size": 2147483647,
+ "_source": {
+ "includes": [
+ "cameraId",
+ "cameraName",
+ "cameraAddr",
+ "targetInfo.targetScore",
+ "picDate",
+ "updateTime",
+ "picMaxUrl",
+ "targetInfo.belongsTargetId",
+ "targetInfo.targetLocation",
+ "picWH"
+ ]
+ }
+}`
+ //fmt.Println(personUrl)
+ //fmt.Println(personBody)
+ source := make(map[string]interface{})
+ queryStartTime := time.Now()
+ buf, err := EsReq("POST", personUrl, []byte(personBody))
+ if err != nil {
+ return nil, err
+ }
+ queryUseTime := time.Now().Sub(queryStartTime).Seconds() * 1000
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return nil, err
+ }
+ resData, err := PerSonAnalysis(sources)
+ source["result"] = resData
+ source["total"] = len(resData)
+ source["queryUseTime"] = queryUseTime
+ //println(sources)
+ return source, nil
+
+}
+
+// 鏍规嵁鏃堕棿鑼冨洿锛屾憚鍍忔満鍒楄〃锛屽垎缁勮仛鍚堜汉鑴稿垪琛�,杩斿洖鍒嗙粍鏁版嵁
+func GetFaceDataBucketsByCameraIdAndTimeReturnByGrouped(cameraId []string, personId []string, startTime string, endTime string, thresholdTime float64, serverIp string, ServerPort string, indexName string) (buckersDate map[string]interface{}, err error) {
+ var filterArr []string
+ if cameraId != nil && len(cameraId) > 0 {
+ esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1)
+ filterArr = append(filterArr, `{
+ "terms": {
+ "cameraId": ["`+esCameraId+`"]
+ }
+ }`)
+ }
+ if personId != nil && len(personId) > 0 {
+ esPersonId := strings.Replace(strings.Trim(fmt.Sprint(personId), "[]"), " ", "\",\"", -1)
+ filterArr = append(filterArr, `{
+ "terms": {
+ "baseInfo.targetId": ["`+esPersonId+`"]
+ }
+ }`)
+ }
+ filterArr = append(filterArr, `{
+ "range": {
+ "picDate": {
+ "gte": "`+startTime+`",
+ "lte": "`+endTime+`"
+ }
+ }
+ }`)
+ filterArr = append(filterArr, ` {
+ "term": {
+ "targetInfo.targetType.raw": "FaceDetect"
+ }
+ }`)
+ queryStr := strings.Join(filterArr, ",")
+
+ var buckersUrl = "http://" + serverIp + ":" + ServerPort + "/" + indexName + "/_search"
+ var buckersBody = `{
+ "query": {
+ "bool": {
+ "filter": [
+ ` + queryStr + `
+ ]
+ }
+ },
+ "size": 0,
+ "aggs": {
+ "buckets_aggs": {
+ "composite": {
+ "sources": [
+ {
+ "baseInfo.targetId": {
+ "terms": {
+ "field": "baseInfo.targetId"
+ }
+ }
+ },
+ {
+ "cameraId": {
+ "terms": {
+ "field": "cameraId"
+ }
+ }
+ }
+ ],
+ "size": 10000000
+ },
+ "aggs":{
+ "top_attention_hits":{
+ "top_hits":{
+ "size": 1000000,
+ "sort": [
+ {
+ "picDate": {
+ "order": "asc"
+ }
+ }
+ ],
+ "_source":{
+ "includes":["baseInfo.targetId","cameraId","cameraName","cameraAddr","targetInfo.targetScore","targetInfo.picSmUrl","showLabels","baseInfo.tableId","baseInfo.tableName","baseInfo.bwType","baseInfo.targetName","baseInfo.compareScore","picDate","picMaxUrl","picWH"]
+ }
+ }
+ }
+ }
+ }
+ }
+}`
+ //fmt.Println(buckersUrl)
+ //fmt.Println(buckersBody)
+ sources := make(map[string]interface{})
+ queryStartTime := time.Now()
+ buf, err := EsReq("POST", buckersUrl, []byte(buckersBody))
+ if err != nil {
+ return nil, err
+ }
+ queryUseTime := time.Now().Sub(queryStartTime).Seconds() * 1000
+ //fmt.Println(queryUseTime)
+ tmpSources, err := SourceAggregationsReturnByGrouped(buf, thresholdTime)
+ if err != nil {
+ return nil, err
+ }
+ sources["result"] = tmpSources
+ sources["total"] = len(tmpSources)
+ sources["queryUseTime"] = queryUseTime
+ //println(sources)
+ return sources, nil
+}
+
+// 鏍规嵁鏃堕棿鑼冨洿锛屾憚鍍忔満鍒楄〃锛屽垎缁勮仛鍚堜汉鑴稿垪琛�
+func GetFaceDataBucketsByCameraIdAndTime(cameraId []string, personId []string, startTime string, endTime string, thresholdTime float64, serverIp string, ServerPort string, indexName string) (buckersDate map[string]interface{}, err error) {
+ var filterArr []string
+ if cameraId != nil && len(cameraId) > 0 {
+ esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1)
+ filterArr = append(filterArr, `{
+ "terms": {
+ "cameraId": ["`+esCameraId+`"]
+ }
+ }`)
+ }
+ if personId != nil && len(personId) > 0 {
+ esPersonId := strings.Replace(strings.Trim(fmt.Sprint(personId), "[]"), " ", "\",\"", -1)
+ filterArr = append(filterArr, `{
+ "terms": {
+ "baseInfo.targetId": ["`+esPersonId+`"]
+ }
+ }`)
+ }
+ filterArr = append(filterArr, `{
+ "range": {
+ "picDate": {
+ "gte": "`+startTime+`",
+ "lte": "`+endTime+`"
+ }
+ }
+ }`)
+ filterArr = append(filterArr, ` {
+ "term": {
+ "targetInfo.targetType.raw": "FaceDetect"
+ }
+ }`)
+ queryStr := strings.Join(filterArr, ",")
+
+ var buckersUrl = "http://" + serverIp + ":" + ServerPort + "/" + indexName + "/_search"
+ var buckersBody = `{
+ "query": {
+ "bool": {
+ "filter": [
+ ` + queryStr + `
+ ]
+ }
+ },
+ "size": 0,
+ "aggs": {
+ "buckets_aggs": {
+ "composite": {
+ "sources": [
+ {
+ "baseInfo.targetId": {
+ "terms": {
+ "field": "baseInfo.targetId"
+ }
+ }
+ },
+ {
+ "cameraId": {
+ "terms": {
+ "field": "cameraId"
+ }
+ }
+ }
+ ],
+ "size": 10000000
+ },
+ "aggs":{
+ "top_attention_hits":{
+ "top_hits":{
+ "size": 1000000,
+ "sort": [
+ {
+ "picDate": {
+ "order": "asc"
+ }
+ }
+ ],
+ "_source":{
+ "includes":["baseInfo.targetId","cameraId","cameraName","cameraAddr","targetInfo.targetScore","targetInfo.picSmUrl","showLabels","baseInfo.tableId","baseInfo.tableName","baseInfo.bwType","baseInfo.targetName","baseInfo.compareScore","picDate","picMaxUrl","picWH"]
+ }
+ }
+ }
+ }
+ }
+ }
+}`
+ //fmt.Println(buckersUrl)
+ //fmt.Println(buckersBody)
+ queryStartTime := time.Now()
+ buf, err := EsReq("POST", buckersUrl, []byte(buckersBody))
+ if err != nil {
+ return nil, err
+ }
+ queryUseTime := time.Now().Sub(queryStartTime).Seconds() * 1000
+
+ sources, err := SourceAggregations(buf, thresholdTime, queryUseTime)
+ if err != nil {
+ return nil, err
+ }
+ return sources, nil
+}
+
+// 鏍规嵁鎶撴媿浜哄憳id鏇存柊锛坧icurl锛夊浘鐗囧湴鍧�
+func UpdatePicUrlById(id string, picUrl string, indexName string, serverIp string, serverPort string) (err error) {
+ updateTime := time.Now().Format("2006-01-02 15:04:05")
+ tRes, err := AIOceaninfosbyid([]string{id}, indexName, serverIp, serverPort)
+ if err != nil || len(tRes) == 0 {
+ return err
+ }
+ picMaxUrls := tRes[0].PicMaxUrl
+ sourceStr := `
+ "source": "ctx._source.picMaxUrl.add('` + picUrl + `');ctx._source.updateTime='` + updateTime + `'"
+`
+ if len(picMaxUrls) >= 2 {
+ sourceStr = `"source": "ctx._source.picMaxUrl[1]='` + picUrl + `';ctx._source.updateTime='` + updateTime + `'"`
+ }
+ var info interface{}
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_update_by_query?refresh=true"
+
+ var picUrlInfo = `
{
"script": {
- "source": "ctx._source.videoUrl='` + videoUrl + `'"
+ ` + sourceStr + `
+ },
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "term": {
+ "id": "` + id + `"
+ }
+ }
+ ]
+ }
+ }
+ }
+ `
+ //logPrint("url: ", url, videoUrlInfo)
+ //fmt.Println(url, picUrlInfo)
+ buf, err := EsReq("POST", url, []byte(picUrlInfo))
+ if err != nil {
+ logPrint("http request videoUrlInfo info is err!")
+ return err
+ }
+ json.Unmarshal(buf, &info)
+ //logPrint(info)
+ out, ok := info.(map[string]interface{})
+ if !ok {
+ logPrint("http response interface can not change map[string]interface{}")
+ return errors.New("http response interface can not change map[string]interface{}")
+ }
+ middle, ok := out["updated"].(float64)
+ if !ok {
+ logPrint("first updated change error!", out)
+ return errors.New("first updated change error!")
+ }
+ if middle == 1 {
+ return nil
+ }
+ if middle == 0 {
+ return errors.New("宸茬粡淇敼")
+ }
+ return nil
+}
+
+// 鏍规嵁鎶撴媿浜哄憳id鏇存柊锛坴ideourl锛夋憚鍍忔満鍦板潃
+func UpdateVideourlById(id string, videoUrl string, indexName string, serverIp string, serverPort string, command int) (statu int, err error) {
+
+ var info interface{}
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_update_by_query?refresh=true"
+ sourceStr := "ctx._source.videoUrl='" + videoUrl + "'"
+ if command >= 0 {
+ sourceStr = "ctx._source.linkTagInfo[" + strconv.Itoa(command) + "].videoUrl='" + videoUrl + "'"
+ }
+ var videoUrlInfo = `
+ {
+ "script": {
+ "source": "` + sourceStr + `"
},
"query": {
"term": {
- "id": "` +id+ `"
+ "id": "` + id + `"
}
}
}
`
- buf, err := EsReq("POST", "http://"+serverIp+":"+serverPort+"/"+indexName+"/_update_by_query",[]byte(videoUrlInfo))
- if err != nil {
- fmt.Println("http request videoUrlInfo info is err!")
- statu = 500
- return
- }
- json.Unmarshal(buf, &info)
- out, ok := info.(map[string]interface{})
- if !ok {
- fmt.Println("http response interface can not change map[string]interface{}")
- statu = 500
- return
- }
+ //logPrint("url: ", url, videoUrlInfo)
+ buf, err := EsReq("POST", url, []byte(videoUrlInfo))
+ if err != nil {
+ logPrint("http request videoUrlInfo info is err!")
+ statu = 500
+ return statu, err
+ }
+ json.Unmarshal(buf, &info)
+ //logPrint(info)
+ out, ok := info.(map[string]interface{})
+ if !ok {
+ logPrint("http response interface can not change map[string]interface{}")
+ statu = 500
+ return statu, errors.New("http response interface can not change map[string]interface{}")
+ }
+ middle, ok := out["updated"].(float64)
+ batches, ok1 := out["batches"].(float64)
+ if !ok || !ok1 {
+ logPrint("first updated change error!")
+ statu = 500
+ return statu, errors.New("first updated change error!")
+ }
+ if batches == 0 {
+ logPrint("no such doc in database")
+ statu = 400
+ return statu, errors.New("鐩爣鏁版嵁涓嶅瓨鍦�")
+ } else {
+ if middle == 1 {
+ statu = 200
+ return statu, nil
+ }
+ if middle == 0 {
+ statu = 201
+ return statu, errors.New("宸茬粡淇敼")
+ }
+ }
+ return statu, nil
+}
- middle, ok := out["updated"].(float64)
- if !ok {
- fmt.Println("first updated change error!")
- statu = 500
- return
+// 鑾峰彇褰撳墠鑺傜偣鎶撴媿搴撴墍鏈変汉鍛業D*缂撳瓨*
+func GetAllLocalVideopersonsId(compareArgs protomsg.CompareArgs, indexName string, serverIp string, serverPort string, alarmLevelTypes string) (capturetable []string) {
+ queryStr := ""
+ queryBody := compareArgs.InputValue
+ //妫�绱㈡
+ if queryBody != "" {
+ queryStr = "\"must\":[{\"multi_match\":{\"query\":\"" + queryBody + "\",\"fields\":[\"cameraAddr^1.5\",\"taskName^1.5\",\"sdkName^1.5\",\"showLabels^3.0\",\"baseInfo.tableName^1.5\",\"baseInfo.targetName^1.5\",\"baseInfo.labels^1.5\",\"alarmRules.alarmLevel^1.5\",\"linkTag^1.5\"]," +
+ "\"type\":\"cross_fields\",\"operator\":\"OR\",\"slop\":0,\"prefix_length\":0,\"max_expansions\":50,\"zero_terms_query\":\"NONE\",\"auto_generate_synonyms_phrase_query\":true,\"fuzzy_transpositions\":true,\"boost\":1}}],"
+ }
+ if compareArgs.SearchTime == nil || len(compareArgs.SearchTime) != 2 {
+ return nil
+ }
+ gteDate := compareArgs.SearchTime[0]
+ lteDate := compareArgs.SearchTime[1]
+ //鍒ゆ柇浠诲姟ID
+ taskIdStr := ""
+ taskId := compareArgs.Tasks
+ if taskId != nil && len(taskId) > 0 {
+ esTaskId := strings.Replace(strings.Trim(fmt.Sprint(taskId), "[]"), " ", "\",\"", -1)
+ taskIdStr = "{\"terms\":{\"taskId\":[\"" + esTaskId + "\"]}},"
+ }
+ //鍒ゆ柇鎽勫儚鏈篒D
+ cameraIdStr := ""
+ cameraId := compareArgs.TreeNodes
+ if cameraId != nil && len(cameraId) > 0 {
+ esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1)
+ cameraIdStr = "{\"terms\":{\"cameraId\":[\"" + esCameraId + "\"]}},"
+ }
+
+ //鍒ゆ柇搴撹〃ID
+ tableId := compareArgs.Tabs
+ esTableId := ""
+ esTableIdStr := ""
+ if tableId != nil && len(tableId) > 0 {
+ esTableId = strings.Replace(strings.Trim(fmt.Sprint(tableId), "[]"), " ", "\",\"", -1)
+ esTableIdStr = "{\"terms\":{\"baseInfo.tableId\":[\"" + esTableId + "\"]}},"
+ }
+ isCollectStr := ""
+ isCollect := compareArgs.Collection
+ if isCollect != "" {
+ //isCollectStr = "{\"term\":{\"isCollect\":\"" + isCollect + "\"}},"
+ if isCollect == "1" {
+ isCollectStr = "{\"term\":{\"isCollect\":true}},"
+ } else if isCollect == "0" {
+ isCollectStr = "{\"term\":{\"isCollect\":false}},"
+ }
+ }
+
+ //鍒ゆ柇甯冮槻绛夌骇
+ alarmLevelStr := ""
+ if alarmLevelTypes != "" {
+ alarmLevelStr = "{\"terms\":{\"alarmRules.alarmLevel.raw\":[\"" + alarmLevelTypes + "\"]}},"
+ }
+
+ //浣跨敤es搴曞眰鏈哄埗澶勭悊鍒嗛〉
+
+ analyServerFilterStr := ""
+ analyServerId := compareArgs.AnalyServerId
+ if analyServerId != "" {
+ analyServerFilterStr = "{\"term\":{\"analyServerId\":\"" + analyServerId + "\"}},"
+ }
+
+ ts := time.Now()
+ //棣栨璇锋眰澶�
+ url := "http://" + serverIp + ":" + serverPort +
+ "/" + indexName + "/_search?search_type=dfs_query_then_fetch;scroll=1m"
+
+ var lock sync.RWMutex
+ var wg sync.WaitGroup
+
+ for i := 0; i < 48; i++ {
+ //璇锋眰浣�
+ prama := "{" +
+ "\"slice\":{\"id\":" + strconv.Itoa(i) + ",\"max\":48}," +
+ "\"size\":\"1000\"," +
+ "\"query\":{\"bool\":{" + queryStr +
+ "\"filter\":[" +
+ "{\"term\":{\"targetInfo.targetType.raw\":\"FaceDetect\"}}," +
+ cameraIdStr +
+ alarmLevelStr +
+ taskIdStr +
+ isCollectStr +
+ esTableIdStr +
+ analyServerFilterStr +
+ "{\"range\":{\"picDate\":{\"from\":\"" + gteDate + "\",\"to\":\"" + lteDate + "\",\"include_lower\":true,\"include_upper\":true,\"boost\":1}}}]}}," +
+ "\"_source\":[\"id\"]" +
+ "}"
+ wg.Add(1)
+ go func(reqParam string) {
+ defer wg.Done()
+
+ logPrint(url)
+ logPrint(prama)
+ buf, err := EsReq("POST", url, []byte(reqParam))
+
+ if err != nil {
+ logPrint("http request videoUrlInfo info is err!")
+ logPrint(len(capturetable))
+ return
+ }
+
+ sources, err := Sourcelistforscroll(buf)
+
+ if err != nil {
+ logPrint(len(capturetable))
+ return
+ }
+ for _, source := range sources["sourcelist"].([]map[string]interface{}) {
+ lock.Lock()
+ capturetable = append(capturetable, source["id"].(string))
+ lock.Unlock()
+ }
+
+ scroll_id := sources["scroll_id"].(string)
+
+ //scroll璇锋眰澶�
+ scroll_url := "http://" + serverIp + ":" + serverPort + "/_search/scroll"
+
+ for {
+ var tmpList []string
+ next_scroll_id := ""
+ if next_scroll_id != "" {
+ scroll_id = next_scroll_id
+ }
+ jsonDSL := `{
+ "scroll": "1m",
+ "scroll_id" : "` + scroll_id + `"
+ }`
+ logPrint(scroll_url)
+ logPrint(jsonDSL)
+ buf, err := EsReq("POST", scroll_url, []byte(jsonDSL))
+
+ if err != nil {
+ logPrint("lenth1: ", len(capturetable))
+ return
+ }
+ nextSources, err := Sourcelistforscroll(buf)
+
+ if nextSources == nil {
+ return
+ }
+
+ nextM := nextSources["sourcelist"].([]map[string]interface{})
+ //logPrint("id",nextSources)
+ if nextM == nil || len(nextM) == 0 {
+ //logPrint("lenth: ", len(capturetable))
+ return
+ }
+ //logPrint("id")
+ for _, source := range nextM {
+ tmpList = append(tmpList, source["id"].(string))
+ }
+ //logPrint("tmpList: ", len(tmpList))
+ lock.Lock()
+ capturetable = append(capturetable, tmpList...)
+ lock.Unlock()
+
+ next_scroll_id = nextSources["scroll_id"].(string)
+ }
+
+ }(prama)
+ }
+ wg.Wait()
+
+ logPrint("lenth_all: ", len(capturetable))
+ logPrint("鑰楁椂锛�", time.Since(ts))
+ return capturetable
+}
+
+// 鍒濆鍖栧疄鏃舵姄鎷�
+func InitRealTimeCapture(serverIp string, serverPort string, indexName string, isAlarm string, category string, quantity int) ([]protomsg.AIOcean, error) {
+ var aIOceanInfo []protomsg.AIOcean
+ url := "http://" + serverIp + ":" + serverPort +
+ "/" + indexName + "/_search"
+ var filterArr []string
+ if isAlarm != "all" {
+ filterArr = append(filterArr, ` {
+ "term":{
+ "isAlarm":"`+isAlarm+`"
+ }
+ }`)
+ }
+
+ if category != "all" {
+ filterArr = append(filterArr, ` {
+ "term":{
+ "targetInfo.targetType.raw":"`+category+`"
+ }
+ }`)
+
+ }
+
+ queryStr := `"query":{
+ "bool":{
+ "filter":[
+ ` + strings.Join(filterArr, ",") + `
+ ]
+ }
+ },`
+
+ DSLJson := `{
+ "size":` + strconv.Itoa(quantity) + `,
+ ` + queryStr + `
+ "sort":[{"picDate":{"order":"desc"}}],
+ "_source": {"includes":[],"excludes":["*.feature"]}
+ }`
+ //logPrint(DSLJson)
+ buf, err := EsReq("POST", url, []byte(DSLJson))
+ if err != nil {
+ return aIOceanInfo, err
+ }
+
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return aIOceanInfo, err
+ }
+
+ aIOcean := AIOceanAnalysis(sources)
+ //logPrint(len(videoperson))
+ return aIOcean, nil
+}
+
+// 瀹炴椂鎶撴媿
+func RealTimeCapture(serverIp string, serverPort string, indexName string, isAlarm bool) ([]protomsg.AIOcean, error) {
+ var aIOceanInfo []protomsg.AIOcean
+ url := "http://" + serverIp + ":" + serverPort +
+ "/" + indexName + "/_search"
+ DSLJson := `{
+ "size":20,
+ "query":{
+ "bool":{
+ "filter":[
+ {
+ "range":{
+ "picDate":{
+ "gte":"now+8h-30s",
+ "lt":"now+8h"
+ }
+ }
+ },
+ {
+ "term":{
+ "isAlarm":` + strconv.FormatBool(isAlarm) + `
+ }
+ }
+ ]
+ }
+ },
+ "_source": {"includes":[],"excludes":["*.feature"]}
+ }`
+
+ buf, err := EsReq("POST", url, []byte(DSLJson))
+ if err != nil {
+ return aIOceanInfo, err
+ }
+
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return aIOceanInfo, err
+ }
+
+ aIOcean := AIOceanAnalysis(sources)
+ logPrint(len(aIOcean))
+ return aIOcean, nil
+}
+
+// 缁煎悎缁熻
+func StatisticsComprehensive(serverIp string, serverPort string, indexName string, isAlarm string) (total int, err error) {
+ url := "http://" + serverIp + ":" + serverPort +
+ "/" + indexName + "/_search"
+ isAlarmStr := ""
+ if isAlarm != "all" {
+ isAlarmStr = ` {
+ "term":{
+ "isAlarm":"` + isAlarm + `"
+ }
+ },`
+
+ }
+ DSLJson := `{
+ "size":0,
+ "query":{
+ "bool":{
+ "filter":[
+ ` + isAlarmStr + `
+ {
+ "range":{
+ "picDate":{
+ "gte":"now+8h/d"
+ }
+ }
+ }
+ ]
+ }
+ }
+ }`
+ buf, err := EsReq("POST", url, []byte(DSLJson))
+ if err != nil {
+ return total, err
+ }
+ var info interface{}
+ json.Unmarshal(buf, &info)
+ out, ok := info.(map[string]interface{})
+ if !ok {
+ return total, errors.New("http response interface can not change map[string]interface{}")
+ }
+ middle, ok := out["hits"].(map[string]interface{})
+ if !ok {
+ return total, errors.New("first hits change error!")
+ }
+ total = int(middle["total"].(float64))
+ //logPrint(total)
+ return total, nil
+}
+
+// 瀹炴椂鎶ヨ浠诲姟姣旂巼
+func RealTimeAlarmTaskRate(serverIp string, serverPort string, indexName string) (sources []map[string]interface{}, err error) {
+ url := "http://" + serverIp + ":" + serverPort +
+ "/" + indexName + "/_search"
+ DSLJson := `{
+ "size":0,
+ "query":{
+ "bool":{
+ "filter":[{
+ "range":{
+ "picDate":{
+ "gte":"now+8h/d"
+ }
+ }
+ }]
+ }
+ },
+ "aggs":{
+ "taskName_status":{
+ "terms":{
+ "field":"taskName.raw"
+ }
+ }
+ }
+}`
+ buf, err := EsReq("POST", url, []byte(DSLJson))
+ if err != nil {
+ return nil, err
+ }
+ var info interface{}
+ json.Unmarshal(buf, &info)
+ out, ok := info.(map[string]interface{})
+ if !ok {
+ return nil, errors.New("http response interface can not change map[string]interface{}")
+ }
+ middle, ok := out["aggregations"].(map[string]interface{})
+ if !ok {
+ return nil, errors.New("first hits change error!")
+ }
+ sdkName_status, ok := middle["taskName_status"].(map[string]interface{})
+ if !ok {
+ return nil, errors.New("first hits change error!")
+ }
+ //fmt.Println(sdkName_status)
+ for _, in := range sdkName_status["buckets"].([]interface{}) {
+ var source = make(map[string]interface{}, 0)
+ tmpbuf, ok := in.(map[string]interface{})
+ if !ok {
+ logPrint("change to source error!")
+ continue
+ }
+ sdkName := tmpbuf["key"].(string)
+ count := int(tmpbuf["doc_count"].(float64))
+ source["name"] = sdkName
+ source["value"] = count
+ sources = append(sources, source)
+ }
+ //logPrint("tmpSource",sources)
+ return sources, nil
+}
+
+// 鑱氬悎浠诲姟鍒楄〃锛宼askId+taskName
+func AggregateTaskList(serverIp string, serverPort string, indexName string, analyServerId string, cameraIds []string) (sources []map[string]interface{}, err error) {
+ url := "http://" + serverIp + ":" + serverPort +
+ "/" + indexName + "/_search"
+ serverFilterStr := ""
+ cameIdFilterStr := ""
+ if cameraIds != nil && len(cameraIds) > 0 {
+ cameIdsStr := strings.Replace(strings.Trim(fmt.Sprint(cameraIds), "[]"), " ", "\",\"", -1)
+ cameIdFilterStr = `,{
+ "term": {
+ "cameraId": "` + cameIdsStr + `"
+ }
+ }`
+ }
+ if analyServerId != "" {
+ serverFilterStr = `,
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "term": {
+ "analyServerId": "` + analyServerId + `"
+ }
+ }
+ ` + cameIdFilterStr + `
+ ]
+ }
+ }`
+ }
+ DSLJson := `{
+ "size": 0,
+ "aggs": {
+ "task_status": {
+ "composite": {
+ "sources": [
+ {
+ "taskId": {
+ "terms": {
+ "field": "taskId"
+ }
+ }
+ },
+ {
+ "taskName": {
+ "terms": {
+ "field": "taskName.raw"
+ }
+ }
+ }
+ ],
+ "size":"1000000"
+ }
+ }
}
- if middle == 1{
- statu = 200
- return
+ ` + serverFilterStr + `
+}`
+ buf, err := EsReq("POST", url, []byte(DSLJson))
+ if err != nil {
+ return nil, err
+ }
+ var info interface{}
+ json.Unmarshal(buf, &info)
+ out, ok := info.(map[string]interface{})
+ if !ok {
+ return nil, errors.New("http response interface can not change map[string]interface{}")
+ }
+ middle, ok := out["aggregations"].(map[string]interface{})
+ if !ok {
+ return nil, errors.New("first hits change error!")
+ }
+ task_status, ok := middle["task_status"].(map[string]interface{})
+ if !ok {
+ return nil, errors.New("first hits change error!")
+ }
+
+ for _, in := range task_status["buckets"].([]interface{}) {
+ var source = make(map[string]interface{}, 0)
+ tmpbuf, ok := in.(map[string]interface{})
+ if !ok {
+ logPrint("change to source error!")
+ continue
+ }
+ task := tmpbuf["key"].(map[string]interface{})
+ count := int(tmpbuf["doc_count"].(float64))
+ taskName := task["taskName"].(string)
+ taskId := task["taskId"].(string)
+ source["taskName"] = taskName
+ source["taskId"] = taskId
+ source["count"] = count
+ sources = append(sources, source)
+ }
+ //logPrint("tmpSource",sources)
+ return sources, nil
+
+}
+
+// 娣诲姞鍗冲皢鍒犻櫎淇″彿
+func AddDeleteSignal() {
+
+}
+
+/****************************************浠ヤ笅涓簊dkCompare姣斿缂撳瓨浣跨敤鏂规硶*********************************************/
+//鑾峰彇鏌ヨ鎬绘暟 *缂撳瓨*
+func GetTotal(serverIp string, serverPort string, indexName string, shards string, targetType string) (total int) {
+ JsonDSL := `{
+ "size": 0,
+ "query": {
+ "bool": {
+ "filter": [{
+ "term": {
+ "targetInfo.targetType.raw": "` + targetType + `"
+ }
+ }]
+ }
+ }
+}`
+
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search?preference=_shards:" + shards + "|_only_local"
+ buf, err := EsReq("POST", url, []byte(JsonDSL))
+ if err != nil {
+ return
+ }
+ var info interface{}
+ json.Unmarshal(buf, &info)
+ out, ok := info.(map[string]interface{})
+ if !ok {
+ return
+ }
+ middle, ok := out["hits"].(map[string]interface{})
+ if !ok {
+ return
+ }
+ total = int(middle["total"].(float64))
+ return total
+
+}
+
+// 鏌ヨ鏃堕棿娈垫暟鎹� *缂撳瓨*
+func GetPeriodInfos(serverIp string, serverPort string, startTime string, endTime string, indexName string, shards string, targetType string) ([]*protomsg.MultiFeaCache, error) {
+ var capdbinfo []*protomsg.MultiFeaCache
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search?preference=_shards:" + shards + "|_only_local"
+ var source []string
+ switch targetType {
+ case "face", "FaceDetect":
+ source = []string{"id", "targetInfo.feature", "analyServerId", "cameraId"}
+ case "track":
+ source = []string{"id", "targetInfo.feature", "analyServerId", "cameraId", "targetInfo.attachTarget.feature", "targetInfo.targetLocation", "linkTagInfo.targetInfo.feature", "linkTagInfo.targetInfo.attachTarget.feature", "linkTagInfo.cameraId", "linkTagInfo.targetInfo.targetLocation"}
+ }
+ JsonDSL := `
+ {
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "term": {
+ "targetInfo.targetType.raw": "` + targetType + `"
+ }
+ },
+ {
+ "range": {
+ "picDate": {
+ "gte": "` + startTime + `",
+ "lt": "` + endTime + `"
+ }
+ }
+ }
+ ]
+ }
+ },
+ "size": 1000000,
+ "_source": ["` + strings.Replace(strings.Trim(fmt.Sprint(source), "[]"), " ", "\",\"", -1) + `"]
+ }
+ `
+ //logger.Debug(url)
+ //logger.Debug(JsonDSL)
+ //logPrint(JsonDSL)
+ buf, err := EsReq("POST", url, []byte(JsonDSL))
+ if err != nil {
+ return capdbinfo, errors.New("http request dbtablename info is err!")
+ }
+
+ // 杩斿洖 _source 鏁扮粍
+ sources, err := Sourcelist(buf)
+ if err != nil {
+ return capdbinfo, err
+ }
+ //logPrint(sources)
+ // 杩斿洖鎵�鏈夋煡璇㈢殑鏁版嵁
+ capdbinfos := Parsesources(sources)
+ return capdbinfos, nil
+}
+
+// 鏌ヨ搴曞簱浜哄憳淇℃伅*缂撳瓨*
+func GetOceanFeatures(serverIp string, serverPort string, queryNums int, indexName string, shards string, targetType string) ([]*protomsg.MultiFeaCache, error) {
+ //queryIndexNum int
+ //var dbinfos []*protomsg.MultiFeaCache
+ dbinfos := make([]*protomsg.MultiFeaCache, 0)
+ //dbinfosss := make([]*protomsg.MultiFeaCache,0)
+ //dbinfoss = append(dbinfoss, dbinfosss...)
+
+ JsonDSL := ""
+ var source []string
+ switch targetType {
+ case "face", "FaceDetect":
+ source = []string{"id", "targetInfo.feature", "analyServerId"}
+ case "track":
+ source = []string{"id", "targetInfo.feature", "analyServerId", "targetInfo.attachTarget.feature", "targetInfo.targetLocation", "linkTagInfo.targetInfo.feature", "linkTagInfo.targetInfo.attachTarget.feature", "linkTagInfo.targetInfo.targetLocation"}
+ }
+
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search?preference=_shards:" + shards + "|_only_local;scroll=1m"
+
+ var lock sync.RWMutex
+ var wg sync.WaitGroup
+
+ for i := 0; i < 48; i++ {
+ //璇锋眰浣�
+ JsonDSL = ` {
+ "slice": {
+ "id": "` + strconv.Itoa(i) + `",
+ "max": 48
+ },
+ "size":` + strconv.Itoa(queryNums) + `,
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "term": {
+ "targetInfo.targetType.raw": "` + targetType + `"
+ }
+ }
+ ]
+ }
+ },
+ "_source": ["` + strings.Replace(strings.Trim(fmt.Sprint(source), "[]"), " ", "\",\"", -1) + `"]
+ }`
+ wg.Add(1)
+ go func(reqJsonDSL string) {
+ defer wg.Done()
+
+ //fmt.Println(url)
+ //fmt.Println(prama)
+ //logPrint("url: ",url)
+ //logPrint("url: ",reqJsonDSL)
+ buf, err := EsReq("POST", url, []byte(reqJsonDSL))
+ if err != nil {
+ logPrint("EsReq: ", err)
+ return
+ }
+
+ // 杩斿洖 _source 鏁扮粍
+ sources, err := Sourcelistforscroll(buf)
+ if err != nil {
+ logPrint("EsReq: ", err)
+ return
+ }
+ // 杩斿洖鎵�鏈夋煡璇㈢殑鏁版嵁
+ ftmpDatas := Parsesources(sources["sourcelist"].([]map[string]interface{}))
+ lock.Lock()
+ dbinfos = append(dbinfos, ftmpDatas...)
+ //logPrint("prsLen: ", len(Parsesources(sources["sourcelist"].([]map[string]interface{}))))
+ //logPrint("dbinfosLen: ", len(dbinfos))
+ lock.Unlock()
+
+ scroll_id := sources["scroll_id"].(string)
+
+ //scroll璇锋眰澶�
+ scroll_url := "http://" + serverIp + ":" + serverPort + "/_search/scroll"
+ for {
+ next_scroll_id := ""
+ if next_scroll_id != "" {
+ scroll_id = next_scroll_id
+ }
+ jsonDSL := `{
+ "scroll": "1m",
+ "scroll_id" : "` + scroll_id + `"
+ }`
+ //fmt.Println(scroll_url)
+ //fmt.Println(jsonDSL)
+ buf, err := EsReq("POST", scroll_url, []byte(jsonDSL))
+
+ if err != nil {
+ //fmt.Println("lenth1: ", len(dbinfos))
+ return
+ }
+ nextSources, err := Sourcelistforscroll(buf)
+
+ if nextSources == nil {
+ return
+ }
+
+ nextM := nextSources["sourcelist"].([]map[string]interface{})
+ //fmt.Println("id",nextSources)
+ if nextM == nil || len(nextM) == 0 {
+ //fmt.Println("lenth: ", len(capturetable))
+ return
+ }
+ tmpDatas := Parsesources(nextM)
+ lock.Lock()
+ dbinfos = append(dbinfos, tmpDatas...)
+ //logPrint("tmpDatasLen: ", len(tmpDatas))
+ //logPrint("AdbinfosLen: ", len(dbinfos))
+ lock.Unlock()
+
+ next_scroll_id = nextSources["scroll_id"].(string)
+ }
+
+ }(JsonDSL)
+ }
+ wg.Wait()
+
+ //fmt.Println("lenth_all: ", len(dbinfos))
+
+ return dbinfos, nil
+}
+
+// ************************CORN TASK*******************************
+// 鏌ヨ鏃ユ湡鑼冨洿鍐呮槸鍚﹁繕瀛樺湪鏁版嵁
+func QueryAnalyServerData(serverIp string, serverPort string, indexName string, startTime string, endTime string, analyServerId string) (result bool, err error) {
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+ deleteJson := `{
+ "query":{
+ "bool":{
+ "filter":[{
+ "range":{
+ "updateTime":{
+ "gte":"` + startTime + `",
+ "lte":"` + endTime + `"
+ }
+ }
+ },
+ {
+ "term":{
+ "analyServerId":"` + analyServerId + `"
+ }
+ }
+ ]
+ }
+ }
+} `
+ buf, err := EsReq("POST", url, []byte(deleteJson))
+ if err != nil {
+ return false, errors.New("璇锋眰澶辫触")
+ }
+ resTotal, err := SourceTotal(buf)
+ if err != nil {
+ return false, errors.New("瑙ｇ爜澶辫触")
+ }
+ if resTotal == -1 || resTotal == 0 {
+ result = false
+ } else {
+ result = true
+ }
+ return result, nil
+}
+
+// 鎸夋棩鏈熻寖鍥达紝鏈嶅姟鍣↖d鍒犻櫎鏁版嵁
+func DeleteByDocumentNumber(docNumber []string, serverIp string, serverPort string, indexName string) (total int, err error) {
+
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_delete_by_query"
+ docNumbers := strings.Replace(strings.Trim(fmt.Sprint(docNumber), "[]"), " ", "\",\"", -1)
+ deleteJson := `{
+ "query":{
+ "bool":{
+ "filter":[
+ {
+ "terms":{
+ "documentNumber":["` + docNumbers + `"]
+ }
+ }
+ ]
+ }
+ }
+} `
+ //fmt.Println(url)
+ //fmt.Println(deleteJson)
+ //return
+ buf, err := EsReq("POST", url, []byte(deleteJson))
+ if err != nil {
+ return -1, errors.New("璇锋眰澶辫触")
+ }
+ deleteRes, err := SourceDeleted(buf)
+ if err != nil {
+ return -1, errors.New("瑙ｇ爜澶辫触")
+ }
+ return deleteRes, nil
+}
+
+//func GetCaptureDaysByDocumentNumber(docNumber string, comId string, indexName string, serverIp string, serverPort string){
+// url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_search"
+// queryDSL := ``
+//}
+
+// 鎸夋棩鏈熻寖鍥达紝鏈嶅姟鍣↖d鍒犻櫎鏁版嵁
+func DeleteAnalyServerData(serverIp string, serverPort string, indexName string, startTime string, endTime string, analyServerId string) (total int, err error) {
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_delete_by_query"
+ deleteJson := `{
+ "query":{
+ "bool":{
+ "filter":[{
+ "range":{
+ "updateTime":{
+ "gte":"` + startTime + `",
+ "lte":"` + endTime + `"
+ }
+ }
+ },
+ {
+ "term":{
+ "analyServerId":"` + analyServerId + `"
+ }
+ }
+ ]
+ }
+ }
+} `
+ fmt.Println(url)
+ fmt.Println(deleteJson)
+ buf, err := EsReq("POST", url, []byte(deleteJson))
+ if err != nil {
+ return -1, errors.New("璇锋眰澶辫触")
+ }
+ deleteRes, err := SourceDeleted(buf)
+ if err != nil {
+ return -1, errors.New("瑙ｇ爜澶辫触")
+ }
+ return deleteRes, nil
+}
+
+// 缁欐墍鏈夎妭鐐硅拷鍔犲垹闄や换鍔′俊鎭�
+func AddDelTask(serverIp string, serverPort string, indexName string, startTime string, endTime string, analyServerId string) (result bool, err error) {
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_update_by_query"
+ addJson := `{
+ "script": {
+ "lang":"painless",
+ "inline": "ctx._source.instantTask.add(params.newtask)",
+ "params": {
+ "newtask": {
+ "instantClearId": "` + analyServerId + `",
+ "startTime": "` + startTime + `",
+ "endTime": "` + endTime + `"
+ }
+ }
+ },
+ "query": {
+ "bool": {
+ "filter": [
+ {
+ "term": {
+ "application": "loopCoverage"
+ }
+ }
+ ]
+ }
}
- if middle == 0{
- statu = 201
- return
+}`
+ buf, err := EsReq("POST", url, []byte(addJson))
+ if err != nil {
+ return false, errors.New("璇锋眰澶辫触")
+ }
+ updateRes, err := SourceUpdated(buf)
+ if err != nil {
+ return false, errors.New("瑙ｇ爜澶辫触")
+ }
+ if updateRes == -1 {
+ result = false
+ } else {
+ result = true
+ }
+ return result, nil
+}
+
+// 绉婚櫎宸叉墽琛屽畬鐨勫垹闄や换鍔�
+func DeleteDelTask(serverIp string, serverPort string, indexName string, analyServerId string) (result bool, err error) {
+ url := "http://" + serverIp + ":" + serverPort + "/" + indexName + "/_update_by_query"
+ deleteJson := `{
+ "script": {
+ "lang":"painless",
+ "inline": "ctx._source.instantTask.remove(0)"
+ },
+ "query": {
+ "bool": {
+ "filter":[{
+ "term":{
+ "id":"` + analyServerId + `"
+ }
+ }]
+ }
}
- return statu
-}
\ No newline at end of file
+}`
+ buf, err := EsReq("POST", url, []byte(deleteJson))
+ if err != nil {
+ return false, errors.New("璇锋眰澶辫触")
+ }
+ updateRes, err := SourceUpdated(buf)
+ if err != nil {
+ return false, errors.New("瑙ｇ爜澶辫触")
+ }
+ if updateRes == -1 {
+ result = false
+ } else {
+ result = true
+ }
+ return result, nil
+}
+
+type ShardInfo struct {
+ ShardIndex string `json:"shardIndex"` //鍒嗙墖鎵�灞炵储寮曞悕绉�
+ ShardNum int `json:"shardNum"` //鍒嗙墖鍙�
+ ShardRole string `json:"shardRole"` //鍒嗙墖瑙掕壊(涓诲垎鐗囷細primary 鍓湰鍒嗙墖锛歳eplica)
+ ShardState string `json:"shardState"` //鍒嗙墖鐘舵��(鍚敤锛歋TARTED 鏈惎鐢細UNASSIGNED)
+ ShardDocs int `json:"shardDocs"` //鍒嗙墖宸蹭繚瀛樻枃妗ｆ暟
+ ShardStore string `json:"shardStore"` //鍒嗙墖褰撳墠瀛樺偍鏁版嵁澶у皬
+ ShardIp string `json:"shardIp"` //鍒嗙墖鎵�鍦ㄨ妭鐐筰p
+ ShardNode string `json:"shardNode"` //鍒嗙墖鎵�鍦ㄨ妭鐐瑰悕绉�
+}
+
+// 鑾峰彇绱㈠紩鍒嗙墖淇℃伅
+func GetShardsByIndex(serverIp string, serverPort string, indexName string) ([]ShardInfo, error) {
+ url := "http://" + serverIp + ":" + serverPort + "/_cat/shards?v"
+ buf, err := EsReq("GET", url, []byte(""))
+ if err != nil {
+ return nil, err
+ }
+ var inf = []ShardInfo{}
+ res := strings.Split(string(buf), "\n")[1:]
+ for _, r := range res {
+ if r != "" {
+
+ inx := strings.Fields(r)
+ index := inx[0]
+ shard, _ := strconv.Atoi(inx[1])
+ prired := inx[2]
+ if prired == "r" {
+ prired = "replica"
+ }
+ if prired == "p" {
+ prired = "primary"
+ }
+ state := inx[3]
+ docs := 0
+ store := ""
+ ip := ""
+ node := ""
+ if state == "STARTED" {
+ docs, _ = strconv.Atoi(inx[4])
+ store = inx[5]
+ ip = inx[6]
+ node = inx[7]
+ }
+ if index == indexName {
+ inf = append(inf, ShardInfo{
+ ShardIndex: index,
+ ShardNum: shard,
+ ShardRole: prired,
+ ShardState: state,
+ ShardDocs: docs,
+ ShardStore: store,
+ ShardIp: ip,
+ ShardNode: node,
+ })
+
+ }
+ }
+
+ }
+ return inf, nil
+}
--
Gitblit v1.8.0