futures_util/future/
recover.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
use core::marker::PhantomData;

use futures_core::{Future, Poll, Async};
use futures_core::task;

/// Future for the `recover` combinator, handling errors by converting them into
/// an `Item`, compatible with any error type of the caller's choosing.
#[must_use = "futures do nothing unless polled"]
#[derive(Debug)]
pub struct Recover<A, E, F> {
    inner: A,
    f: Option<F>,
    err: PhantomData<E>,
}

pub fn new<A, E, F>(future: A, f: F) -> Recover<A, E, F>
    where A: Future
{
    Recover { inner: future, f: Some(f), err: PhantomData }
}

impl<A, E, F> Future for Recover<A, E, F>
    where A: Future,
          F: FnOnce(A::Error) -> A::Item,
{
    type Item = A::Item;
    type Error = E;

    fn poll(&mut self, cx: &mut task::Context) -> Poll<A::Item, E> {
        match self.inner.poll(cx) {
            Err(e) => Ok(Async::Ready((self.f.take().expect("Polled future::Recover after completion"))(e))),
            Ok(x) => Ok(x),
        }
    }
}