src/main/java/org/onap/aai/sa/searchdbabstraction/searchapi/DateRangeAggregation.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * Copyright © 2017-2018 Amdocs
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *       http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21 package org.onap.aai.sa.searchdbabstraction.searchapi;
  22
  23 import com.fasterxml.jackson.annotation.JsonProperty;
  24
  25 /**
  26  * An example of a date_range aggregation:
  27  *
  28  * <p>
  29  *
  30  * <pre>
  31  * {
  32  *    "aggs": {
  33  *        "range": {
  34  *            "date_range": {
  35  *                "field": "date",
  36  *                "format": "MM-yyy",
  37  *                "ranges": [
  38  *                    { "to": "now-10M/M" },
  39  *                    { "from": "now-10M/M" }
  40  *                ]
  41  *            }
  42  *        }
  43  *    }
  44  * }
  45  * </pre>
  46  *
  47  * @author sye
  48  */
  49 public class DateRangeAggregation extends AbstractAggregation {
  50
  51
  52     private String format;
  53
  54     @JsonProperty("ranges")
  55     private DateRange[] dateRanges;
  56
  57
  58     public String getFormat() {
  59         return format;
  60     }
  61
  62     public void setFormat(String format) {
  63         this.format = format;
  64     }
  65
  66     public DateRange[] getDateRanges() {
  67         return dateRanges;
  68     }
  69
  70     public void setDateRanges(DateRange[] dateRanges) {
  71         this.dateRanges = dateRanges;
  72     }
  73
  74     @Override
  75     public String toElasticSearch() {
  76         StringBuilder sb = new StringBuilder();
  77
  78         sb.append("\"date_range\": {\"field\": \"");
  79         sb.append(field);
  80         sb.append("\"");
  81
  82         if (format != null) {
  83             sb.append(", \"format\": \"");
  84             sb.append(format);
  85             sb.append("\"");
  86         }
  87
  88         if (dateRanges != null && dateRanges.length > 0) {
  89             sb.append(", \"ranges\": [");
  90
  91             for (int i = 0; i < dateRanges.length; i++) {
  92                 if (i > 0) {
  93                     sb.append(",");
  94                 }
  95                 sb.append(dateRanges[i].toElasticSearch());
  96             }
  97
  98             sb.append("]");
  99         }
 100
 101         if (size != null) {
 102             sb.append(", \"size\": ");
 103             sb.append(size);
 104         }
 105
 106         if (minThreshold != null) {
 107             sb.append(", \"min_doc_count\": ").append(minThreshold);
 108         }
 109
 110         sb.append("}");
 111
 112         return sb.toString();
 113     }
 114
 115     @Override
 116     public String toString() {
 117         StringBuffer sb = new StringBuffer();
 118         sb.append("date-range: {field: " + field + ", format: " + format + ", size: " + size + ", minThreshold: "
 119                 + minThreshold + "ranges: [");
 120         for (int i = 0; i < dateRanges.length; i++) {
 121             if (i > 0) {
 122                 sb.append(",");
 123             }
 124             sb.append(dateRanges[i].toString());
 125         }
 126         sb.append("]");
 127
 128         return sb.toString();
 129     }
 130
 131 }