Adding override for RangeAggregator as well

Signed-off-by: Ankit Jain <jainankitk@apache.org>

Author: jainankitk

server/src/main/java/org/opensearch/search/aggregations/LeafBucketCollector.java

@@ -145,6 +145,8 @@ public void collect(DocIdStream stream) throws IOException {
      */
     @ExperimentalApi
     public void collect(DocIdStream stream, long owningBucketOrd) throws IOException {
+        // Different aggregator implementations should override this method even if to just delegate to super for
+        // helping the performance: when the super call inlines, calls to #collect(int, long) become monomorphic.
         stream.forEach((doc) -> collect(doc, owningBucketOrd));
     }
 

server/src/main/java/org/opensearch/search/aggregations/bucket/range/RangeAggregator.java

@@ -33,6 +33,7 @@
 
 import org.apache.lucene.index.LeafReaderContext;
 import org.apache.lucene.search.DocIdSetIterator;
+import org.apache.lucene.search.DocIdStream;
 import org.apache.lucene.search.ScoreMode;
 import org.apache.lucene.util.FixedBitSet;
 import org.opensearch.core.ParseField;
@@ -353,6 +354,11 @@ public void collect(int doc, long bucket) throws IOException {
                 }
             }
 
+            @Override
+            public void collect(DocIdStream stream, long owningBucketOrd) throws IOException {
+                super.collect(stream, owningBucketOrd);
+            }
+
             private int collect(int doc, double value, long owningBucketOrdinal, int lowBound) throws IOException {
                 MatchedRange range = new MatchedRange(ranges, lowBound, value, maxTo);
                 for (int i = range.startLo; i <= range.endHi; ++i) {