anystore 0.2.1

Polymorphic, type-safe, composable async API for arbitrary stores
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
41
42

use tokio::{
    sync::Mutex,
    time::{sleep, Duration, Instant},
};

pub struct Ratelimiter {
    pub duration: Duration,
    pub count: usize,

    value: Mutex<(Instant, usize)>,
}

impl Ratelimiter {
    pub fn new(duration: Duration, count: usize) -> Ratelimiter {
        Ratelimiter {
            duration,
            count,
            value: Mutex::new((Instant::now(), 0)),
        }
    }
    pub async fn ask(&self) {
        loop {
            let now = Instant::now();

            let left = {
                let mut v = self.value.lock().await;

                if now - v.0 > self.duration {
                    *v = (now, 0);
                    return;
                } else if v.1 < self.count - 1 {
                    v.1 += 1;
                    return;
                }

                self.duration - (now - v.0)
            };

            sleep(left).await;
        }
    }
}