chj_util/
time_guard.rs

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
55
56
57
//! Tools for performance debugging.

use std::{time::Instant, fmt::Debug};


#[macro_export]
macro_rules! time {
    ($name:expr; $($code:tt)*) => {{
        let msg = format!("time {}", $name);
        let now = std::time::Instant::now();
        let r = {
            $($code)*
        };
        let elapsed = now.elapsed();
        eprintln!("{msg}: {elapsed:?} at {:?} line {}", file!(), line!());
        r
    }}
}

#[macro_export]
macro_rules! notime {
    ($name:expr; $($code:tt)*) => {{
        $($code)*
    }}
}



// Reminiscent of dt.rs (DtGuard)

pub struct TimeGuard<S: Debug> {
    pub name: S,
    pub start: Instant
}

impl<S: Debug> Drop for TimeGuard<S> {
    fn drop(&mut self) {
        let elapsed = self.start.elapsed();
        eprintln!("{:?}: {:#?}", self.name, elapsed);
    }
}

#[macro_export]
macro_rules! time_guard {
    ($namestr:expr) => {
        let _guard = time_guard::TimeGuard {
            name: $namestr,
            start: std::time::Instant::now()
        };
    }
}

#[macro_export]
macro_rules! notime_guard {
    ($namestr:expr) => {}
}