limtr 0.2.0

Implement rate limits for all sort of applications and scenarios.
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

use std::collections::HashMap;
use std::sync::Arc;
use once_cell::sync::OnceCell;
use tokio::sync::Mutex;
use tokio::task::JoinHandle;
use crate::back::*;
use crate::{Error, FeatureRaw, Limtr};
use indexmap::IndexMap;


pub(crate) type Blocks = Arc<Mutex<HashMap<FeatureRaw, Usage>>>;
pub(crate) type IndexMapLimtr = IndexMap::<String, Blocks>;

pub(crate) static LIMTR: OnceCell<Limtr> = OnceCell::new();

pub(crate) fn run(buffer: usize) -> Limtr {
    use tokio::sync::mpsc;

    let (tx, mut rx) = mpsc::channel::<Directive>(buffer);

    let _: JoinHandle<Result<(), Error>> = tokio::spawn(async move {
        let mut index_map = IndexMapLimtr::new();

        while let Some(directive) = rx.recv().await {
            directive.handle(&mut index_map).await;
        }

        Err(Error::LimtrClosed)
    });

    Limtr {
        tx,
    }
}