tower-http-cache-plus 0.0.2

HTTP response caching middleware with integrated compression
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use {
    ::axum::{
        http::header::*,
        response::{IntoResponse, Response},
    },
    duration_str::*,
    kutil::http::*,
    std::{result::Result, time::*},
};

//
// Headers
//

/// Headers.
pub trait Headers<IntoResponseT>
where
    Self: Sized,
    IntoResponseT: IntoResponse,
{
    /// Set `XX-Encode` header to "false".
    fn do_not_encode(self) -> Response {
        self.set_header_bool(XX_ENCODE, false)
    }

    /// Set `XX-Cache` header to "false".
    fn do_not_cache(self) -> Response {
        self.set_header_bool(XX_CACHE, false)
    }

    /// Set `XX-Cache-Duration` header.
    fn with_duration(self, duration: Duration) -> Response;

    /// Set `XX-Cache-Duration` header.
    fn with_duration_str(self, duration: &str) -> Result<Response, InvalidHeaderValue>;

    /// Set a header to a boolean value.
    fn set_header_bool(self, name: HeaderName, value: bool) -> Response;
}

impl<IntoResponseT> Headers<IntoResponseT> for IntoResponseT
where
    IntoResponseT: IntoResponse,
{
    fn with_duration(self, duration: Duration) -> Response {
        let mut response = self.into_response();
        let headers = response.headers_mut();
        headers.remove(XX_CACHE_DURATION);
        let duration = HeaderValue::from_str(duration.human_format().as_str())
            .expect("duration in HTTP header");
        headers.set_into_header_value(XX_CACHE_DURATION, duration);
        response
    }

    fn with_duration_str(self, duration: &str) -> Result<Response, InvalidHeaderValue> {
        let mut response = self.into_response();
        let headers = response.headers_mut();
        headers.set_string_value(XX_CACHE_DURATION, duration)?;
        Ok(response)
    }

    fn set_header_bool(self, name: HeaderName, value: bool) -> Response {
        let mut response = self.into_response();
        response.headers_mut().set_bool_value(name, value);
        response
    }
}