neighborly 0.0.1

Tools for managing distributed workloads.
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

/// Util API

use tokio::time::Instant;

use super::SimpleRateLimiter;

impl SimpleRateLimiter {
    /// Remove expired timestamps from the history.
    pub fn clean(&self) {
        let mut history = self
            .history
            .lock()
            .expect("rate limiter mutex was poisoned");
        self.clean_history(&mut history, Instant::now());
    }
    /// Returns the number of scheduled, inflight requests. This is
    /// equal to the length of the underlying `VecDeque<Instant>` after
    /// discarding expired entries.
    pub fn len(&self) -> usize {
        let mut history = self
            .history
            .lock()
            .expect("rate limiter mutex was poisoned");
        self.clean_history(&mut history, Instant::now());
        history.len()
    }
    /// Remove all timestamps from the history. Once the history is wiped,
    /// we can no longer ensure we do not violate the rate limit.
    pub unsafe fn clear(&self) {
        let mut history = self
            .history
            .lock()
            .expect("rate limiter mutex was poisoned");
        history.clear();
    }
}