benchmark-rs 0.1.3

Benchmarks for Rust libraries
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use std::fmt::{Display, Formatter};
use std::ops::AddAssign;
use std::time::{Duration, Instant};

use chrono::{DateTime, Utc};

/// Measure elapsed time
pub struct StopWatch {
    accumulated: Duration,
    checkpoint: Instant,
    is_stopped: bool,
}

impl StopWatch {
    /// Create a new [StopWatch]
    pub fn new() -> StopWatch {
        StopWatch {
            accumulated: Duration::from_secs(0),
            checkpoint: Instant::now(),
            is_stopped: true,
        }
    }

    /// Start duration measurement
    pub fn start(&mut self) {
        if self.is_stopped {
            self.checkpoint = Instant::now();
            self.is_stopped = false;
        }
    }

    /// Resume duration measurement
    pub fn resume(&mut self) {
        self.start()
    }

    /// Reset duration measurement
    pub fn reset(&mut self) {
        self.accumulated = Duration::from_secs(0);
        self.checkpoint = Instant::now();
        self.is_stopped = true
    }

    /// Stop duration measurement
    pub fn stop(&mut self) {
        if !self.is_stopped {
            self.accumulated.add_assign(Duration::from_nanos(
                self.checkpoint.elapsed().as_nanos() as u64
            ));
            self.is_stopped = true;
        }
    }

    /// Pause duration measurement
    pub fn pause(&mut self) {
        self.stop()
    }

    /// Get accumulated duration
    pub fn accumulated(&self) -> Duration {
        self.accumulated
    }
}

impl Display for StopWatch {
    /// Format accumulated duration as time
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        let mut accumulated = self.accumulated;
        if !self.is_stopped {
            accumulated.add_assign(Duration::from_nanos(
                self.checkpoint.elapsed().as_nanos() as u64
            ));
        }
        let datetime = DateTime::<Utc>::from_timestamp(
            accumulated.as_secs() as i64,
            accumulated.subsec_nanos(),
        )
        .unwrap();
        let formatted_time = datetime.format("%H:%M:%S.%3f").to_string();
        f.write_fmt(format_args!("{}", formatted_time))
    }
}

impl Default for StopWatch {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(test)]
mod tests {
    use std::thread;
    use std::time::Duration;

    use super::*;

    #[test]
    fn test_pause() {
        let mut stop_watch = StopWatch::new();
        stop_watch.resume();
        thread::sleep(Duration::from_millis(3));
        stop_watch.pause();
        thread::sleep(Duration::from_millis(3));
        stop_watch.resume();
        assert!(stop_watch.accumulated() >= Duration::from_millis(3));
        assert!(stop_watch.accumulated() <= Duration::from_millis(6));
    }

    #[test]
    fn test_stopwatch() {
        let mut stopwatch = StopWatch::new();
        assert_eq!("00:00:00.000", stopwatch.to_string());
        stopwatch.start();
        thread::sleep(Duration::from_millis(3));
        assert!("00:00:00.003".to_string() <= stopwatch.to_string());
        thread::sleep(Duration::from_millis(1000));
        assert!("00:00:01.003".to_string() <= stopwatch.to_string());
    }
}