From 713cbdc67f3122fa1c9da81b97e638468202f954 Mon Sep 17 00:00:00 2001 From: Hatter Jiang Date: Fri, 1 May 2020 22:04:18 +0800 Subject: [PATCH] fix size --- src/main.rs | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/main.rs b/src/main.rs index 948e422..066473f 100644 --- a/src/main.rs +++ b/src/main.rs @@ -50,9 +50,14 @@ impl CountCell { self.cell.get() } + #[inline] + fn add(&self, i: u64) { + self.cell.set(self.cell.get() + i); + } + #[inline] fn add_one(&self) { - self.cell.set(self.cell.get() + 1); + self.add(1); } } @@ -86,7 +91,7 @@ fn find_huge_files(options: &Options, dir_path: &Path) { let len = metadata.len(); if len >= options.parsed_huge_file_size { huge_file_count.add_one(); - huge_file_size.add_one(); + huge_file_size.add(len); clear_n_print_message(MessageType::OK, &format!("{} [{}]", p_str, get_display_size(len as i64))); } },