sshping 0.2.4

SSH-based ping that measures interactive character echo latency and file transfer throughput. Pronounced "shipping".
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

use std::time::Duration;

use num_format::{Buffer, CustomFormat};
use size::{Base, Size, Style};

pub struct Formatter {
    // Formatter style for large number
    // Only used when human_readable is false
    format: Option<CustomFormat>,
}

impl Formatter {
    pub fn new(human_readable: bool, delimit: Option<char>) -> Self {
        let format = (!human_readable).then(|| {
            CustomFormat::builder()
                .separator(delimit.map(|ch| ch.to_string()).unwrap_or_default())
                .build()
                .unwrap()
        });

        Self { format }
    }

    pub fn format_duration(&self, time: Duration) -> String {
        if let Some(format) = &self.format {
            let mut buffer = Buffer::new();
            buffer.write_formatted(&time.as_nanos(), format);
            buffer.as_str().to_string() + "ns"
        } else {
            let formatted = humantime::format_duration(time).to_string();
            let parts = formatted.split(" ").collect::<Vec<&str>>();
            if parts.len() > 2 {
                parts[..2].join(" ")
            } else {
                formatted
            }
        }
    }

    pub fn format_size(&self, size: u64) -> String {
        if let Some(format) = &self.format {
            let mut buffer = Buffer::new();
            buffer.write_formatted(&size, format);
            buffer.as_str().to_string() + " B"
        } else {
            Size::from_bytes(size)
                .format()
                .with_base(Base::Base10)
                .with_style(Style::Abbreviated)
                .to_string()
        }
    }
}