Tocket
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