diff options
author | Blas Rodriguez Irizar <rodrigblas@gmail.com> | 2020-09-02 11:52:31 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-09-02 11:52:31 +0200 |
commit | 5cdb6f8fd6be35f971d8ef7ea8984f4d01965620 (patch) | |
tree | 924787133275b3fd92d3b6e53382e2e32024c79f /tokio/src/stream/mod.rs | |
parent | 5a1a6dc90c6d5a7eb5f31ae215f9ec383d6767aa (diff) |
time: move throttle to StreamExt (#2752)
Ref: #2727
Diffstat (limited to 'tokio/src/stream/mod.rs')
-rw-r--r-- | tokio/src/stream/mod.rs | 30 |
1 files changed, 29 insertions, 1 deletions
diff --git a/tokio/src/stream/mod.rs b/tokio/src/stream/mod.rs index 7b061efe..bc6eea23 100644 --- a/tokio/src/stream/mod.rs +++ b/tokio/src/stream/mod.rs @@ -71,7 +71,9 @@ use take_while::TakeWhile; cfg_time! { mod timeout; use timeout::Timeout; - use std::time::Duration; + use crate::time::Duration; + mod throttle; + use crate::stream::throttle::{throttle, Throttle}; } pub use futures_core::Stream; @@ -819,6 +821,32 @@ pub trait StreamExt: Stream { { Timeout::new(self, duration) } + /// Slows down a stream by enforcing a delay between items. + /// + /// # Example + /// + /// Create a throttled stream. + /// ```rust,no_run + /// use std::time::Duration; + /// use tokio::stream::StreamExt; + /// + /// # async fn dox() { + /// let mut item_stream = futures::stream::repeat("one").throttle(Duration::from_secs(2)); + /// + /// loop { + /// // The string will be produced at most every 2 seconds + /// println!("{:?}", item_stream.next().await); + /// } + /// # } + /// ``` + #[cfg(all(feature = "time"))] + #[cfg_attr(docsrs, doc(cfg(feature = "time")))] + fn throttle(self, duration: Duration) -> Throttle<Self> + where + Self: Sized, + { + throttle(duration, self) + } } impl<St: ?Sized> StreamExt for St where St: Stream {} |