Disable timeout on waiting commit types

quickwit-oss · Aug 30, 2024 · fbbdc0f · fbbdc0f
1 parent 1e18d5d
commit fbbdc0f
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/quickwit/quickwit-ingest/src/ingest_v2/router.rs b/quickwit/quickwit-ingest/src/ingest_v2/router.rs
@@ -606,9 +606,14 @@ impl IngestRouterService for IngestRouter {
             .try_acquire_many_owned(request_size_bytes as u32)
             .map_err(|_| IngestV2Error::TooManyRequests(RateLimitingCause::RouterLoadShedding))?;
 
-        let ingest_res = self
-            .ingest_timeout(ingest_request, ingest_request_timeout())
-            .await;
+        let ingest_res = if ingest_request.commit_type() == CommitTypeV2::Auto {
+            self.ingest_timeout(ingest_request, ingest_request_timeout())
+                .await
+        } else {
+            Ok(self
+                .retry_batch_persist(ingest_request, MAX_PERSIST_ATTEMPTS)
+                .await)
+        };
 
         update_ingest_metrics(&ingest_res, num_subrequests);