3 unstable releases

0.2.1 May 8, 2022
0.2.0 May 8, 2022
0.1.1 Feb 28, 2022
0.1.0 Feb 28, 2022

#7 in #token-bucket

MIT/Apache

37KB
702 lines

Tocket

crates.io docs.rs build

This library provides implementation of token bucket algorithm and some storage implementations.

[dependencies]
tocket = "0.2"

Example

use tocket::{TokenBucket, InMemoryStorage};
use std::sync::Arc;
use std::time::Duration;

fn main() {
    let tb = TokenBucket::new(InMemoryStorage::new(100));
    let tb = Arc::new(tb);

    for i in 0..4 {
        std::thread::spawn({
            let tb = Arc::clone(&tb);
            move || {
                loop {
                    std::thread::sleep(Duration::from_secs(i));
                    
                    match tb.try_acquire_one() {
                        Ok(_) => println!("token acquired, limit not exceeded"),
                        Err(err) => eprintln!("token acquiring failed: {}", err),
                    }
                }
            }
        });
    }
    
    loop {}
}

Features

  • redis-impl - redis storage implementation
  • distributed-impl - distributed storage implementation

License

Licensed under either of Apache License, Version 2.0 or MIT license at your option.
Unless you explicitly state otherwise, any contribution intentionally submitted for inclusion in this crate by you, as defined in the Apache-2.0 license, shall be dual licensed as above, without any additional terms or conditions.

Dependencies

~2–12MB
~133K SLoC