fix number of total shards in partial search response (#543) (#565)

(cherry picked from commit 0402b6d) Signed-off-by: panguixin <[email protected]> Signed-off-by: Marc Handalian <[email protected]> Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
opensearch-project · Apr 30, 2024 · 29c75e5 · 29c75e5
1 parent 857e3ee
commit 29c75e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../java/org/opensearch/search/asynchronous/listener/AsynchronousSearchProgressListener.java b/.../java/org/opensearch/search/asynchronous/listener/AsynchronousSearchProgressListener.java
@@ -68,7 +68,7 @@ public SearchResponse partialResponse() {
     protected void onListShards(List<SearchShard> shards, List<SearchShard> skippedShards, SearchResponse.Clusters clusters,
                                 boolean fetchPhase) {
         partialResultsHolder.hasFetchPhase.set(fetchPhase);
-        partialResultsHolder.totalShards.set(shards.size());
+        partialResultsHolder.totalShards.set(shards.size() + skippedShards.size());
         partialResultsHolder.skippedShards.set(skippedShards.size());
         partialResultsHolder.successfulShards.set(skippedShards.size());
         partialResultsHolder.clusters.set(clusters);