feat: Add stream::from_iter

pull/415/head
k-nasa 5 years ago
parent da795dec7b
commit 40c4e1a29d

@ -0,0 +1,44 @@
use std::pin::Pin;
use pin_project_lite::pin_project;
use crate::stream::Stream;
use crate::task::{Context, Poll};
pin_project! {
#[derive(Debug)]
pub struct FromIter<I> {
iter: I,
}
}
/// # Examples
///```
/// # async_std::task::block_on(async {
/// #
/// use async_std::prelude::*;
/// use async_std::stream;
///
/// let mut s = stream::from_iter(vec![0, 1, 2, 3]);
///
/// assert_eq!(s.next().await, Some(0));
/// assert_eq!(s.next().await, Some(1));
/// assert_eq!(s.next().await, Some(2));
/// assert_eq!(s.next().await, Some(3));
/// assert_eq!(s.next().await, None);
/// #
/// # })
///````
pub fn from_iter<I: IntoIterator>(iter: I) -> FromIter<<I as std::iter::IntoIterator>::IntoIter> {
FromIter {
iter: iter.into_iter(),
}
}
impl<I: Iterator> Stream for FromIter<I> {
type Item = I::Item;
fn poll_next(mut self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
Poll::Ready(self.iter.next())
}
}

@ -302,6 +302,7 @@
pub use empty::{empty, Empty};
pub use from_fn::{from_fn, FromFn};
pub use from_iter::{from_iter, FromIter};
pub use once::{once, Once};
pub use repeat::{repeat, Repeat};
pub use repeat_with::{repeat_with, RepeatWith};
@ -313,6 +314,7 @@ pub(crate) mod stream;
mod empty;
mod from_fn;
mod from_iter;
mod once;
mod repeat;
mod repeat_with;

Loading…
Cancel
Save