fix(analytics): set aggregation size to retrieve all terms (e.g HTTP …

…Response Status) fixes gravitee-io/issues#2813
gravitee-io · Nov 1, 2019 · fa959b3 · fa959b3
1 parent 2797f47
commit fa959b3
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/src/main/java/io/gravitee/repository/analytics/query/Aggregation.java b/src/main/java/io/gravitee/repository/analytics/query/Aggregation.java
@@ -24,4 +24,6 @@ public interface Aggregation {
     AggregationType type();
 
     String field();
+
+    Integer size();
 }
diff --git a/src/main/java/io/gravitee/repository/analytics/query/DateHistogramQueryBuilder.java b/src/main/java/io/gravitee/repository/analytics/query/DateHistogramQueryBuilder.java
@@ -35,6 +35,10 @@ public DateHistogramQueryBuilder aggregation(Aggregation aggregation) {
     }
 
     public DateHistogramQueryBuilder aggregation(AggregationType type, String field) {
+        return aggregation(type, field, null);
+    }
+
+    public DateHistogramQueryBuilder aggregation(AggregationType type, String field, Integer size) {
         query.aggregations().add(new Aggregation() {
             @Override
             public AggregationType type() {
@@ -45,6 +49,11 @@ public AggregationType type() {
             public String field() {
                 return field;
             }
+
+            @Override
+            public Integer size() {
+                return size;
+            }
         });
 
         return this;

diff --git a/src/main/java/io/gravitee/repository/healthcheck/query/DateHistogramQueryBuilder.java b/src/main/java/io/gravitee/repository/healthcheck/query/DateHistogramQueryBuilder.java
@@ -38,6 +38,10 @@ public DateHistogramQueryBuilder aggregation(Aggregation aggregation) {
     }
 
     public DateHistogramQueryBuilder aggregation(AggregationType type, String field) {
+        return aggregation(type, field, null);
+    }
+
+    public DateHistogramQueryBuilder aggregation(AggregationType type, String field, Integer size) {
         query.aggregations().add(new Aggregation() {
             @Override
             public AggregationType type() {
@@ -48,6 +52,11 @@ public AggregationType type() {
             public String field() {
                 return field;
             }
+
+            @Override
+            public Integer size() {
+                return size;
+            }
         });
 
         return this;