Skip to content

Commit 448407c

Browse files
committed
refactor(aggregator): remove sleep call in test and reduce duplicate code to create log message
1 parent b5e416f commit 448407c

File tree

1 file changed

+13
-20
lines changed

1 file changed

+13
-20
lines changed

mithril-aggregator/src/tools/vacuum_tracker.rs

Lines changed: 13 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -54,25 +54,20 @@ impl VacuumTracker {
5454
let duration_since_last = Utc::now() - (last_vacuum);
5555

5656
let should_vacuum = duration_since_last >= self.min_interval;
57-
58-
if should_vacuum {
59-
info!(
60-
self.logger,
61-
"Sufficient time has passed since last vacuum";
62-
"last_vacuum" => last_vacuum.to_string(),
63-
"elapsed_days" => duration_since_last.num_days(),
64-
"min_interval_days" => self.min_interval.num_days()
65-
);
57+
let info_message = if should_vacuum {
58+
"Sufficient time has passed since last vacuum"
6659
} else {
67-
info!(
68-
self.logger,
69-
"Not enough time elapsed since last vacuum";
70-
"last_vacuum" => last_vacuum.to_string(),
71-
"elapsed_days" => duration_since_last.num_days(),
72-
"min_interval_days" => self.min_interval.num_days()
73-
);
60+
"Not enough time elapsed since last vacuum"
7461
};
7562

63+
info!(
64+
self.logger,
65+
"{}", info_message;
66+
"last_vacuum" => last_vacuum.to_string(),
67+
"elapsed_days" => duration_since_last.num_days(),
68+
"min_interval_days" => self.min_interval.num_days()
69+
);
70+
7671
Ok((should_vacuum, Some(last_vacuum)))
7772
}
7873

@@ -93,8 +88,6 @@ impl VacuumTracker {
9388

9489
#[cfg(test)]
9590
mod tests {
96-
use std::thread::sleep;
97-
9891
use mithril_common::temp_dir_create;
9992

10093
use crate::test_tools::TestLogger;
@@ -159,8 +152,8 @@ mod tests {
159152
let min_interval = TimeDelta::milliseconds(10);
160153
let tracker = VacuumTracker::new(&temp_dir_create!(), min_interval, TestLogger::stdout());
161154

162-
let saved_timestamp = tracker.update_last_vacuum_time().unwrap();
163-
sleep(min_interval.to_std().unwrap());
155+
let saved_timestamp = Utc::now() - TimeDelta::milliseconds(10);
156+
fs::write(tracker.clone().tracker_file, saved_timestamp.to_rfc3339()).unwrap();
164157

165158
let (is_vacuum_needed, last_timestamp) = tracker.check_vacuum_needed().unwrap();
166159

0 commit comments

Comments
 (0)