spacegate_kernel/helper_layers/function/
handler.rs

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
#![allow(non_snake_case)]
use futures_util::Future;
use hyper::{Request, Response};

use crate::SgBody;

use super::Inner;
use crate::extractor::Extract;
// F: Fn(Request<SgBody>, Inner) -> Fut + Send + Sync + Clone + 'static,
// Fut: Future<Output = Response<SgBody>> + Send + 'static,
pub trait HandlerFn<T, Fut>
where
    Fut: Future<Output = Response<SgBody>> + Send + 'static,
{
    fn apply(&self, request: Request<SgBody>, inner: Inner) -> Fut;
}

macro_rules! impl_handler {
    // entry
    ($($t:ident,)*) => {
        impl_handler!(@);
        impl_handler!(; $($t,)*);
    };
    // next
    ($($t:ident,)*;$next:ident, $($rest:ident,)*) => {
        impl_handler!(@$($t,)*$next,);
        impl_handler!($($t,)*$next,;$($rest,)*);
    };
    // exit
    ($($t:ident,)*;) => {};
    // do impl
    (@$($t:ident,)*) => {
        #[allow(unused_variables, unused_parens)]
        impl<F, Fut, $($t),* > HandlerFn<($($t),*), Fut> for F
        where
            F: (Fn(Request<SgBody>, Inner, $($t,)*) -> Fut) + Send + Sync + Clone + 'static ,
            Fut: Future<Output = Response<SgBody>> + Send + 'static,
            $($t: Extract),*
        {
            fn apply(&self, request: Request<SgBody>, inner: Inner) -> Fut {
                $(let $t = $t::extract(&request);)*
                (self.clone())(request, inner, $($t,)*)
            }
        }
    };

}

impl Inner {
    pub fn invoke<T, Fut, H>(self, request: Request<SgBody>, handler: H) -> Fut
    where
        H: HandlerFn<T, Fut>,
        Fut: Future<Output = Response<SgBody>> + Send + 'static,
    {
        handler.apply(request, self)
    }
}

impl_handler!(T0, T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14, T15,);