aboutsummaryrefslogtreecommitdiffstats
path: root/src/main/java/org/onap/aai/sa/searchdbabstraction/searchapi/DateRangeAggregation.java
blob: 1ee42c9fde1ae668cd48a90edd38b039b8b4f9c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
 * Copyright © 2017-2018 Amdocs
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */
package org.onap.aai.sa.searchdbabstraction.searchapi;

import com.fasterxml.jackson.annotation.JsonProperty;

/**
 * An example of a date_range aggregation:
 *
 * <p><pre>
 * {
 *    "aggs": {
 *        "range": {
 *            "date_range": {
 *                "field": "date",
 *                "format": "MM-yyy",
 *                "ranges": [
 *                    { "to": "now-10M/M" },
 *                    { "from": "now-10M/M" }
 *                ]
 *            }
 *        }
 *    }
 * }
 * </pre>
 *
 * @author sye
 */
public class DateRangeAggregation extends AbstractAggregation {


  private String format;

  @JsonProperty("ranges")
  private DateRange[] dateRanges;


  public String getFormat() {
    return format;
  }

  public void setFormat(String format) {
    this.format = format;
  }

  public DateRange[] getDateRanges() {
    return dateRanges;
  }

  public void setDateRanges(DateRange[] dateRanges) {
    this.dateRanges = dateRanges;
  }

  @Override
  public String toElasticSearch() {
    StringBuilder sb = new StringBuilder();

    sb.append("\"date_range\": {\"field\": \"");
    sb.append(field);
    sb.append("\"");

    if (format != null) {
      sb.append(", \"format\": \"");
      sb.append(format);
      sb.append("\"");
    }

    if (dateRanges != null && dateRanges.length > 0) {
      sb.append(", \"ranges\": [");

      for (int i = 0; i < dateRanges.length; i++) {
        if (i > 0) {
          sb.append(",");
        }
        sb.append(dateRanges[i].toElasticSearch());
      }

      sb.append("]");
    }

    if (size != null) {
      sb.append(", \"size\": ");
      sb.append(size);
    }

    if (minThreshold != null) {
      sb.append(", \"min_doc_count\": ").append(minThreshold);
    }

    sb.append("}");

    return sb.toString();
  }

  @Override
  public String toString() {
    StringBuffer sb = new StringBuffer();
    sb.append("date-range: {field: " + field + ", format: " + format + ", size: " + size
        + ", minThreshold: " + minThreshold + "ranges: [");
    for (int i = 0; i < dateRanges.length; i++) {
      if (i > 0) {
        sb.append(",");
      }
      sb.append(dateRanges[i].toString());
    }
    sb.append("]");

    return sb.toString();
  }

}