blob: 23a99819bd9f316a1f4cfd718f78243576324675 [file] [log] [blame]
// This should really be in the stream module,
// but `pub(crate)` isn't available until Rust 1.18,
// and pre-1.18 there isn't a really good way to have a sub-module
// available to the crate, but not without it.
use core::marker::PhantomData;
use {Poll, Async};
use stream::Stream;
/// A stream combinator used to convert a `Stream<Item=T,Error=E>`
/// to a `Stream<Item=Result<T,E>>`.
///
/// A poll on this stream will never return an `Err`. As such the
/// actual error type is parameterized, so it can match whatever error
/// type is needed.
///
/// This structure is produced by the `Stream::results` method.
#[derive(Debug)]
#[must_use = "streams do nothing unless polled"]
pub struct Results<S: Stream, E> {
inner: S,
phantom: PhantomData<E>
}
pub fn new<S, E>(s: S) -> Results<S, E> where S: Stream {
Results {
inner: s,
phantom: PhantomData
}
}
impl<S: Stream, E> Stream for Results<S, E> {
type Item = Result<S::Item, S::Error>;
type Error = E;
fn poll(&mut self) -> Poll<Option<Result<S::Item, S::Error>>, E> {
match self.inner.poll() {
Ok(Async::Ready(Some(item))) => Ok(Async::Ready(Some(Ok(item)))),
Err(e) => Ok(Async::Ready(Some(Err(e)))),
Ok(Async::Ready(None)) => Ok(Async::Ready(None)),
Ok(Async::NotReady) => Ok(Async::NotReady)
}
}
}