tower_http/decompression/
mod.rs1mod request;
95
96mod body;
97mod future;
98mod layer;
99mod service;
100
101pub use self::{
102 body::DecompressionBody, future::ResponseFuture, layer::DecompressionLayer,
103 service::Decompression,
104};
105
106pub use self::request::future::RequestDecompressionFuture;
107pub use self::request::layer::RequestDecompressionLayer;
108pub use self::request::service::RequestDecompression;
109
110#[cfg(test)]
111mod tests {
112 use std::convert::Infallible;
113 use std::io::Write;
114
115 use super::*;
116 use crate::test_helpers::Body;
117 use crate::{compression::Compression, test_helpers::WithTrailers};
118 use flate2::write::GzEncoder;
119 use http::Response;
120 use http::{HeaderMap, HeaderName, Request};
121 use http_body_util::BodyExt;
122 use tower::{service_fn, Service, ServiceExt};
123
124 #[tokio::test]
125 async fn works() {
126 let mut client = Decompression::new(Compression::new(service_fn(handle)));
127
128 let req = Request::builder()
129 .header("accept-encoding", "gzip")
130 .body(Body::empty())
131 .unwrap();
132 let res = client.ready().await.unwrap().call(req).await.unwrap();
133
134 let body = res.into_body();
136 let collected = body.collect().await.unwrap();
137 let trailers = collected.trailers().cloned().unwrap();
138 let decompressed_data = String::from_utf8(collected.to_bytes().to_vec()).unwrap();
139
140 assert_eq!(decompressed_data, "Hello, World!");
141
142 assert_eq!(trailers["foo"], "bar");
144 }
145
146 async fn handle(_req: Request<Body>) -> Result<Response<WithTrailers<Body>>, Infallible> {
147 let mut trailers = HeaderMap::new();
148 trailers.insert(HeaderName::from_static("foo"), "bar".parse().unwrap());
149 let body = Body::from("Hello, World!").with_trailers(trailers);
150 Ok(Response::builder().body(body).unwrap())
151 }
152
153 #[tokio::test]
154 async fn decompress_multi_gz() {
155 let mut client = Decompression::new(service_fn(handle_multi_gz));
156
157 let req = Request::builder()
158 .header("accept-encoding", "gzip")
159 .body(Body::empty())
160 .unwrap();
161 let res = client.ready().await.unwrap().call(req).await.unwrap();
162
163 let body = res.into_body();
165 let decompressed_data =
166 String::from_utf8(body.collect().await.unwrap().to_bytes().to_vec()).unwrap();
167
168 assert_eq!(decompressed_data, "Hello, World!");
169 }
170
171 #[tokio::test]
172 async fn decompress_multi_zstd() {
173 let mut client = Decompression::new(service_fn(handle_multi_zstd));
174
175 let req = Request::builder()
176 .header("accept-encoding", "zstd")
177 .body(Body::empty())
178 .unwrap();
179 let res = client.ready().await.unwrap().call(req).await.unwrap();
180
181 let body = res.into_body();
183 let decompressed_data =
184 String::from_utf8(body.collect().await.unwrap().to_bytes().to_vec()).unwrap();
185
186 assert_eq!(decompressed_data, "Hello, World!");
187 }
188
189 async fn handle_multi_gz(_req: Request<Body>) -> Result<Response<Body>, Infallible> {
190 let mut buf = Vec::new();
191 let mut enc1 = GzEncoder::new(&mut buf, Default::default());
192 enc1.write_all(b"Hello, ").unwrap();
193 enc1.finish().unwrap();
194
195 let mut enc2 = GzEncoder::new(&mut buf, Default::default());
196 enc2.write_all(b"World!").unwrap();
197 enc2.finish().unwrap();
198
199 let mut res = Response::new(Body::from(buf));
200 res.headers_mut()
201 .insert("content-encoding", "gzip".parse().unwrap());
202 Ok(res)
203 }
204
205 async fn handle_multi_zstd(_req: Request<Body>) -> Result<Response<Body>, Infallible> {
206 let mut buf = Vec::new();
207 let mut enc1 = zstd::Encoder::new(&mut buf, Default::default()).unwrap();
208 enc1.write_all(b"Hello, ").unwrap();
209 enc1.finish().unwrap();
210
211 let mut enc2 = zstd::Encoder::new(&mut buf, Default::default()).unwrap();
212 enc2.write_all(b"World!").unwrap();
213 enc2.finish().unwrap();
214
215 let mut res = Response::new(Body::from(buf));
216 res.headers_mut()
217 .insert("content-encoding", "zstd".parse().unwrap());
218 Ok(res)
219 }
220
221 #[allow(dead_code)]
222 async fn is_compatible_with_hyper() {
223 let client =
224 hyper_util::client::legacy::Client::builder(hyper_util::rt::TokioExecutor::new())
225 .build_http();
226 let mut client = Decompression::new(client);
227
228 let req = Request::new(Body::empty());
229
230 let _: Response<DecompressionBody<_>> =
231 client.ready().await.unwrap().call(req).await.unwrap();
232 }
233}