Allow specifying slices and sliceId in /document/v1

1 files changed, 4 insertions, 0 deletions

diff --git a/documentapi/src/main/java/com/yahoo/documentapi/VisitorIterator.java b/documentapi/src/main/java/com/yahoo/documentapi/VisitorIterator.java
index 0ccc64bb8f3..17847a56579 100755
--- a/documentapi/src/main/java/com/yahoo/documentapi/VisitorIterator.java
+++ b/documentapi/src/main/java/com/yahoo/documentapi/VisitorIterator.java
@@ -119,6 +119,10 @@ public class VisitorIterator {
                     throw new IllegalArgumentException("Total bucket count in existing progress is not "
                             + "consistent with that of the current document selection");
                 }
+                if (slices > progressToken.getTotalBucketCount()) {
+                    throw new IllegalArgumentException("slices may not exceed number of super-buckets, but got " +
+                                                       slices + " > " + progressToken.getTotalBucketCount());
+                }
             }
 
             if (!progress.isFinished()) {