crux_time/protocol/
duration.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
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
use serde::{Deserialize, Serialize};

/// The number of nanoseconds in seconds.
const NANOS_PER_SEC: u32 = 1_000_000_000;
/// The number of nanoseconds in a millisecond.
const NANOS_PER_MILLI: u32 = 1_000_000;

/// Represents a duration of time, internally stored as nanoseconds
#[derive(Clone, Copy, Debug, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Duration {
    pub(crate) nanos: u64,
}

impl Duration {
    /// Create a new `Duration` from the given number of nanoseconds.
    pub fn new(nanos: u64) -> Self {
        Self { nanos }
    }

    /// Create a new `Duration` from the given number of milliseconds.
    ///
    /// Panics if the number of milliseconds
    /// would overflow when converted to nanoseconds.
    pub fn from_millis(millis: u64) -> Self {
        let nanos = millis
            .checked_mul(NANOS_PER_MILLI as u64)
            .expect("millis overflow");
        Self { nanos }
    }

    /// Create a new `Duration` from the given number of seconds.
    ///
    /// Panics if the number of seconds
    /// would overflow when converted to nanoseconds.
    pub fn from_secs(seconds: u64) -> Self {
        let nanos = seconds
            .checked_mul(NANOS_PER_SEC as u64)
            .expect("seconds overflow");
        Self { nanos }
    }
}

impl From<std::time::Duration> for Duration {
    fn from(duration: std::time::Duration) -> Self {
        Duration {
            nanos: duration.as_nanos() as u64,
        }
    }
}

impl From<Duration> for std::time::Duration {
    fn from(duration: Duration) -> Self {
        std::time::Duration::from_nanos(duration.nanos)
    }
}

#[cfg(test)]
mod test {
    use super::Duration;
    use std::time::Duration as StdDuration;

    #[test]
    fn duration_from_millis() {
        let duration = Duration::from_millis(1_000);
        assert_eq!(duration.nanos, 1_000_000_000);
    }

    #[test]
    fn duration_from_secs() {
        let duration = Duration::from_secs(1);
        assert_eq!(duration.nanos, 1_000_000_000);
    }

    #[test]
    fn std_into_duration() {
        let actual: Duration = StdDuration::from_millis(100).into();
        let expected = Duration { nanos: 100_000_000 };
        assert_eq!(actual, expected);
    }

    #[test]
    fn duration_into_std() {
        let actual: StdDuration = Duration { nanos: 100_000_000 }.into();
        let expected = StdDuration::from_nanos(100_000_000);
        assert_eq!(actual, expected);
    }
}