warp/filter/
untuple_one.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, Internal, Tuple};
9
10#[derive(Clone, Copy, Debug)]
11pub struct UntupleOne<F> {
12    pub(super) filter: F,
13}
14
15impl<F, T> FilterBase for UntupleOne<F>
16where
17    F: Filter<Extract = (T,)>,
18    T: Tuple,
19{
20    type Extract = T;
21    type Error = F::Error;
22    type Future = UntupleOneFuture<F>;
23    #[inline]
24    fn filter(&self, _: Internal) -> Self::Future {
25        UntupleOneFuture {
26            extract: self.filter.filter(Internal),
27        }
28    }
29}
30
31#[allow(missing_debug_implementations)]
32#[pin_project]
33pub struct UntupleOneFuture<F: Filter> {
34    #[pin]
35    extract: F::Future,
36}
37
38impl<F, T> Future for UntupleOneFuture<F>
39where
40    F: Filter<Extract = (T,)>,
41    T: Tuple,
42{
43    type Output = Result<T, F::Error>;
44
45    #[inline]
46    fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
47        match ready!(self.project().extract.try_poll(cx)) {
48            Ok((t,)) => Poll::Ready(Ok(t)),
49            Err(err) => Poll::Ready(Err(err)),
50        }
51    }
52}