burster 0.1.2

Lightweight stack allocated rate limiter implementations
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

use burster::Limiter;
use rand::Rng;
use std::{
    thread,
    time::{Duration, Instant},
};

fn main() {
    // Set up the limiter to allow 100 consumes per second (window_width = 1s)
    let mut bucket = burster::fixed_window(100, 1000);

    let start = Instant::now();
    let mut pass = 0;
    let mut rng = rand::thread_rng();

    println!("Trying to consume 100 000 tokens with varying intervals");
    for _ in 0..100_000 {
        if bucket.try_consume_one().is_ok() {
            pass += 1
        }
        thread::sleep(Duration::from_micros(rng.gen_range(0..10)));
    }

    // Resulting rate should be around 100/s.
    // There can be some variation since this is not a perfect simulation.
    let elapsed = start.elapsed().as_secs_f64();
    let rate_per_s = pass as f64 / elapsed;
    println!("Average pass rate {rate_per_s}");
}