rate_limiters 0.1.13

Flexible and easy-to-use Rust library for implementing rate limits.
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
32
33
34
35
36
37
38
39
40

// cargo run --example token_bucket_usage
use std::thread;
use std::time::{Duration, Instant};

use rate_limiters::token_bucket::{TokenBucket, TokenBucketConfig};
use rate_limiters::RateLimiter;

fn main() {
    let start = Instant::now();
    let mut bucket = TokenBucket::new(TokenBucketConfig {
        capacity: 5,
        refill_rate: 2,
    });

    for i in 0..100 {
        bucket.refresh();
        let limit = bucket.get_limit();
        let remaining = bucket.get_remaining();
        let used = bucket.get_used();
        let reset = bucket.get_reset();
        let is_acquired = bucket.try_acquire(1);

        let elapsed = start.elapsed().as_secs_f32();
        println!(
            "[{elapsed:5.2}s] Request #{:03} | {:<12} | Limit: {:2} | Remaining: {:2} | Used: {:2} | Reset: {}",
            i + 1,
            if is_acquired {
                "Allowed"
            } else {
                "Rate limited"
            },
            limit,
            remaining,
            used,
            reset
        );

        thread::sleep(Duration::from_millis(300));
    }
}