hatter/rust_util
1
0
Fork 0
mirror of https://github.com/jht5945/rust_util.git synced 2025-12-27 15:40:03 +08:00
Code Issues Releases Wiki Activity

add parse_size()

Browse Source
This commit is contained in:
Hatter Jiang
2019-07-22 01:22:09 +08:00
parent 6a08f8f92f
commit 37bde2e066
1 changed files with 24 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/lib.rs
View File

@@ -152,6 +152,26 @@ pub fn print_lastline(line: &str) {
flush_stdout();
}
pub fn parse_size(size: &str) -> XResult<i64> {
let lower_size = size.to_lowercase();
if lower_size.ends_with("b") {
let no_last_b_size = &lower_size[0..lower_size.len()-1];
if no_last_b_size.ends_with("k") {
return Ok((SIZE_KB as f64 * no_last_b_size[0..no_last_b_size.len()-1].parse::<f64>()?) as i64);
} else if no_last_b_size.ends_with("m") {
return Ok((SIZE_MB as f64 * no_last_b_size[0..no_last_b_size.len()-1].parse::<f64>()?) as i64);
} else if no_last_b_size.ends_with("g") {
return Ok((SIZE_GB as f64 * no_last_b_size[0..no_last_b_size.len()-1].parse::<f64>()?) as i64);
} else if no_last_b_size.ends_with("t") {
return Ok((SIZE_TB as f64 * no_last_b_size[0..no_last_b_size.len()-1].parse::<f64>()?) as i64);
} else if no_last_b_size.ends_with("p") {
return Ok((SIZE_PB as f64 * no_last_b_size[0..no_last_b_size.len()-1].parse::<f64>()?) as i64);
}
}
Err(new_box_error(&format!("Cannot parse size: {}", size)))
}
pub fn get_display_size(size: i64) -> String {
if size < SIZE_KB {
return size.to_string();
@@ -159,12 +179,12 @@ pub fn get_display_size(size: i64) -> String {
return format!("{:.*}KB", 2, (size as f64) / 1024.);
} else if size < SIZE_GB {
return format!("{:.*}MB", 2, (size as f64) / 1024. / 1024.);
} else if size < SIZE_PB {
return format!("{:.*}GB", 2, (size as f64) / 1024. / 1024. / 1024.);
} else if size < SIZE_TB {
return format!("{:.*}PB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024.);
return format!("{:.*}GB", 2, (size as f64) / 1024. / 1024. / 1024.);
} else if size < SIZE_PB {
return format!("{:.*}TB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024.);
} else {
return format!("{:.*}TB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024. / 1024.);
return format!("{:.*}PB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024. / 1024.);
}
}