Add tokio sleep backend and use it when reqwest is used

Author: paolobarbolini

Cargo.toml

@@ -34,6 +34,7 @@ futures-util = { version = "0.3.31", default-features = false, features = ["io"]
 
 [target.'cfg(not(target_arch = "wasm32"))'.dependencies]
 jsonwebtoken = { version = "9.3.1", default-features = false }
+tokio = { version = "1.38", optional = true, features = ["time"] }
 
 [target.'cfg(target_arch = "wasm32")'.dependencies]
 uuid = { version = "1.17.0", default-features = false, features = ["v4", "js"] }
@@ -42,7 +43,7 @@ wasm-bindgen-futures = "0.4"
 
 [features]
 default = ["reqwest"]
-reqwest = ["dep:reqwest", "pin-project-lite", "bytes"]
+reqwest = ["dep:reqwest", "dep:tokio", "pin-project-lite", "bytes"]
 futures-unsend = []
 
 [dev-dependencies]

src/client.rs

@@ -1146,7 +1146,7 @@ impl<Http: HttpClient> Client<Http> {
     }
 
     fn sleep_backend(&self) -> SleepBackend {
-        SleepBackend::infer()
+        SleepBackend::infer(self.http_client.is_tokio())
     }
 }
 

src/request.rs

@@ -101,6 +101,10 @@ pub trait HttpClient: Clone + Send + Sync {
         content_type: &str,
         expected_status_code: u16,
     ) -> Result<Output, Error>;
+
+    fn is_tokio(&self) -> bool {
+        false
+    }
 }
 
 pub fn parse_response<Output: DeserializeOwned>(

src/reqwest.rs

@@ -112,6 +112,10 @@ impl HttpClient for ReqwestClient {
 
         parse_response(status, expected_status_code, &body, url.to_string())
     }
+
+    fn is_tokio(&self) -> bool {
+        true
+    }
 }
 
 fn verb<Q, B>(method: &Method<Q, B>) -> reqwest::Method {

src/utils.rs

@@ -2,14 +2,23 @@ use std::time::Duration;
 
 #[derive(Debug, Copy, Clone)]
 pub(crate) enum SleepBackend {
+    #[cfg(all(not(target_arch = "wasm32"), feature = "reqwest"))]
+    Tokio,
     #[cfg(not(target_arch = "wasm32"))]
     Thread,
     #[cfg(target_arch = "wasm32")]
     Javascript,
 }
 
 impl SleepBackend {
-    pub(crate) fn infer() -> Self {
+    pub(crate) fn infer(is_tokio: bool) -> Self {
+        #[cfg(all(not(target_arch = "wasm32"), feature = "reqwest"))]
+        if is_tokio {
+            return Self::Tokio;
+        }
+        #[cfg(any(target_arch = "wasm32", not(feature = "reqwest")))]
+        let _ = is_tokio;
+
         #[cfg(not(target_arch = "wasm32"))]
         return Self::Thread;
 
@@ -19,6 +28,10 @@ impl SleepBackend {
 
     pub(crate) async fn sleep(self, interval: Duration) {
         match self {
+            #[cfg(all(not(target_arch = "wasm32"), feature = "reqwest"))]
+            Self::Tokio => {
+                tokio::time::sleep(interval).await;
+            }
             #[cfg(not(target_arch = "wasm32"))]
             Self::Thread => {
                 let (sender, receiver) = futures_channel::oneshot::channel::<()>();
@@ -54,6 +67,17 @@ mod test {
     use super::*;
     use meilisearch_test_macro::meilisearch_test;
 
+    #[cfg(all(not(target_arch = "wasm32"), feature = "reqwest"))]
+    #[meilisearch_test]
+    async fn sleep_tokio() {
+        let sleep_duration = Duration::from_millis(10);
+        let now = std::time::Instant::now();
+
+        SleepBackend::Tokio.sleep(sleep_duration).await;
+
+        assert!(now.elapsed() >= sleep_duration);
+    }
+
     #[cfg(not(target_arch = "wasm32"))]
     #[meilisearch_test]
     async fn sleep_thread() {