Introduce query parameters for default max hits/groups in grouping queries

4 files changed, 53 insertions, 7 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/grouping/GroupingQueryParser.java b/container-search/src/main/java/com/yahoo/search/grouping/GroupingQueryParser.java
index f4603e5ea28..d3f0278e00f 100644
--- a/container-search/src/main/java/com/yahoo/search/grouping/GroupingQueryParser.java
+++ b/container-search/src/main/java/com/yahoo/search/grouping/GroupingQueryParser.java
@@ -37,6 +37,8 @@ public class GroupingQueryParser extends Searcher {
     public static final CompoundName PARAM_CONTINUE = new CompoundName("continue");
     public static final CompoundName PARAM_REQUEST = new CompoundName(Select.SELECT);
     public static final CompoundName PARAM_TIMEZONE = new CompoundName("timezone");
+    public static final CompoundName PARAM_DEFAULT_MAX_HITS = new CompoundName("grouping.defaultMaxHits");
+    public static final CompoundName PARAM_DEFAULT_MAX_GROUPS = new CompoundName("grouping.defaultMaxGroups");
     private static final ThreadLocal<ZoneCache> zoneCache = new ThreadLocal<>();
 
     @Override
@@ -52,6 +54,8 @@ public class GroupingQueryParser extends Searcher {
                 grpRequest.setRootOperation(op);
                 grpRequest.setTimeZone(zone);
                 grpRequest.continuations().addAll(continuations);
+                grpRequest.setDefaultMaxGroups(query.properties().getInteger(PARAM_DEFAULT_MAX_GROUPS, -1));
+                grpRequest.setDefaultMaxHits(query.properties().getInteger(PARAM_DEFAULT_MAX_HITS, -1));
             }
             return execution.search(query);
         }
diff --git a/container-search/src/main/java/com/yahoo/search/grouping/GroupingRequest.java b/container-search/src/main/java/com/yahoo/search/grouping/GroupingRequest.java
index 6b21b2c6b5c..0c163aaacae 100644
--- a/container-search/src/main/java/com/yahoo/search/grouping/GroupingRequest.java
+++ b/container-search/src/main/java/com/yahoo/search/grouping/GroupingRequest.java
@@ -1,6 +1,7 @@
 // Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.search.grouping;
 
+import com.yahoo.api.annotations.Beta;
 import com.yahoo.net.URI;
 import com.yahoo.search.Query;
 import com.yahoo.search.Result;
