src/main/java/org/openecomp/sa/searchdbabstraction/searchapi/DateRangeAggregation.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  * Copyright © 2017 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  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  22  */
  23 package org.openecomp.sa.searchdbabstraction.searchapi;
  24
  25 import com.fasterxml.jackson.annotation.JsonProperty;
  26
  27 /**
  28  * An example of a date_range aggregation:
  29  *
  30  * <p><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
 119         + ", minThreshold: " + 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 }