338 lines
12 KiB
Rust
338 lines
12 KiB
Rust
use std::fmt;
|
|
use std::time::Duration;
|
|
|
|
use number_prefix::NumberPrefix;
|
|
|
|
const SECOND: Duration = Duration::from_secs(1);
|
|
const MINUTE: Duration = Duration::from_secs(60);
|
|
const HOUR: Duration = Duration::from_secs(60 * 60);
|
|
const DAY: Duration = Duration::from_secs(24 * 60 * 60);
|
|
const WEEK: Duration = Duration::from_secs(7 * 24 * 60 * 60);
|
|
const YEAR: Duration = Duration::from_secs(365 * 24 * 60 * 60);
|
|
|
|
/// Wraps an std duration for human basic formatting.
|
|
#[derive(Debug)]
|
|
pub struct FormattedDuration(pub Duration);
|
|
|
|
/// Wraps an std duration for human readable formatting.
|
|
#[derive(Debug)]
|
|
pub struct HumanDuration(pub Duration);
|
|
|
|
/// Formats bytes for human readability
|
|
#[derive(Debug)]
|
|
pub struct HumanBytes(pub u64);
|
|
|
|
/// Formats bytes for human readability using SI prefixes
|
|
#[derive(Debug)]
|
|
pub struct DecimalBytes(pub u64);
|
|
|
|
/// Formats bytes for human readability using ISO/IEC prefixes
|
|
#[derive(Debug)]
|
|
pub struct BinaryBytes(pub u64);
|
|
|
|
/// Formats counts for human readability using commas
|
|
#[derive(Debug)]
|
|
pub struct HumanCount(pub u64);
|
|
|
|
/// Formats counts for human readability using commas for floats
|
|
#[derive(Debug)]
|
|
pub struct HumanFloatCount(pub f64);
|
|
|
|
impl fmt::Display for FormattedDuration {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
let mut t = self.0.as_secs();
|
|
let seconds = t % 60;
|
|
t /= 60;
|
|
let minutes = t % 60;
|
|
t /= 60;
|
|
let hours = t % 24;
|
|
t /= 24;
|
|
if t > 0 {
|
|
let days = t;
|
|
write!(f, "{days}d {hours:02}:{minutes:02}:{seconds:02}")
|
|
} else {
|
|
write!(f, "{hours:02}:{minutes:02}:{seconds:02}")
|
|
}
|
|
}
|
|
}
|
|
|
|
// `HumanDuration` should be as intuitively understandable as possible.
|
|
// So we want to round, not truncate: otherwise 1 hour and 59 minutes
|
|
// would display an ETA of "1 hour" which underestimates the time
|
|
// remaining by a factor 2.
|
|
//
|
|
// To make the precision more uniform, we avoid displaying "1 unit"
|
|
// (except for seconds), because it would be displayed for a relatively
|
|
// long duration compared to the unit itself. Instead, when we arrive
|
|
// around 1.5 unit, we change from "2 units" to the next smaller unit
|
|
// (e.g. "89 seconds").
|
|
//
|
|
// Formally:
|
|
// * for n >= 2, we go from "n+1 units" to "n units" exactly at (n + 1/2) units
|
|
// * we switch from "2 units" to the next smaller unit at (1.5 unit minus half of the next smaller unit)
|
|
|
|
impl fmt::Display for HumanDuration {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
let mut idx = 0;
|
|
for (i, &(cur, _, _)) in UNITS.iter().enumerate() {
|
|
idx = i;
|
|
match UNITS.get(i + 1) {
|
|
Some(&next) if self.0.saturating_add(next.0 / 2) >= cur + cur / 2 => break,
|
|
_ => continue,
|
|
}
|
|
}
|
|
|
|
let (unit, name, alt) = UNITS[idx];
|
|
// FIXME when `div_duration_f64` is stable
|
|
let mut t = (self.0.as_secs_f64() / unit.as_secs_f64()).round() as usize;
|
|
if idx < UNITS.len() - 1 {
|
|
t = Ord::max(t, 2);
|
|
}
|
|
|
|
match (f.alternate(), t) {
|
|
(true, _) => write!(f, "{t}{alt}"),
|
|
(false, 1) => write!(f, "{t} {name}"),
|
|
(false, _) => write!(f, "{t} {name}s"),
|
|
}
|
|
}
|
|
}
|
|
|
|
const UNITS: &[(Duration, &str, &str)] = &[
|
|
(YEAR, "year", "y"),
|
|
(WEEK, "week", "w"),
|
|
(DAY, "day", "d"),
|
|
(HOUR, "hour", "h"),
|
|
(MINUTE, "minute", "m"),
|
|
(SECOND, "second", "s"),
|
|
];
|
|
|
|
impl fmt::Display for HumanBytes {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
match NumberPrefix::binary(self.0 as f64) {
|
|
NumberPrefix::Standalone(number) => write!(f, "{number:.0} B"),
|
|
NumberPrefix::Prefixed(prefix, number) => write!(f, "{number:.2} {prefix}B"),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for DecimalBytes {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
match NumberPrefix::decimal(self.0 as f64) {
|
|
NumberPrefix::Standalone(number) => write!(f, "{number:.0} B"),
|
|
NumberPrefix::Prefixed(prefix, number) => write!(f, "{number:.2} {prefix}B"),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for BinaryBytes {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
match NumberPrefix::binary(self.0 as f64) {
|
|
NumberPrefix::Standalone(number) => write!(f, "{number:.0} B"),
|
|
NumberPrefix::Prefixed(prefix, number) => write!(f, "{number:.2} {prefix}B"),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for HumanCount {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
use fmt::Write;
|
|
|
|
let num = self.0.to_string();
|
|
let len = num.len();
|
|
for (idx, c) in num.chars().enumerate() {
|
|
let pos = len - idx - 1;
|
|
f.write_char(c)?;
|
|
if pos > 0 && pos % 3 == 0 {
|
|
f.write_char(',')?;
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for HumanFloatCount {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
use fmt::Write;
|
|
|
|
let num = format!("{:.4}", self.0);
|
|
let (int_part, frac_part) = match num.split_once('.') {
|
|
Some((int_str, fract_str)) => (int_str.to_string(), fract_str),
|
|
None => (self.0.trunc().to_string(), ""),
|
|
};
|
|
let len = int_part.len();
|
|
for (idx, c) in int_part.chars().enumerate() {
|
|
let pos = len - idx - 1;
|
|
f.write_char(c)?;
|
|
if pos > 0 && pos % 3 == 0 {
|
|
f.write_char(',')?;
|
|
}
|
|
}
|
|
let frac_trimmed = frac_part.trim_end_matches('0');
|
|
if !frac_trimmed.is_empty() {
|
|
f.write_char('.')?;
|
|
f.write_str(frac_trimmed)?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
const MILLI: Duration = Duration::from_millis(1);
|
|
|
|
#[test]
|
|
fn human_duration_alternate() {
|
|
for (unit, _, alt) in UNITS {
|
|
assert_eq!(format!("2{alt}"), format!("{:#}", HumanDuration(2 * *unit)));
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_less_than_one_second() {
|
|
assert_eq!(
|
|
"0 seconds",
|
|
format!("{}", HumanDuration(Duration::from_secs(0)))
|
|
);
|
|
assert_eq!("0 seconds", format!("{}", HumanDuration(MILLI)));
|
|
assert_eq!("0 seconds", format!("{}", HumanDuration(499 * MILLI)));
|
|
assert_eq!("1 second", format!("{}", HumanDuration(500 * MILLI)));
|
|
assert_eq!("1 second", format!("{}", HumanDuration(999 * MILLI)));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_less_than_two_seconds() {
|
|
assert_eq!("1 second", format!("{}", HumanDuration(1499 * MILLI)));
|
|
assert_eq!("2 seconds", format!("{}", HumanDuration(1500 * MILLI)));
|
|
assert_eq!("2 seconds", format!("{}", HumanDuration(1999 * MILLI)));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_one_unit() {
|
|
assert_eq!("1 second", format!("{}", HumanDuration(SECOND)));
|
|
assert_eq!("60 seconds", format!("{}", HumanDuration(MINUTE)));
|
|
assert_eq!("60 minutes", format!("{}", HumanDuration(HOUR)));
|
|
assert_eq!("24 hours", format!("{}", HumanDuration(DAY)));
|
|
assert_eq!("7 days", format!("{}", HumanDuration(WEEK)));
|
|
assert_eq!("52 weeks", format!("{}", HumanDuration(YEAR)));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_less_than_one_and_a_half_unit() {
|
|
// this one is actually done at 1.5 unit - half of the next smaller unit - epsilon
|
|
// and should display the next smaller unit
|
|
let d = HumanDuration(MINUTE + MINUTE / 2 - SECOND / 2 - MILLI);
|
|
assert_eq!("89 seconds", format!("{d}"));
|
|
let d = HumanDuration(HOUR + HOUR / 2 - MINUTE / 2 - MILLI);
|
|
assert_eq!("89 minutes", format!("{d}"));
|
|
let d = HumanDuration(DAY + DAY / 2 - HOUR / 2 - MILLI);
|
|
assert_eq!("35 hours", format!("{d}"));
|
|
let d = HumanDuration(WEEK + WEEK / 2 - DAY / 2 - MILLI);
|
|
assert_eq!("10 days", format!("{d}"));
|
|
let d = HumanDuration(YEAR + YEAR / 2 - WEEK / 2 - MILLI);
|
|
assert_eq!("78 weeks", format!("{d}"));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_one_and_a_half_unit() {
|
|
// this one is actually done at 1.5 unit - half of the next smaller unit
|
|
// and should still display "2 units"
|
|
let d = HumanDuration(MINUTE + MINUTE / 2 - SECOND / 2);
|
|
assert_eq!("2 minutes", format!("{d}"));
|
|
let d = HumanDuration(HOUR + HOUR / 2 - MINUTE / 2);
|
|
assert_eq!("2 hours", format!("{d}"));
|
|
let d = HumanDuration(DAY + DAY / 2 - HOUR / 2);
|
|
assert_eq!("2 days", format!("{d}"));
|
|
let d = HumanDuration(WEEK + WEEK / 2 - DAY / 2);
|
|
assert_eq!("2 weeks", format!("{d}"));
|
|
let d = HumanDuration(YEAR + YEAR / 2 - WEEK / 2);
|
|
assert_eq!("2 years", format!("{d}"));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_two_units() {
|
|
assert_eq!("2 seconds", format!("{}", HumanDuration(2 * SECOND)));
|
|
assert_eq!("2 minutes", format!("{}", HumanDuration(2 * MINUTE)));
|
|
assert_eq!("2 hours", format!("{}", HumanDuration(2 * HOUR)));
|
|
assert_eq!("2 days", format!("{}", HumanDuration(2 * DAY)));
|
|
assert_eq!("2 weeks", format!("{}", HumanDuration(2 * WEEK)));
|
|
assert_eq!("2 years", format!("{}", HumanDuration(2 * YEAR)));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_less_than_two_and_a_half_units() {
|
|
let d = HumanDuration(2 * SECOND + SECOND / 2 - MILLI);
|
|
assert_eq!("2 seconds", format!("{d}"));
|
|
let d = HumanDuration(2 * MINUTE + MINUTE / 2 - MILLI);
|
|
assert_eq!("2 minutes", format!("{d}"));
|
|
let d = HumanDuration(2 * HOUR + HOUR / 2 - MILLI);
|
|
assert_eq!("2 hours", format!("{d}"));
|
|
let d = HumanDuration(2 * DAY + DAY / 2 - MILLI);
|
|
assert_eq!("2 days", format!("{d}"));
|
|
let d = HumanDuration(2 * WEEK + WEEK / 2 - MILLI);
|
|
assert_eq!("2 weeks", format!("{d}"));
|
|
let d = HumanDuration(2 * YEAR + YEAR / 2 - MILLI);
|
|
assert_eq!("2 years", format!("{d}"));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_two_and_a_half_units() {
|
|
let d = HumanDuration(2 * SECOND + SECOND / 2);
|
|
assert_eq!("3 seconds", format!("{d}"));
|
|
let d = HumanDuration(2 * MINUTE + MINUTE / 2);
|
|
assert_eq!("3 minutes", format!("{d}"));
|
|
let d = HumanDuration(2 * HOUR + HOUR / 2);
|
|
assert_eq!("3 hours", format!("{d}"));
|
|
let d = HumanDuration(2 * DAY + DAY / 2);
|
|
assert_eq!("3 days", format!("{d}"));
|
|
let d = HumanDuration(2 * WEEK + WEEK / 2);
|
|
assert_eq!("3 weeks", format!("{d}"));
|
|
let d = HumanDuration(2 * YEAR + YEAR / 2);
|
|
assert_eq!("3 years", format!("{d}"));
|
|
}
|
|
|
|
#[test]
|
|
fn human_duration_three_units() {
|
|
assert_eq!("3 seconds", format!("{}", HumanDuration(3 * SECOND)));
|
|
assert_eq!("3 minutes", format!("{}", HumanDuration(3 * MINUTE)));
|
|
assert_eq!("3 hours", format!("{}", HumanDuration(3 * HOUR)));
|
|
assert_eq!("3 days", format!("{}", HumanDuration(3 * DAY)));
|
|
assert_eq!("3 weeks", format!("{}", HumanDuration(3 * WEEK)));
|
|
assert_eq!("3 years", format!("{}", HumanDuration(3 * YEAR)));
|
|
}
|
|
|
|
#[test]
|
|
fn human_count() {
|
|
assert_eq!("42", format!("{}", HumanCount(42)));
|
|
assert_eq!("7,654", format!("{}", HumanCount(7654)));
|
|
assert_eq!("12,345", format!("{}", HumanCount(12345)));
|
|
assert_eq!("1,234,567,890", format!("{}", HumanCount(1234567890)));
|
|
}
|
|
|
|
#[test]
|
|
fn human_float_count() {
|
|
assert_eq!("42", format!("{}", HumanFloatCount(42.0)));
|
|
assert_eq!("7,654", format!("{}", HumanFloatCount(7654.0)));
|
|
assert_eq!("12,345", format!("{}", HumanFloatCount(12345.0)));
|
|
assert_eq!(
|
|
"1,234,567,890",
|
|
format!("{}", HumanFloatCount(1234567890.0))
|
|
);
|
|
assert_eq!("42.5", format!("{}", HumanFloatCount(42.5)));
|
|
assert_eq!("42.5", format!("{}", HumanFloatCount(42.500012345)));
|
|
assert_eq!("42.502", format!("{}", HumanFloatCount(42.502012345)));
|
|
assert_eq!("7,654.321", format!("{}", HumanFloatCount(7654.321)));
|
|
assert_eq!("7,654.321", format!("{}", HumanFloatCount(7654.3210123456)));
|
|
assert_eq!("12,345.6789", format!("{}", HumanFloatCount(12345.6789)));
|
|
assert_eq!(
|
|
"1,234,567,890.1235",
|
|
format!("{}", HumanFloatCount(1234567890.1234567))
|
|
);
|
|
assert_eq!(
|
|
"1,234,567,890.1234",
|
|
format!("{}", HumanFloatCount(1234567890.1234321))
|
|
);
|
|
}
|
|
}
|