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
use serde::{Deserialize, Serialize};
use std::convert::TryFrom;

#[derive(
    Serialize,
    Deserialize,
    Default,
    Clone,
    Debug,
    Hash,
    PartialEq,
    Eq,
    PartialOrd,
    Ord,
    juniper::GraphQLObject,
)]
pub struct Duration {
    secs: i32,
    nanos: i32,
}

impl Duration {
    pub fn secs(&self) -> u32 {
        u32::try_from(self.secs).unwrap()
    }

    pub fn nanos(&self) -> u32 {
        u32::try_from(self.nanos).unwrap()
    }

    pub fn from_std(
        duration: std::time::Duration,
    ) -> std::result::Result<Duration, Box<dyn std::error::Error>> {
        let total_nanos = duration.as_nanos();
        let secs = total_nanos / 1_000_000_000;
        let nanos = total_nanos - (secs * 1_000_000_000);
        Ok(Duration {
            secs: i32::try_from(secs)?,
            nanos: i32::try_from(nanos)?,
        })
    }

    pub fn to_std(&self) -> std::result::Result<std::time::Duration, Box<dyn std::error::Error>> {
        Ok(std::time::Duration::new(
            u64::try_from(self.secs)?,
            u32::try_from(self.nanos)?,
        ))
    }
}

pub fn format(duration: &std::time::Duration) -> String {
    match duration.as_micros() > 999 {
        // greater than 999 microseconds
        false => format!("{}\u{00b5}s", duration.as_micros()),
        true => match duration.as_millis() > 999 {
            // greater that 999 milliseconds
            false => format!("{}ms", duration.as_millis()),
            true => match duration.as_secs() > 99 {
                // greater than 99 seconds
                false => format!("{}s", duration.as_secs()),
                true => match duration.as_secs() > 5940 {
                    // greater than 99 minutes
                    false => format!("{}m", duration.as_secs() / 60),
                    true => match duration.as_secs() > 356400 {
                        // greater than 99 hours
                        false => format!("{}h", duration.as_secs() / 3600),
                        true => match duration.as_secs() > 8553600 {
                            // greater than 99 days
                            false => format!("{}d", duration.as_secs() / 86400),
                            true => format!("{} months", duration.as_secs() / (86400 * 30)),
                        },
                    },
                },
            },
        },
    }
}

#[cfg(test)]
use crate::duration;

#[test]
fn usage() {}

#[test]
fn format_test() {
    assert_eq!(
        "15µs",
        duration::format(&std::time::Duration::from_micros(15))
    );

    let d = Duration::from_std(std::time::Duration::new(70_000, 4321)).unwrap();
    assert_eq!(70_000, d.secs());
    assert_eq!(4321, d.nanos());
    let d2 = d.to_std().unwrap();
    assert_eq!(70_000, d2.as_secs());
}