@@ -13,6 +14,7 @@ import com.yahoo.search.result.Hit;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.OptionalInt;
 import java.util.TimeZone;
 
 /**
@@ -30,6 +32,8 @@ public class GroupingRequest {
     private final List<Continuation> continuations = new ArrayList<>();
     private GroupingOperation root;
     private TimeZone timeZone;
+    private int defaultMaxHits = -1;
+    private int defaultMaxGroups = -1;
 
     private GroupingRequest(Select parent) {
         this.parent = parent;
@@ -38,16 +42,20 @@ public class GroupingRequest {
     private GroupingRequest(Select parent,
                             List<Continuation> continuations,
                             GroupingOperation root,
-                            TimeZone timeZone) {
+                            TimeZone timeZone,
+                            int defaultMaxHits,
+                            int defaultMaxGroups) {
         this.parent = parent;
         continuations.forEach(item -> this.continuations.add(item.copy()));
         this.root = root != null ? root.copy(null) : null;
         this.timeZone = timeZone;
+        this.defaultMaxHits = defaultMaxHits;
+        this.defaultMaxGroups = defaultMaxGroups;
     }
 
     /** Returns a deep copy of this */
     public GroupingRequest copy(Select parentOfCopy) {
-        return new GroupingRequest(parentOfCopy, continuations, root, timeZone);
+        return new GroupingRequest(parentOfCopy, continuations, root, timeZone, defaultMaxHits, defaultMaxGroups);
     }
 
     /**
@@ -131,6 +139,20 @@ public class GroupingRequest {
         return continuations;
     }
 
+    @Beta
+    public OptionalInt defaultMaxHits() {
+        return defaultMaxHits >= 0 ? OptionalInt.of(defaultMaxHits) : OptionalInt.empty();
+    }
+
+    @Beta public void setDefaultMaxHits(int v) { this.defaultMaxHits = v; }
+
+    @Beta
+    public OptionalInt defaultMaxGroups() {
+        return defaultMaxGroups >= 0 ? OptionalInt.of(defaultMaxGroups) : OptionalInt.empty();
+    }
+
+    @Beta public void setDefaultMaxGroups(int v) { this.defaultMaxGroups = v; }
+
     /**
      * Creates a new grouping request and adds it to the query.getSelect().getGrouping() list
      *
diff --git a/container-search/src/main/java/com/yahoo/search/grouping/vespa/GroupingExecutor.java b/container-search/src/main/java/com/yahoo/search/grouping/vespa/GroupingExecutor.java
index 3e707b0cd38..c09502110b1 100644
--- a/container-search/src/main/java/com/yahoo/search/grouping/vespa/GroupingExecutor.java
+++ b/container-search/src/main/java/com/yahoo/search/grouping/vespa/GroupingExecutor.java
@@ -150,6 +150,8 @@ public class GroupingExecutor extends Searcher {
         builder.setDefaultSummaryName(query.getPresentation().getSummary());
         builder.setTimeZone(req.getTimeZone());
         builder.addContinuations(req.continuations());
+        req.defaultMaxGroups().ifPresent(builder::setDefaultMaxGroups);
+        req.defaultMaxHits().ifPresent(builder::setDefaultMaxHits);
         builder.build();
 
         RequestContext ctx = new RequestContext(req, builder.getTransform());
diff --git a/container-search/src/main/java/com/yahoo/search/grouping/vespa/RequestBuilder.java b/container-search/src/main/java/com/yahoo/search/grouping/vespa/RequestBuilder.java
index e636c0d63a6..5ee6f7bc604 100644
--- a/container-search/src/main/java/com/yahoo/search/grouping/vespa/RequestBuilder.java
+++ b/container-search/src/main/java/com/yahoo/search/grouping/vespa/RequestBuilder.java
@@ -27,6 +27,8 @@ class RequestBuilder {
     private final GroupingTransform transform;
     private GroupingOperation root;
     private int tag = 0;
+    private int defaultMaxHits = -1;
+    private int defaultMaxGroups = -1;
 
     /**
      * Constructs a new instance of this class.
@@ -138,6 +140,10 @@ class RequestBuilder {
         return this;
     }
 
+    public RequestBuilder setDefaultMaxGroups(int v) { this.defaultMaxGroups = v; return this; }
+
+    public RequestBuilder setDefaultMaxHits(int v) { this.defaultMaxHits = v; return this; }
+
     private void processRequestNode(BuildFrame frame) {
         int level = frame.astNode.getLevel();
         if (level > 2) {
@@ -173,10 +179,16 @@ class RequestBuilder {
                 grpLevel.setPrecision(frame.state.precision + offset);
                 frame.state.precision = null;
             }
+            int max = -1;
             if (frame.state.max != null) {
-                transform.putMax(tag, frame.state.max, "group list");
-                grpLevel.setMaxGroups(LOOKAHEAD + frame.state.max + offset);
+                max = frame.state.max;
                 frame.state.max = null;
+            } else if (defaultMaxGroups >= 0) {
+                max = defaultMaxGroups;
+            }
+            if (max >= 0) {
+                transform.putMax(tag, max, "group list");
+                grpLevel.setMaxGroups(LOOKAHEAD + max + offset);
             }
             frame.grouping.getLevels().add(grpLevel);
         }
@@ -285,11 +297,17 @@ class RequestBuilder {
                 throw new UnsupportedOperationException("Can not label expression '" + exp + "'.");
             }
             HitsAggregationResult hits = (HitsAggregationResult)result;
+            int max = -1;
             if (frame.state.max != null) {
-                transform.putMax(tag, frame.state.max, "hit list");
-                int offset = transform.getOffset(tag);
-                hits.setMaxHits(LOOKAHEAD + frame.state.max + offset);
+                max = frame.state.max;
                 frame.state.max = null;
+            } else if (defaultMaxHits >= 0) {
+                max = defaultMaxHits;
+            }
+            if (max >= 0) {
+                transform.putMax(tag, max, "hit list");
+                int offset = transform.getOffset(tag);
+                hits.setMaxHits(LOOKAHEAD + max + offset);
             }
             transform.putLabel(group.getTag(), tag, frame.state.label, "hit list");
         } else {