warp/filter/
map.rs

1use std::future::Future;
2use std::pin::Pin;
3use std::task::{Context, Poll};
4
5use futures_util::{ready, TryFuture};
6use pin_project::pin_project;
7
8use super::{Filter, FilterBase, Func, Internal};
9
10#[derive(Clone, Copy, Debug)]
11pub struct Map<T, F> {
12    pub(super) filter: T,
13    pub(super) callback: F,
14}
15
16impl<T, F> FilterBase for Map<T, F>
17where
18    T: Filter,
19    F: Func<T::Extract> + Clone + Send,
20{
21    type Extract = (F::Output,);
22    type Error = T::Error;
23    type Future = MapFuture<T, F>;
24    #[inline]
25    fn filter(&self, _: Internal) -> Self::Future {
26        MapFuture {
27            extract: self.filter.filter(Internal),
28            callback: self.callback.clone(),
29        }
30    }
31}
32
33#[allow(missing_debug_implementations)]
34#[pin_project]
35pub struct MapFuture<T: Filter, F> {
36    #[pin]
37    extract: T::Future,
38    callback: F,
39}
40
41impl<T, F> Future for MapFuture<T, F>
42where
43    T: Filter,
44    F: Func<T::Extract>,
45{
46    type Output = Result<(F::Output,), T::Error>;
47
48    #[inline]
49    fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
50        let pin = self.project();
51        match ready!(pin.extract.try_poll(cx)) {
52            Ok(ex) => {
53                let ex = (pin.callback.call(ex),);
54                Poll::Ready(Ok(ex))
55            }
56            Err(err) => Poll::Ready(Err(err)),
57        }
58    }
59}
warp/filter/map.rs

warp/filter/
map.rs
