futures-stream-ext 0.5.0

Asynchronous stream extensions for the futures(-core) crate
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

// SPDX-FileCopyrightText: The futures-stream-ext authors
// SPDX-License-Identifier: MPL-2.0

use std::time::Duration;

use futures_core::Stream;
use tokio::time::Sleep;

use crate::{StreamExt, ThrottleIntervalConfig, Throttled};

mod debounce;

mod throttle;
pub use self::throttle::IntervalThrottler;

impl crate::Sleep for tokio::time::Sleep {
    fn sleep(duration: Duration) -> Self {
        tokio::time::sleep(duration)
    }
}

impl<S: Stream> StreamExt for S {
    type Sleep = Sleep;
    type IntervalThrottler = IntervalThrottler<S::Item>;

    fn throttle_interval(
        self,
        config: ThrottleIntervalConfig,
        poll_next_max_ready_count: std::num::NonZeroUsize,
    ) -> Throttled<Self, Self::IntervalThrottler>
    where
        Self: Sized,
    {
        let throttler = IntervalThrottler::new(config);
        self.throttle(throttler, poll_next_max_ready_count)
    }
}