forked from mirror/async-std
42 lines
866 B
Rust
42 lines
866 B
Rust
use std::pin::Pin;
|
|
|
|
use crate::task::{Context, Poll};
|
|
|
|
/// Creates a stream that yields a single item.
|
|
///
|
|
/// # Examples
|
|
///
|
|
/// ```
|
|
/// # #![feature(async_await)]
|
|
/// # fn main() { async_std::task::block_on(async {
|
|
/// #
|
|
/// use async_std::{prelude::*, stream};
|
|
///
|
|
/// let mut s = stream::once(7);
|
|
///
|
|
/// assert_eq!(s.next().await, Some(7));
|
|
/// assert_eq!(s.next().await, None);
|
|
/// #
|
|
/// # }) }
|
|
/// ```
|
|
pub fn once<T>(t: T) -> Once<T> {
|
|
Once { value: Some(t) }
|
|
}
|
|
|
|
/// A stream that yields a single item.
|
|
///
|
|
/// This stream is constructed by the [`once`] function.
|
|
///
|
|
/// [`once`]: fn.once.html
|
|
#[derive(Debug)]
|
|
pub struct Once<T> {
|
|
value: Option<T>,
|
|
}
|
|
|
|
impl<T: Unpin> futures::Stream for Once<T> {
|
|
type Item = T;
|
|
|
|
fn poll_next(mut self: Pin<&mut Self>, _: &mut Context<'_>) -> Poll<Option<T>> {
|
|
Poll::Ready(self.value.take())
|
|
}
|
|
}
|