mirror of
https://github.com/async-rs/async-std.git
synced 2025-03-28 20:16:41 +00:00
feat: Add stream::from_iter
This commit is contained in:
parent
da795dec7b
commit
40c4e1a29d
2 changed files with 46 additions and 0 deletions
44
src/stream/from_iter.rs
Normal file
44
src/stream/from_iter.rs
Normal file
|
@ -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 empty::{empty, Empty};
|
||||||
pub use from_fn::{from_fn, FromFn};
|
pub use from_fn::{from_fn, FromFn};
|
||||||
|
pub use from_iter::{from_iter, FromIter};
|
||||||
pub use once::{once, Once};
|
pub use once::{once, Once};
|
||||||
pub use repeat::{repeat, Repeat};
|
pub use repeat::{repeat, Repeat};
|
||||||
pub use repeat_with::{repeat_with, RepeatWith};
|
pub use repeat_with::{repeat_with, RepeatWith};
|
||||||
|
@ -313,6 +314,7 @@ pub(crate) mod stream;
|
||||||
|
|
||||||
mod empty;
|
mod empty;
|
||||||
mod from_fn;
|
mod from_fn;
|
||||||
|
mod from_iter;
|
||||||
mod once;
|
mod once;
|
||||||
mod repeat;
|
mod repeat;
|
||||||
mod repeat_with;
|
mod repeat_with;
|
||||||
|
|
Loading…
Reference in a new issue