diff options
Diffstat (limited to 'vendor/tower-http/src/decompression/mod.rs')
| -rw-r--r-- | vendor/tower-http/src/decompression/mod.rs | 233 |
1 files changed, 0 insertions, 233 deletions
diff --git a/vendor/tower-http/src/decompression/mod.rs b/vendor/tower-http/src/decompression/mod.rs deleted file mode 100644 index 50d4d5fa..00000000 --- a/vendor/tower-http/src/decompression/mod.rs +++ /dev/null @@ -1,233 +0,0 @@ -//! Middleware that decompresses request and response bodies. -//! -//! # Examples -//! -//! #### Request -//! -//! ```rust -//! use bytes::Bytes; -//! use flate2::{write::GzEncoder, Compression}; -//! use http::{header, HeaderValue, Request, Response}; -//! use http_body_util::{Full, BodyExt}; -//! use std::{error::Error, io::Write}; -//! use tower::{Service, ServiceBuilder, service_fn, ServiceExt}; -//! use tower_http::{BoxError, decompression::{DecompressionBody, RequestDecompressionLayer}}; -//! -//! # #[tokio::main] -//! # async fn main() -> Result<(), BoxError> { -//! // A request encoded with gzip coming from some HTTP client. -//! let mut encoder = GzEncoder::new(Vec::new(), Compression::default()); -//! encoder.write_all(b"Hello?")?; -//! let request = Request::builder() -//! .header(header::CONTENT_ENCODING, "gzip") -//! .body(Full::from(encoder.finish()?))?; -//! -//! // Our HTTP server -//! let mut server = ServiceBuilder::new() -//! // Automatically decompress request bodies. -//! .layer(RequestDecompressionLayer::new()) -//! .service(service_fn(handler)); -//! -//! // Send the request, with the gzip encoded body, to our server. -//! let _response = server.ready().await?.call(request).await?; -//! -//! // Handler receives request whose body is decoded when read -//! async fn handler( -//! mut req: Request<DecompressionBody<Full<Bytes>>>, -//! ) -> Result<Response<Full<Bytes>>, BoxError>{ -//! let data = req.into_body().collect().await?.to_bytes(); -//! assert_eq!(&data[..], b"Hello?"); -//! Ok(Response::new(Full::from("Hello, World!"))) -//! } -//! # Ok(()) -//! # } -//! ``` -//! -//! #### Response -//! -//! ```rust -//! use bytes::Bytes; -//! use http::{Request, Response}; -//! use http_body_util::{Full, BodyExt}; -//! use std::convert::Infallible; -//! use tower::{Service, ServiceExt, ServiceBuilder, service_fn}; -//! use tower_http::{compression::Compression, decompression::DecompressionLayer, BoxError}; -//! # -//! # #[tokio::main] -//! # async fn main() -> Result<(), tower_http::BoxError> { -//! # async fn handle(req: Request<Full<Bytes>>) -> Result<Response<Full<Bytes>>, Infallible> { -//! # let body = Full::from("Hello, World!"); -//! # Ok(Response::new(body)) -//! # } -//! -//! // Some opaque service that applies compression. -//! let service = Compression::new(service_fn(handle)); -//! -//! // Our HTTP client. -//! let mut client = ServiceBuilder::new() -//! // Automatically decompress response bodies. -//! .layer(DecompressionLayer::new()) -//! .service(service); -//! -//! // Call the service. -//! // -//! // `DecompressionLayer` takes care of setting `Accept-Encoding`. -//! let request = Request::new(Full::<Bytes>::default()); -//! -//! let response = client -//! .ready() -//! .await? -//! .call(request) -//! .await?; -//! -//! // Read the body -//! let body = response.into_body(); -//! let bytes = body.collect().await?.to_bytes().to_vec(); -//! let body = String::from_utf8(bytes).map_err(Into::<BoxError>::into)?; -//! -//! assert_eq!(body, "Hello, World!"); -//! # -//! # Ok(()) -//! # } -//! ``` - -mod request; - -mod body; -mod future; -mod layer; -mod service; - -pub use self::{ - body::DecompressionBody, future::ResponseFuture, layer::DecompressionLayer, - service::Decompression, -}; - -pub use self::request::future::RequestDecompressionFuture; -pub use self::request::layer::RequestDecompressionLayer; -pub use self::request::service::RequestDecompression; - -#[cfg(test)] -mod tests { - use std::convert::Infallible; - use std::io::Write; - - use super::*; - use crate::test_helpers::Body; - use crate::{compression::Compression, test_helpers::WithTrailers}; - use flate2::write::GzEncoder; - use http::Response; - use http::{HeaderMap, HeaderName, Request}; - use http_body_util::BodyExt; - use tower::{service_fn, Service, ServiceExt}; - - #[tokio::test] - async fn works() { - let mut client = Decompression::new(Compression::new(service_fn(handle))); - - let req = Request::builder() - .header("accept-encoding", "gzip") - .body(Body::empty()) - .unwrap(); - let res = client.ready().await.unwrap().call(req).await.unwrap(); - - // read the body, it will be decompressed automatically - let body = res.into_body(); - let collected = body.collect().await.unwrap(); - let trailers = collected.trailers().cloned().unwrap(); - let decompressed_data = String::from_utf8(collected.to_bytes().to_vec()).unwrap(); - - assert_eq!(decompressed_data, "Hello, World!"); - - // maintains trailers - assert_eq!(trailers["foo"], "bar"); - } - - async fn handle(_req: Request<Body>) -> Result<Response<WithTrailers<Body>>, Infallible> { - let mut trailers = HeaderMap::new(); - trailers.insert(HeaderName::from_static("foo"), "bar".parse().unwrap()); - let body = Body::from("Hello, World!").with_trailers(trailers); - Ok(Response::builder().body(body).unwrap()) - } - - #[tokio::test] - async fn decompress_multi_gz() { - let mut client = Decompression::new(service_fn(handle_multi_gz)); - - let req = Request::builder() - .header("accept-encoding", "gzip") - .body(Body::empty()) - .unwrap(); - let res = client.ready().await.unwrap().call(req).await.unwrap(); - - // read the body, it will be decompressed automatically - let body = res.into_body(); - let decompressed_data = - String::from_utf8(body.collect().await.unwrap().to_bytes().to_vec()).unwrap(); - - assert_eq!(decompressed_data, "Hello, World!"); - } - - #[tokio::test] - async fn decompress_multi_zstd() { - let mut client = Decompression::new(service_fn(handle_multi_zstd)); - - let req = Request::builder() - .header("accept-encoding", "zstd") - .body(Body::empty()) - .unwrap(); - let res = client.ready().await.unwrap().call(req).await.unwrap(); - - // read the body, it will be decompressed automatically - let body = res.into_body(); - let decompressed_data = - String::from_utf8(body.collect().await.unwrap().to_bytes().to_vec()).unwrap(); - - assert_eq!(decompressed_data, "Hello, World!"); - } - - async fn handle_multi_gz(_req: Request<Body>) -> Result<Response<Body>, Infallible> { - let mut buf = Vec::new(); - let mut enc1 = GzEncoder::new(&mut buf, Default::default()); - enc1.write_all(b"Hello, ").unwrap(); - enc1.finish().unwrap(); - - let mut enc2 = GzEncoder::new(&mut buf, Default::default()); - enc2.write_all(b"World!").unwrap(); - enc2.finish().unwrap(); - - let mut res = Response::new(Body::from(buf)); - res.headers_mut() - .insert("content-encoding", "gzip".parse().unwrap()); - Ok(res) - } - - async fn handle_multi_zstd(_req: Request<Body>) -> Result<Response<Body>, Infallible> { - let mut buf = Vec::new(); - let mut enc1 = zstd::Encoder::new(&mut buf, Default::default()).unwrap(); - enc1.write_all(b"Hello, ").unwrap(); - enc1.finish().unwrap(); - - let mut enc2 = zstd::Encoder::new(&mut buf, Default::default()).unwrap(); - enc2.write_all(b"World!").unwrap(); - enc2.finish().unwrap(); - - let mut res = Response::new(Body::from(buf)); - res.headers_mut() - .insert("content-encoding", "zstd".parse().unwrap()); - Ok(res) - } - - #[allow(dead_code)] - async fn is_compatible_with_hyper() { - let client = - hyper_util::client::legacy::Client::builder(hyper_util::rt::TokioExecutor::new()) - .build_http(); - let mut client = Decompression::new(client); - - let req = Request::new(Body::empty()); - - let _: Response<DecompressionBody<_>> = - client.ready().await.unwrap().call(req).await.unwrap(); - } -} |
