implement workaround for BatchIterator interface change

apache · Sep 18, 2023 · 94c3b65 · 94c3b65
1 parent b404563
commit 94c3b65
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/processing/src/main/java/org/apache/druid/collections/bitmap/BatchIteratorAdapter.java b/processing/src/main/java/org/apache/druid/collections/bitmap/BatchIteratorAdapter.java
@@ -25,11 +25,11 @@
 
 public final class BatchIteratorAdapter implements BatchIterator
 {
-  private final IntIterator iterator;
+  private final PeekableIteratorAdapter<IntIterator> iterator;
 
   public BatchIteratorAdapter(IntIterator iterator)
   {
-    this.iterator = Preconditions.checkNotNull(iterator, "iterator");
+    this.iterator = new PeekableIteratorAdapter<>(Preconditions.checkNotNull(iterator, "iterator"));
   }
 
   @Override
@@ -49,6 +49,12 @@ public boolean hasNext()
     return iterator.hasNext();
   }
 
+  @Override
+  public void advanceIfNeeded(int target)
+  {
+    iterator.advanceIfNeeded(target);
+  }
+
   @Override
   public BatchIterator clone()
   {