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