rocket_async_compression 0.6.1

Response compression in both gzip and brotli formats for the Rocket webserver using the `async-compression` library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use rocket::response::{self, Responder, Response};
use rocket::Request;

use super::CompressionUtils;

/// Compresses responses with Brotli or Gzip compression using the `async-compression` crate.
///
/// The `Compress` type implements brotli and gzip compression for responses in
/// accordance with the `Accept-Encoding` header. If accepted, brotli
/// compression is preferred over gzip.
///
/// Responses that already have a `Content-Encoding` header are not compressed.
///
/// # Usage
///
/// Compress responses by wrapping a `Responder` inside `Compress`:
///
/// ```rust
/// use rocket_async_compression::Compress;
///
/// let response = Compress::default("Hi.");
/// ```
#[derive(Debug)]
pub struct Compress<R>(pub R, pub async_compression::Level);

impl<'r, 'o: 'r, R: Responder<'r, 'o>> Compress<R> {
    pub fn default(r: R) -> Compress<R> {
        Compress(r, async_compression::Level::Default)
    }

    pub fn best(r: R) -> Compress<R> {
        Compress(r, async_compression::Level::Best)
    }

    pub fn fastest(r: R) -> Compress<R> {
        Compress(r, async_compression::Level::Fastest)
    }
}

impl<'r, 'o: 'r, R: Responder<'r, 'o>> Responder<'r, 'o> for Compress<R> {
    #[inline(always)]
    fn respond_to(self, request: &'r Request<'_>) -> response::Result<'o> {
        let mut response = Response::build()
            .merge(self.0.respond_to(request)?)
            .finalize();

        CompressionUtils::compress_response(request, &mut response, &[], self.1);
        Ok(response)
    }
}