forked from mirror/async-std
feat: Add stream::from_iter
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())
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue