Merge pull request #880 from hatoo/refactor-timeout

refactor: replace tokio::select with tokio::time::timeout for request…

Author: hatoo

src/client.rs

@@ -776,13 +776,9 @@ impl Client {
         };
 
         if let Some(timeout) = self.timeout {
-            tokio::select! {
-                res = do_req => {
-                    res
-                }
-                _ = tokio::time::sleep(timeout) => {
-                    Err(ClientError::Timeout)
-                }
+            match tokio::time::timeout(timeout, do_req).await {
+                Ok(res) => res,
+                Err(_) => Err(ClientError::Timeout),
             }
         } else {
             do_req.await
@@ -923,13 +919,9 @@ impl Client {
         };
 
         if let Some(timeout) = self.timeout {
-            tokio::select! {
-                res = do_req => {
-                    res
-                }
-                _ = tokio::time::sleep(timeout) => {
-                    Err(ClientError::Timeout)
-                }
+            match tokio::time::timeout(timeout, do_req).await {
+                Ok(res) => res,
+                Err(_) => Err(ClientError::Timeout),
             }
         } else {
             do_req.await

src/client_h3.rs

@@ -175,13 +175,9 @@ impl Client {
         };
 
         if let Some(timeout) = self.timeout {
-            tokio::select! {
-                res = do_req => {
-                    res
-                }
-                _ = tokio::time::sleep(timeout) => {
-                    Err(ClientError::Timeout)
-                }
+            match tokio::time::timeout(timeout, do_req).await {
+                Ok(res) => res,
+                Err(_) => Err(ClientError::Timeout),
             }
         } else {
             do_req.await