play around grow

Signed-off-by: bowenlan-amzn <[email protected]>
bowenlan-amzn · Jan 30, 2025 · 8978019 · 8978019
1 parent 45f486b
commit 8978019
Showing 1 changed file with 21 additions and 8 deletions.
diff --git a/...main/java/org/opensearch/search/aggregations/bucket/filterrewrite/PointTreeTraversal.java b/...main/java/org/opensearch/search/aggregations/bucket/filterrewrite/PointTreeTraversal.java
@@ -122,6 +122,12 @@ private static void intersectWithRanges(
 
     private static PointValues.IntersectVisitor getIntersectVisitor(RangeCollectorForPointTree collector) {
         return new PointValues.IntersectVisitor() {
+
+            @Override
+            public void grow(int count) {
+                collector.grow(count);
+            }
+
             @Override
             public void visit(int docID) {
                 // this branch should be unreachable
@@ -222,21 +228,28 @@ private void count() {
         }
 
         private void collectDocId(int docId) {
-            if (docIdSetBuilders[activeIndex] == null) {
-                // TODO hard code for now, should be controlled by intersector grow
-                docIdSetBuilders[activeIndex] = disBuilderSupplier.get();
-                currentAdder = docIdSetBuilders[activeIndex].grow(1000);
-            }
+            // if (docIdSetBuilders[activeIndex] == null) {
+            //     // TODO hard code for now, should be controlled by intersector grow
+            //     docIdSetBuilders[activeIndex] = disBuilderSupplier.get();
+            //     currentAdder = docIdSetBuilders[activeIndex].grow(1000);
+            // }
             currentAdder.add(docId);
         }
 
         private void collectDocIdSet(DocIdSetIterator iter) throws IOException {
+            // if (docIdSetBuilders[activeIndex] == null) {
+            //     // TODO hard code for now, should be controlled by intersector grow
+            //     docIdSetBuilders[activeIndex] = disBuilderSupplier.get();
+            //     currentAdder = docIdSetBuilders[activeIndex].grow(1000);
+            // }
+            currentAdder.add(iter);
+        }
+
+        private void grow(int count) {
             if (docIdSetBuilders[activeIndex] == null) {
-                // TODO hard code for now, should be controlled by intersector grow
                 docIdSetBuilders[activeIndex] = disBuilderSupplier.get();
-                currentAdder = docIdSetBuilders[activeIndex].grow(1000);
+                currentAdder = docIdSetBuilders[activeIndex].grow(count);
             }
-            currentAdder.add(iter);
         }
 
         private void countNode(int count) {