leaky-bucket 1.1.2

A token-based rate limiter based on the leaky bucket algorithm.
Documentation 
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

use std::future::Future;
use std::sync::Arc;
use std::task::Context;

use leaky_bucket::RateLimiter;

struct Waker;

impl std::task::Wake for Waker {
    fn wake(self: Arc<Self>) {}
}

#[tokio::main]
async fn main() {
    let limiter = Arc::new(RateLimiter::builder().build());

    let waker = Arc::new(Waker).into();
    let mut cx = Context::from_waker(&waker);

    let mut a0 = Box::pin(limiter.acquire(1));
    // Poll once to ensure that the core task is assigned.
    assert!(a0.as_mut().poll(&mut cx).is_pending());
    assert!(a0.is_core());

    // We leak the core task, preventing the rate limiter from making progress
    // by assigning new core tasks.
    std::mem::forget(a0);

    println!("Blocking forever...");
    limiter.acquire(1).await;
}