zon_middleware 0.0.6

part of a new WIP, very incomplete async http service stack
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

use std::future::Future;

use zon_core::{HttpMiddleware, HttpService};

/// Middleware to modify a response.
///
/// Currently, this middleware only accepts an `Fn(http::Response<B>) ->
/// http::Response<B>`. `FromRequest[Parts]` and `IntoResponse` return types are
/// going to be supported in the future.
#[derive(Clone, Copy, Debug)]
pub struct MapResponse<F> {
    map_fn: F,
}

impl<F> MapResponse<F> {
    pub fn new(map_fn: F) -> Self {
        Self { map_fn }
    }
}

impl<S, F> HttpMiddleware<S> for MapResponse<F> {
    type Service = MapResponseService<S, F>;

    fn apply(self, inner: S) -> Self::Service {
        MapResponseService::new(inner, self.map_fn)
    }
}

pub struct MapResponseService<S, F> {
    inner: S,
    map_fn: F,
}

impl<S, F> MapResponseService<S, F> {
    pub fn new(inner: S, map_fn: F) -> Self {
        Self { inner, map_fn }
    }
}

impl<S, F, Fut, B, ResB> HttpService<B> for MapResponseService<S, F>
where
    S: HttpService<B>,
    F: Fn(http::Response<S::ResponseBody>) -> Fut + Sync,
    Fut: Future<Output = http::Response<ResB>> + Send,
    B: Send,
{
    type ResponseBody = ResB;

    async fn call(&self, request: http::Request<B>) -> http::Response<Self::ResponseBody> {
        let response = self.inner.call(request).await;
        (self.map_fn)(response).await
    }
}