| | |
| | | |
| | | func queryEsLocation(esClient *elasticsearch.Client, locationModel *LocationModel, documentNumbers []string) ([]*LocationRecord, error) { |
| | | var buf bytes.Buffer |
| | | nowTime := time.Now() |
| | | startTime := nowTime.Add(-time.Duration(locationModel.Duration) * 24 * time.Hour) |
| | | //nowTime := time.Now() |
| | | //startTime := nowTime.Add(-time.Duration(locationModel.Duration) * 24 * time.Hour) |
| | | |
| | | // 构建过滤条件 |
| | | var filters []map[string]interface{} |
| | |
| | | } |
| | | filters = append(filters, addrParams) |
| | | } |
| | | |
| | | //// 重点人员过滤 |
| | | //if len(locationModel.KeyPersonType) > 0 { |
| | | //var queryTimes []map[string]interface{} |
| | | //for date := startTime; date.Before(nowTime); date = date.Add(24 * time.Hour) { |
| | | // start := time.Date(date.Year(), date.Month(), date.Day(), locationModel.StartTime, 0, 0, 0, date.Location()) |
| | | // end := time.Date(date.Year(), date.Month(), date.Day(), locationModel.EndTime, 0, 0, 0, date.Location()) |
| | | // |
| | | // //queryTimes = append(queryTimes, map[string]interface{}{ |
| | | // // "range": map[string]interface{}{ |
| | | // // "picDate": map[string]interface{}{ |
| | | // // "gte": start.Format(time.DateTime), // "2006-01-02 15:04:05d" |
| | | // // "lte": end.Format(time.DateTime), |
| | | // // }, |
| | | // // }, |
| | | // //}) |
| | | // |
| | | // filters = append(filters, map[string]interface{}{ |
| | | // "terms": map[string]interface{}{ |
| | | // "keyPersonType": strings.Split(locationModel.KeyPersonType, ","), |
| | | // "range": map[string]interface{}{ |
| | | // "picDate": map[string]interface{}{ |
| | | // "gte": start.Format(time.DateTime), |
| | | // "lte": end.Format(time.DateTime), |
| | | // }, |
| | | // }, |
| | | // }) |
| | | //} |
| | | |
| | | // 时间范围 |
| | | //filters = append(filters, map[string]interface{}{ |
| | | // "range": map[string]interface{}{ |
| | | // "picDate": map[string]interface{}{ |
| | | // "gte": start.Format(time.DateTime), |
| | | // "lt": now.Format(time.DateTime), |
| | | // }, |
| | | // "bool": map[string]interface{}{ |
| | | // "filter": queryTimes, |
| | | // }, |
| | | //}) |
| | | var queryTimes []map[string]interface{} |
| | | for date := startTime; date.Before(nowTime); date = date.Add(24 * time.Hour) { |
| | | start := time.Date(date.Year(), date.Month(), date.Day(), locationModel.StartTime, 0, 0, 0, date.Location()) |
| | | end := time.Date(date.Year(), date.Month(), date.Day(), locationModel.EndTime, 0, 0, 0, date.Location()) |
| | | |
| | | queryTimes = append(queryTimes, map[string]interface{}{ |
| | | "range": map[string]interface{}{ |
| | | "picDate": map[string]interface{}{ |
| | | "gte": start.Format(time.DateTime), // "2006-01-02 15:04:05d" |
| | | "lte": end.Format(time.DateTime), |
| | | }, |
| | | }, |
| | | }) |
| | | } |
| | | filters = append(filters, map[string]interface{}{ |
| | | "bool": map[string]interface{}{ |
| | | "should": queryTimes, |
| | | }, |
| | | }) |
| | | logger.Debugf("filters--------------------------------------- %s", filters) |
| | | query := map[string]interface{}{ |
| | | "query": map[string]interface{}{ |
| | |
| | | if err := json.NewDecoder(res.Body).Decode(&result); err != nil { |
| | | return nil, fmt.Errorf("error parsing response body: %s", err) |
| | | } |
| | | logger.Debugf("result--------------------------------------- %s", result) |
| | | // 解析聚合结果 |
| | | var records []*LocationRecord |
| | | if aggs, ok := result["aggregations"].(map[string]interface{}); ok { |