Add TryFrom<Request> for HttpRequest<Option<Body>>

kvinwang · Aug 26, 2021 · 3879694 · 3879694
1 parent 66c1b48
commit 3879694
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 0 deletions.
diff --git a/src/async_impl/request.rs b/src/async_impl/request.rs
@@ -570,6 +570,31 @@ where
     }
 }
 
+impl TryFrom<Request> for HttpRequest<Body> {
+    type Error = crate::Error;
+
+    fn try_from(req: Request) -> crate::Result<Self> {
+        let Request {
+            method,
+            url,
+            headers,
+            body,
+            version,
+            ..
+        } = req;
+
+        let mut req = HttpRequest::builder()
+            .version(version)
+            .method(method)
+            .uri(url.as_str())
+            .body(body.unwrap_or_else(Body::empty))
+            .map_err(crate::error::builder)?;
+
+        *req.headers_mut() = headers;
+        Ok(req)
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::{Client, HttpRequest, Request, Version};

diff --git a/src/wasm/request.rs b/src/wasm/request.rs
@@ -1,6 +1,7 @@
 use std::convert::TryFrom;
 use std::fmt;
 
+use bytes::Bytes;
 use http::{request::Parts, Method, Request as HttpRequest};
 use serde::Serialize;
 #[cfg(feature = "json")]
@@ -435,3 +436,26 @@ where
         })
     }
 }
+
+impl TryFrom<Request> for HttpRequest<Body> {
+    type Error = crate::Error;
+
+    fn try_from(req: Request) -> crate::Result<Self> {
+        let Request {
+            method,
+            url,
+            headers,
+            body,
+            ..
+        } = req;
+
+        let mut req = HttpRequest::builder()
+            .method(method)
+            .uri(url.as_str())
+            .body(body.unwrap_or_else(|| Body::from(Bytes::default())))
+            .map_err(crate::error::builder)?;
+
+        *req.headers_mut() = headers;
+        Ok(req)
+    }
+}