search-data-service-app/src/main/java/org/onap/aai/sa/searchdbabstraction/searchapi/AggregationStatement.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 import java.util.Arrays;
  25
  26 public class AggregationStatement {
  27
  28     @JsonProperty("group-by")
  29     private GroupByAggregation groupBy;
  30
  31     @JsonProperty("date-range")
  32     private DateRangeAggregation dateRange;
  33
  34     @JsonProperty("date-histogram")
  35     private DateHistogramAggregation dateHist;
  36
  37     @JsonProperty("nested")
  38     private Aggregation[] nested;
  39
  40     @JsonProperty("sub-aggregations")
  41     private Aggregation[] subAggregations;
  42
  43     public GroupByAggregation getGroupBy() {
  44         return groupBy;
  45     }
  46
  47     public void setGroupBy(GroupByAggregation groupBy) {
  48         this.groupBy = groupBy;
  49     }
  50
  51     public DateRangeAggregation getDateRange() {
  52         return dateRange;
  53     }
  54
  55     public void setDateRange(DateRangeAggregation dateRange) {
  56         this.dateRange = dateRange;
  57     }
  58
  59     public DateHistogramAggregation getDateHist() {
  60         return dateHist;
  61     }
  62
  63     public void setDateHist(DateHistogramAggregation dateHist) {
  64         this.dateHist = dateHist;
  65     }
  66
  67     public Aggregation[] getNested() {
  68         return nested;
  69     }
  70
  71     public void setNested(Aggregation[] nested) {
  72         this.nested = nested;
  73     }
  74
  75     public Aggregation[] getSubAggregations() {
  76         return subAggregations;
  77     }
  78
  79     public void setSubAggregations(Aggregation[] subAggregations) {
  80         this.subAggregations = subAggregations;
  81     }
  82
  83     public String toElasticSearch() {
  84         StringBuilder sb = new StringBuilder();
  85
  86         sb.append("{");
  87
  88         if (nested != null && nested.length > 0) {
  89             sb.append("\"nested\": {\"path\": \"");
  90             if (nested[0].getStatement() != null) {
  91                 sb.append(nested[0].getStatement().getNestedPath());
  92             }
  93             sb.append("\"}, \"aggs\": {");
  94             for (int i = 0; i < nested.length; i++) {
  95                 if (i > 0) {
  96                     sb.append(",");
  97                 }
  98                 sb.append(nested[i].toElasticSearch());
  99             }
 100
 101             sb.append("}");
 102         } else {
 103             if (groupBy != null) {
 104                 sb.append(groupBy.toElasticSearch());
 105             } else if (dateRange != null) {
 106                 sb.append(dateRange.toElasticSearch());
 107             } else if (dateHist != null) {
 108                 sb.append(dateHist.toElasticSearch());
 109             }
 110
 111             if (subAggregations != null && subAggregations.length > 0) {
 112                 sb.append(", \"aggs\": {");
 113                 for (int i = 0; i < subAggregations.length; i++) {
 114                     if (i > 0) {
 115                         sb.append(",");
 116                     }
 117                     sb.append(subAggregations[i].toElasticSearch());
 118                 }
 119                 sb.append("}");
 120             }
 121         }
 122
 123         sb.append("}");
 124
 125         return sb.toString();
 126     }
 127
 128     @Override
 129     public String toString() {
 130         StringBuilder sb = new StringBuilder();
 131
 132         if (nested != null) {
 133             sb.append("{nested: ");
 134             sb.append(Arrays.toString(nested));
 135         } else if (groupBy != null) {
 136             sb.append(groupBy.toString());
 137         } else if (dateHist != null) {
 138             sb.append(dateHist.toString());
 139         } else if (dateRange != null) {
 140             sb.append(dateRange.toString());
 141         }
 142
 143         if (subAggregations != null) {
 144             sb.append(", sub-aggregations: ");
 145             sb.append(Arrays.toString(subAggregations));
 146         }
 147
 148         sb.append("}");
 149
 150         return sb.toString();
 151     }
 152
 153     public String getNestedPath() {
 154         String path = null;
 155         String fieldName = null;
 156
 157         if (groupBy != null) {
 158             fieldName = groupBy.getField();
 159         } else if (dateRange != null) {
 160             fieldName = dateRange.getField();
 161         } else if (dateHist != null) {
 162             fieldName = dateHist.getField();
 163         }
 164
 165         if (fieldName != null && fieldName.contains(".")) {
 166             // we have nested field
 167             path = fieldName.substring(0, fieldName.indexOf("."));
 168         }
 169
 170         return path;
 171     }
 172
 173 }