Add estimate for field merge result

ivancea · ivancea · commit 449203fc9bbf · 2025-09-01T18:42:38.000+02:00
diff --git a/server/src/main/java/org/elasticsearch/action/search/QueryPhaseResultConsumer.java b/server/src/main/java/org/elasticsearch/action/search/QueryPhaseResultConsumer.java
@@ -226,6 +226,7 @@ public SearchPhaseController.ReducedQueryPhase reduce() throws Exception {
         // execution for shards on the coordinating node itself
         if (mergeResult != null) {
             consumePartialMergeResult(mergeResult, topDocsList, aggsList);
+            addEstimateAndMaybeBreak(mergeResult.estimatedSize);
         }
         Tuple<TopDocsStats, MergeResult> batchedResult;
         while ((batchedResult = batchedResults.poll()) != null) {

Original file line number	Diff line number	Diff line change
`@@ -226,6 +226,7 @@ public SearchPhaseController.ReducedQueryPhase reduce() throws Exception {`
`226`	`226`	`// execution for shards on the coordinating node itself`
`227`	`227`	`if (mergeResult != null) {`
`228`	`228`	`consumePartialMergeResult(mergeResult, topDocsList, aggsList);`
	`229`	`+ addEstimateAndMaybeBreak(mergeResult.estimatedSize);`
`229`	`230`	`}`
`230`	`231`	`Tuple<TopDocsStats, MergeResult> batchedResult;`
`231`	`232`	`while ((batchedResult = batchedResults.poll()) != null) {`