rtools 0.9.0

Set of tools for my needs
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
43
44
45
46
47
48
49
50
51
52
53
54

use std::{collections::BTreeMap, sync::Mutex};

static STORAGE: Mutex<BTreeMap<String, usize>> = Mutex::new(BTreeMap::new());

pub fn _impl_times(label: impl ToString, counter: usize, mut closure: impl FnMut()) {
    let label = label.to_string();
    let mut storage = STORAGE.lock().unwrap();

    let stored_counter = if let Some(counter) = storage.get_mut(&label) {
        counter
    } else {
        storage.insert(label.clone(), 0);
        storage.get_mut(&label).unwrap()
    };

    if *stored_counter >= counter {
        return;
    }

    closure();

    *stored_counter += 1;
}

#[macro_export]
macro_rules! times {
    ($n:expr, $closure:expr) => {{
        rtools::_impl_times(format!("{}:{}", file!(), line!()), $n, $closure);
    }};
}

#[cfg(test)]
mod test {
    use crate::_impl_times;

    #[test]
    fn test() {
        let val = &mut 0;

        let mut inc = || {
            *val += 1;
        };

        _impl_times("a", 2, &mut inc);
        _impl_times("a", 2, &mut inc);
        _impl_times("a", 2, &mut inc);
        _impl_times("a", 2, &mut inc);
        _impl_times("a", 2, &mut inc);
        _impl_times("a", 2, &mut inc);
        _impl_times("a", 2, &mut inc);

        assert_eq!(*val, 2);
    }
}