feat(http): add multipart for server

cloudwego · Oct 24, 2024 · 865976b · 865976b
1 parent 11c3367
commit 865976b
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 17 deletions.
diff --git a/volo-http/Cargo.toml b/volo-http/Cargo.toml
@@ -60,7 +60,6 @@ tokio-util = { workspace = true, features = ["io"] }
 tracing.workspace = true
 
 # =====optional=====
-
 multer = { workspace = true, optional = true }
 
 # server optional
@@ -86,9 +85,9 @@ sonic-rs = { workspace = true, optional = true }
 async-stream.workspace = true
 libc.workspace = true
 serde = { workspace = true, features = ["derive"] }
-tokio-test.workspace = true
 rand.workspace = true
 reqwest = { workspace = true, features = ["multipart"] }
+tokio-test.workspace = true
 url.workspace = true
 
 [features]

diff --git a/volo-http/src/server/layer/body_limit.rs b/volo-http/src/server/layer/body_limit.rs
@@ -60,7 +60,7 @@ pub struct BodyLimitService<S> {
     limit: usize,
 }
 
-impl<S,B> Service<ServerContext, ServerRequest<B>> for BodyLimitService<S>
+impl<S, B> Service<ServerContext, ServerRequest<B>> for BodyLimitService<S>
 where
     S: Service<ServerContext, ServerRequest<B>> + Send + Sync + 'static,
     S::Response: IntoResponse,
@@ -100,13 +100,17 @@ where
 #[cfg(test)]
 mod tests {
     use http::{Method, StatusCode};
-    use motore::layer::Layer;
-    use motore::Service;
+    use motore::{layer::Layer, Service};
     use rand::Rng;
-    use crate::server::layer::BodyLimitLayer;
-    use crate::server::route::{any, Route};
-    use crate::server::test_helpers::empty_cx;
-    use crate::utils::test_helpers::simple_req;
+
+    use crate::{
+        server::{
+            layer::BodyLimitLayer,
+            route::{any, Route},
+            test_helpers::empty_cx,
+        },
+        utils::test_helpers::simple_req,
+    };
 
     #[tokio::test]
     async fn test_body_limit() {
@@ -125,7 +129,9 @@ mod tests {
         let min_part_size = 4096;
         let mut body: Vec<u8> = vec![0; min_part_size];
         rng.fill(&mut body[..]);
-        let req = simple_req(Method::GET, "/", unsafe { String::from_utf8_unchecked(body) });
+        let req = simple_req(Method::GET, "/", unsafe {
+            String::from_utf8_unchecked(body)
+        });
         let res = service.call(&mut cx, req).await.unwrap();
         assert_eq!(res.status(), StatusCode::PAYLOAD_TOO_LARGE);
 
@@ -134,4 +140,4 @@ mod tests {
         let res = service.call(&mut cx, req).await.unwrap();
         assert_eq!(res.status(), StatusCode::OK);
     }
-}
+}
diff --git a/volo-http/src/server/utils/multipart.rs b/volo-http/src/server/utils/multipart.rs
@@ -16,9 +16,7 @@
 //!     Router,
 //! };
 //!
-//! async fn upload(
-//!     mut multipart: Multipart,
-//! ) -> Result<StatusCode, MultipartRejectionError> {
+//! async fn upload(mut multipart: Multipart) -> Result<StatusCode, MultipartRejectionError> {
 //!     while let Some(field) = multipart.next_field().await? {
 //!         let name = field.name().unwrap();
 //!         let value = field.bytes().await?;
@@ -64,9 +62,7 @@ use crate::{
 ///     server::utils::multipart::{Multipart, MultipartRejectionError},
 /// };
 ///
-/// async fn upload(
-///     mut multipart: Multipart,
-/// ) -> Result<StatusCode, MultipartRejectionError> {
+/// async fn upload(mut multipart: Multipart) -> Result<StatusCode, MultipartRejectionError> {
 ///     while let Some(field) = multipart.next_field().await? {
 ///         todo!()
 ///     }