heimoshuiyu/rust-synapse-compress-state
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add no-progress-bars feature so other packages can hide them (#66)

Browse Source
This commit is contained in:
Azrenbeth
2021-09-27 11:35:40 +01:00
committed by GitHub
parent a069d8765a
commit bc30942e2d
6 changed files with 34 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/database.rs
View File

@@ -373,7 +373,12 @@ fn get_initial_data_from_db(
// Copy the data from the database into a map
let mut state_group_map: BTreeMap<i64, StateGroupEntry> = BTreeMap::new();
let pb = ProgressBar::new_spinner();
let pb: ProgressBar;
if cfg!(feature = "no-progress-bars") {
pb = ProgressBar::hidden();
} else {
pb = ProgressBar::new_spinner();
}
pb.set_style(
ProgressStyle::default_spinner().template("{spinner} [{elapsed}] {pos} rows retrieved"),
);
@@ -533,7 +538,12 @@ pub fn send_changes_to_db(
println!("Writing changes...");
// setup the progress bar
let pb = ProgressBar::new(old_map.len() as u64);
let pb: ProgressBar;
if cfg!(feature = "no-progress-bars") {
pb = ProgressBar::hidden();
} else {
pb = ProgressBar::new(old_map.len() as u64);
}
pb.set_style(
ProgressStyle::default_bar().template("[{elapsed_precise}] {bar} {pos}/{len} {msg}"),
);