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

Signed-off-by: panguixin <[email protected]>
Signed-off-by: Marc Handalian <[email protected]>

src/main/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);