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><pre>
  29  * {
  30  *    "aggs": {
  31  *        "range": {
  32  *            "date_range": {
  33  *                "field": "date",
  34  *                "format": "MM-yyy",
  35  *                "ranges": [
  36  *                    { "to": "now-10M/M" },
  37  *                    { "from": "now-10M/M" }
  38  *                ]
  39  *            }
  40  *        }
  41  *    }
  42  * }
  43  * </pre>
  44  *
  45  * @author sye
  46  */
  47 public class DateRangeAggregation extends AbstractAggregation {
  48
  49
  50   private String format;
  51
  52   @JsonProperty("ranges")
  53   private DateRange[] dateRanges;
  54
  55
  56   public String getFormat() {
  57     return format;
  58   }
  59
  60   public void setFormat(String format) {
  61     this.format = format;
  62   }
  63
  64   public DateRange[] getDateRanges() {
  65     return dateRanges;
  66   }
  67
  68   public void setDateRanges(DateRange[] dateRanges) {
  69     this.dateRanges = dateRanges;
  70   }
  71
  72   @Override
  73   public String toElasticSearch() {
  74     StringBuilder sb = new StringBuilder();
  75
  76     sb.append("\"date_range\": {\"field\": \"");
  77     sb.append(field);
  78     sb.append("\"");
  79
  80     if (format != null) {
  81       sb.append(", \"format\": \"");
  82       sb.append(format);
  83       sb.append("\"");
  84     }
  85
  86     if (dateRanges != null && dateRanges.length > 0) {
  87       sb.append(", \"ranges\": [");
  88
  89       for (int i = 0; i < dateRanges.length; i++) {
  90         if (i > 0) {
  91           sb.append(",");
  92         }
  93         sb.append(dateRanges[i].toElasticSearch());
  94       }
  95
  96       sb.append("]");
  97     }
  98
  99     if (size != null) {
 100       sb.append(", \"size\": ");
 101       sb.append(size);
 102     }
 103
 104     if (minThreshold != null) {
 105       sb.append(", \"min_doc_count\": ").append(minThreshold);
 106     }
 107
 108     sb.append("}");
 109
 110     return sb.toString();
 111   }
 112
 113   @Override
 114   public String toString() {
 115     StringBuffer sb = new StringBuffer();
 116     sb.append("date-range: {field: " + field + ", format: " + format + ", size: " + size
 117         + ", minThreshold: " + minThreshold + "ranges: [");
 118     for (int i = 0; i < dateRanges.length; i++) {
 119       if (i > 0) {
 120         sb.append(",");
 121       }
 122       sb.append(dateRanges[i].toString());
 123     }
 124     sb.append("]");
 125
 126     return sb.toString();
 127   }
 128
 129 }