Skip to content

Commit f5f0cce

Browse files
committed
Refresh the progress bars 100 times less often
1 parent a380c4f commit f5f0cce

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

crates/syn2mas/src/migration.rs

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ async fn migrate_users(
246246
let mut password_buffer = MasWriteBuffer::new(mas, MasWriter::write_passwords);
247247
let mut users_stream = pin!(synapse
248248
.read_users()
249-
.with_progress_bar(user_count_hint as u64, 100));
249+
.with_progress_bar(user_count_hint as u64, 10_000));
250250
// Oversize the capacity, because the count is only an estimate and
251251
// we would like to avoid a reallocation
252252
let mut user_localparts_to_uuid = HashMap::with_capacity(user_count_hint * 9 / 8);
@@ -308,7 +308,9 @@ async fn migrate_threepids(
308308

309309
let mut email_buffer = MasWriteBuffer::new(mas, MasWriter::write_email_threepids);
310310
let mut unsupported_buffer = MasWriteBuffer::new(mas, MasWriter::write_unsupported_threepids);
311-
let mut users_stream = pin!(synapse.read_threepids().with_progress_bar(count_hint, 100));
311+
let mut users_stream = pin!(synapse
312+
.read_threepids()
313+
.with_progress_bar(count_hint, 10_000));
312314

313315
while let Some(threepid_res) = users_stream.next().await {
314316
let SynapseThreepid {
@@ -406,7 +408,7 @@ async fn migrate_external_ids(
406408
let mut write_buffer = MasWriteBuffer::new(mas, MasWriter::write_upstream_oauth_links);
407409
let mut extids_stream = pin!(synapse
408410
.read_user_external_ids()
409-
.with_progress_bar(count_hint, 100));
411+
.with_progress_bar(count_hint, 10_000));
410412

411413
while let Some(extid_res) = extids_stream.next().await {
412414
let SynapseExternalId {
@@ -492,7 +494,7 @@ async fn migrate_devices(
492494
) -> Result<(), Error> {
493495
let start = Instant::now();
494496

495-
let mut devices_stream = pin!(synapse.read_devices().with_progress_bar(count_hint, 100));
497+
let mut devices_stream = pin!(synapse.read_devices().with_progress_bar(count_hint, 10_000));
496498
let mut write_buffer = MasWriteBuffer::new(mas, MasWriter::write_compat_sessions);
497499

498500
while let Some(device_res) = devices_stream.next().await {
@@ -579,7 +581,7 @@ async fn migrate_unrefreshable_access_tokens(
579581

580582
let mut token_stream = pin!(synapse
581583
.read_unrefreshable_access_tokens()
582-
.with_progress_bar(count_hint, 100));
584+
.with_progress_bar(count_hint, 10_000));
583585
let mut write_buffer = MasWriteBuffer::new(mas, MasWriter::write_compat_access_tokens);
584586
let mut deviceless_session_write_buffer =
585587
MasWriteBuffer::new(mas, MasWriter::write_compat_sessions);
@@ -700,7 +702,7 @@ async fn migrate_refreshable_token_pairs(
700702

701703
let mut token_stream = pin!(synapse
702704
.read_refreshable_token_pairs()
703-
.with_progress_bar(count_hint, 100));
705+
.with_progress_bar(count_hint, 10_000));
704706
let mut access_token_write_buffer =
705707
MasWriteBuffer::new(mas, MasWriter::write_compat_access_tokens);
706708
let mut refresh_token_write_buffer =

0 commit comments

Comments
 (0)