tower_http/decompression/
future.rs

1#![allow(unused_imports)]
2
3use super::{body::BodyInner, DecompressionBody};
4use crate::compression_utils::{AcceptEncoding, CompressionLevel, WrapBody};
5use crate::content_encoding::SupportedEncodings;
6use http::{header, Response};
7use http_body::Body;
8use pin_project_lite::pin_project;
9use std::{
10    future::Future,
11    pin::Pin,
12    task::{ready, Context, Poll},
13};
14
15pin_project! {
16    /// Response future of [`Decompression`].
17    ///
18    /// [`Decompression`]: super::Decompression
19    #[derive(Debug)]
20    pub struct ResponseFuture<F> {
21        #[pin]
22        pub(crate) inner: F,
23        pub(crate) accept: AcceptEncoding,
24    }
25}
26
27impl<F, B, E> Future for ResponseFuture<F>
28where
29    F: Future<Output = Result<Response<B>, E>>,
30    B: Body,
31{
32    type Output = Result<Response<DecompressionBody<B>>, E>;
33
34    #[allow(unreachable_code, unused_mut, unused_variables)]
35    fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
36        let res = ready!(self.as_mut().project().inner.poll(cx)?);
37        let (mut parts, body) = res.into_parts();
38
39        let res =
40            if let header::Entry::Occupied(entry) = parts.headers.entry(header::CONTENT_ENCODING) {
41                let body = match entry.get().as_bytes() {
42                    #[cfg(feature = "decompression-gzip")]
43                    b"gzip" if self.accept.gzip() => DecompressionBody::new(BodyInner::gzip(
44                        WrapBody::new(body, CompressionLevel::default()),
45                    )),
46
47                    #[cfg(feature = "decompression-deflate")]
48                    b"deflate" if self.accept.deflate() => DecompressionBody::new(
49                        BodyInner::deflate(WrapBody::new(body, CompressionLevel::default())),
50                    ),
51
52                    #[cfg(feature = "decompression-br")]
53                    b"br" if self.accept.br() => DecompressionBody::new(BodyInner::brotli(
54                        WrapBody::new(body, CompressionLevel::default()),
55                    )),
56
57                    #[cfg(feature = "decompression-zstd")]
58                    b"zstd" if self.accept.zstd() => DecompressionBody::new(BodyInner::zstd(
59                        WrapBody::new(body, CompressionLevel::default()),
60                    )),
61
62                    _ => {
63                        return Poll::Ready(Ok(Response::from_parts(
64                            parts,
65                            DecompressionBody::new(BodyInner::identity(body)),
66                        )))
67                    }
68                };
69
70                entry.remove();
71                parts.headers.remove(header::CONTENT_LENGTH);
72
73                Response::from_parts(parts, body)
74            } else {
75                Response::from_parts(parts, DecompressionBody::new(BodyInner::identity(body)))
76            };
77
78        Poll::Ready(Ok(res))
79    }
80}
tower_http/decompression/future.rs

tower_http/decompression/
future.rs
