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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
use std::{future::Future, marker::PhantomData, pin::Pin, task::Context, task::Poll};
use super::ServiceFactory;
pub struct MapInitErr<A, R, C, F, E> {
a: A,
f: F,
e: PhantomData<fn(R, C) -> E>,
}
impl<A, R, C, F, E> MapInitErr<A, R, C, F, E>
where
A: ServiceFactory<R, C>,
F: Fn(A::InitError) -> E,
{
pub(crate) fn new(a: A, f: F) -> Self {
Self {
a,
f,
e: PhantomData,
}
}
}
impl<A, R, C, F, E> Clone for MapInitErr<A, R, C, F, E>
where
A: Clone,
F: Clone,
{
fn clone(&self) -> Self {
Self {
a: self.a.clone(),
f: self.f.clone(),
e: PhantomData,
}
}
}
impl<A, R, C, F, E> ServiceFactory<R, C> for MapInitErr<A, R, C, F, E>
where
A: ServiceFactory<R, C>,
F: Fn(A::InitError) -> E + Clone,
{
type Response = A::Response;
type Error = A::Error;
type Service = A::Service;
type InitError = E;
type Future = MapInitErrFuture<A, R, C, F, E>;
fn new_service(&self, cfg: C) -> Self::Future {
MapInitErrFuture {
fut: self.a.new_service(cfg),
f: self.f.clone(),
}
}
}
pin_project_lite::pin_project! {
pub struct MapInitErrFuture<A, R, C, F, E>
where
A: ServiceFactory<R, C>,
F: Fn(A::InitError) -> E,
{
f: F,
#[pin]
fut: A::Future,
}
}
impl<A, R, C, F, E> Future for MapInitErrFuture<A, R, C, F, E>
where
A: ServiceFactory<R, C>,
F: Fn(A::InitError) -> E,
{
type Output = Result<A::Service, E>;
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
let this = self.project();
this.fut.poll(cx).map_err(this.f)
}
}
#[cfg(test)]
mod tests {
use crate::{fn_factory_with_config, fn_service, pipeline_factory, ServiceFactory};
#[ntex::test]
async fn map_init_err() {
let factory = pipeline_factory(fn_factory_with_config(|err: bool| async move {
if err {
Err(())
} else {
Ok(fn_service(|i: usize| async move { Ok::<_, ()>(i * 2) }))
}
}))
.map_init_err(|_| std::io::Error::new(std::io::ErrorKind::Other, "err"))
.clone();
assert!(factory.new_service(true).await.is_err());
assert!(factory.new_service(false).await.is_ok());
}
}