Implement ConstantScoreScorer#nextDocsAndScores (#14772)

Co-authored-by: gesong.samuel <[email protected]>

Author: 2 people

lucene/CHANGES.txt

@@ -99,6 +99,8 @@ Optimizations
 
 * GITHUB#14834: Optimize CombinedFieldQuery by nextDocsAndScores. (Ge Song)
 
+* GITHUB#14772: Implement ConstantScoreScorer#nextDocsAndScores. (Ge Song)
+
 Bug Fixes
 ---------------------
 * GITHUB#14654: ValueSource.fromDoubleValuesSource(dvs).getSortField() would throw errors when

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

@@ -17,6 +17,8 @@
 package org.apache.lucene.search;
 
 import java.io.IOException;
+import java.util.Arrays;
+import org.apache.lucene.util.Bits;
 
 /**
  * A constant-scoring {@link Scorer}.
@@ -144,4 +146,21 @@ public int docID() {
   public float score() throws IOException {
     return score;
   }
+
+  @Override
+  public void nextDocsAndScores(int upTo, Bits liveDocs, DocAndFloatFeatureBuffer buffer)
+      throws IOException {
+    int batchSize = 64;
+    buffer.growNoCopy(batchSize);
+    int size = 0;
+    DocIdSetIterator iterator = iterator();
+    for (int doc = iterator.docID(); doc < upTo && size < batchSize; doc = iterator.nextDoc()) {
+      if (liveDocs == null || liveDocs.get(doc)) {
+        buffer.docs[size] = doc;
+        ++size;
+      }
+    }
+    Arrays.fill(buffer.features, 0, size, score);
+    buffer.size = size;
+  }
 }