libqueued 0.13.0

Library for queued
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

use chrono::Utc;

pub(crate) struct Throttler {
  // State.
  count: u64,
  time_base: i64,
  // Configuration.
  max: u64,
  window: i64,
}

impl Throttler {
  pub fn new(max_reqs_per_time_window: u64, time_window_sec: i64) -> Self {
    assert!(time_window_sec > 1);
    Self {
      count: 0,
      time_base: i64::MIN,
      max: max_reqs_per_time_window,
      window: time_window_sec,
    }
  }

  pub fn increment_count(&mut self) -> bool {
    let cur_window = Utc::now().timestamp() / self.time_base;
    if self.window != cur_window {
      self.window = cur_window;
      self.count = 0;
    };
    self.count += 1;
    self.count <= self.max
  }

  pub fn get_max_reqs_per_time_window(&self) -> u64 {
    self.max
  }

  pub fn get_time_window_sec(&self) -> i64 {
    self.time_base
  }
}