From 6a6ee225737781f1aeef83735d57b7393ad07dcc Mon Sep 17 00:00:00 2001
From: Paul Masurel <paul@quickwit.io>
Date: Thu, 26 Sep 2024 11:02:11 +0900
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: trinity-1686a <trinity@quickwit.io>
---
 .../src/elasticsearch_api/rest_handler.rs        | 16 ++++++++++++----
 .../src/search_api/rest_handler.rs               |  3 +--
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/quickwit/quickwit-serve/src/elasticsearch_api/rest_handler.rs b/quickwit/quickwit-serve/src/elasticsearch_api/rest_handler.rs
index f8587705875..126d99df889 100644
--- a/quickwit/quickwit-serve/src/elasticsearch_api/rest_handler.rs
+++ b/quickwit/quickwit-serve/src/elasticsearch_api/rest_handler.rs
@@ -857,8 +857,14 @@ async fn es_scroll(
     };
     let search_response: SearchResponse = search_service.scroll(scroll_request).await?;
     // TODO append_shard_doc depends on the initial request, but we don't have access to it
+
+    // Ideally, we would have wanted to reuse the setting from the initial search request.
+    // However, passing that parameter is cumbersome and not necessary:
+    // if we have a valid `scroll_id` it means that the search request was successful.
+    // We can therefore allow failed splits, it won't make any difference.
+    let allow_failed_splits = true;
     let mut search_response_rest: ElasticsearchResponse =
-        convert_to_es_search_response(search_response, false, None, None, true)?;
+        convert_to_es_search_response(search_response, false, None, None, allow_failed_splits)?;
     search_response_rest.took = start_instant.elapsed().as_millis() as u32;
     Ok(search_response_rest)
 }
@@ -941,6 +947,8 @@ fn convert_to_es_search_response(
         None
     };
     let num_failed_splits = resp.failed_splits.len() as u32;
+    let num_successful_splits = resp.num_successful_splits as u32;
+    let num_total_splits = num_successful_splits + num_failed_splits;
     Ok(ElasticsearchResponse {
         timed_out: false,
         hits: HitsMetadata {
@@ -955,8 +963,8 @@ fn convert_to_es_search_response(
         scroll_id: resp.scroll_id,
         // There is not concept of shards here, but use this to convey split search failures.
         shards: ShardStatistics {
-            total: 0u32,
-            successful: 0u32,
+            total: num_total_splits,
+            successful: num_successful_splits,
             skipped: 0u32,
             failed: num_failed_splits,
             failures: Vec::new(),
@@ -1188,7 +1196,7 @@ mod tests {
                 failed_splits: vec![split_error.clone()],
                 ..Default::default()
             };
-            // Event if we allow partial search results, with a single successful fail, we have a
+            // Event if we allow partial search results, with a fail and no success, we have a
             // failure.
             convert_to_es_search_response(search_response, false, None, None, true).unwrap_err();
         }
diff --git a/quickwit/quickwit-serve/src/search_api/rest_handler.rs b/quickwit/quickwit-serve/src/search_api/rest_handler.rs
index 7cb60ecb222..42a9a0ff44b 100644
--- a/quickwit/quickwit-serve/src/search_api/rest_handler.rs
+++ b/quickwit/quickwit-serve/src/search_api/rest_handler.rs
@@ -313,8 +313,7 @@ async fn search_endpoint(
             .root_search(search_request)
             .await
             .and_then(|search_response| {
-                // We consider case where
-                if allow_failed_splits {
+                if !allow_failed_splits || search_response.num_successful_splits == 0 {
                     if let Some(search_error) =
                         SearchError::from_split_errors(&search_response.failed_splits[..])
                     {