diff --git a/src/main/java/io/gravitee/repository/analytics/query/Aggregation.java b/src/main/java/io/gravitee/repository/analytics/query/Aggregation.java index 221b8b8c..93d9be08 100644 --- 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 index 7e447a36..4c7ece60 100644 --- 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 index 6584cb8e..74bc2c98 100644 --- 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;