bulk visit

Author: gf2121

lucene/core/src/java/org/apache/lucene/index/PointValues.java

@@ -287,6 +287,12 @@ public interface IntersectVisitor {
      */
     void visit(int docID) throws IOException;
 
+    default void visit(int[] docID, int count) throws IOException {
+      for (int i = 0; i < count; i++) {
+        visit(docID[i]);
+      }
+    }
+
     /**
      * Similar to {@link IntersectVisitor#visit(int)}, but a bulk visit and implements may have
      * their optimizations.

lucene/core/src/java/org/apache/lucene/search/PointRangeQuery.java

@@ -180,6 +180,11 @@ public void visit(int docID) {
             adder.add(docID);
           }
 
+          @Override
+          public void visit(int[] docID, int count) throws IOException {
+            adder.add(docID, count);
+          }
+
           @Override
           public void visit(DocIdSetIterator iterator) throws IOException {
             adder.add(iterator);
@@ -219,6 +224,14 @@ public void visit(int docID) {
             cost[0]--;
           }
 
+          @Override
+          public void visit(int[] docID, int count) throws IOException {
+            for (int i = 0; i < count; i++) {
+              result.clear(docID[i]);
+            }
+            cost[0] -= count;
+          }
+
           @Override
           public void visit(DocIdSetIterator iterator) throws IOException {
             result.andNot(iterator);

lucene/core/src/java/org/apache/lucene/util/DocIdSetBuilder.java

@@ -44,6 +44,12 @@ public final class DocIdSetBuilder {
   public abstract static class BulkAdder {
     public abstract void add(int doc);
 
+    public void add(int[] docs, int count) {
+      for (int i=0; i<count; i++) {
+        add(docs[i]);
+      }
+    }
+
     public void add(DocIdSetIterator iterator) throws IOException {
       int docID;
       while ((docID = iterator.nextDoc()) != DocIdSetIterator.NO_MORE_DOCS) {
@@ -92,6 +98,12 @@ private static class BufferAdder extends BulkAdder {
       this.buffer = buffer;
     }
 
+    @Override
+    public void add(int[] docs, int count) {
+      System.arraycopy(docs, 0, buffer.array, buffer.length, count);
+      buffer.length += count;
+    }
+
     @Override
     public void add(int doc) {
       buffer.array[buffer.length++] = doc;

lucene/core/src/java/org/apache/lucene/util/bkd/DocIdsWriter.java

@@ -308,22 +308,16 @@ private static void readContinuousIds(IndexInput in, int count, IntersectVisitor
   private void readDelta16(IndexInput in, int count, IntersectVisitor visitor) throws IOException {
     final int min = in.readVInt();
     forUtil.decode16(in, scratch, count, min);
-    for (int i = 0; i < count; i++) {
-      visitor.visit(scratch[i]);
-    }
+    visitor.visit(scratch, count);
   }
 
   private void readInts24(IndexInput in, int count, IntersectVisitor visitor) throws IOException {
     forUtil.decode24(in, scratch, count);
-    for (int i = 0; i < count; i++) {
-      visitor.visit(scratch[i]);
-    }
+    visitor.visit(scratch, count);
   }
 
   private void readInts32(IndexInput in, int count, IntersectVisitor visitor) throws IOException {
     in.readInts(scratch, 0, count);
-    for (int i = 0; i < count; i++) {
-      visitor.visit(scratch[i]);
-    }
+    visitor.visit(scratch, count);
   }
 }