Use rustc' FxHasher instead of the std one

Author: sandhose

Cargo.lock

@@ -25,6 +25,7 @@ compact_str.workspace = true
 tracing.workspace = true
 tracing-indicatif.workspace = true
 futures-util = "0.3.31"
+rustc-hash = "2.1.1"
 
 rand.workspace = true
 uuid = "1.14.0"

crates/syn2mas/Cargo.toml

@@ -9,6 +9,9 @@ mod synapse_reader;
 mod migration;
 mod progress_stream;
 
+type RandomState = rustc_hash::FxBuildHasher;
+type HashMap<K, V> = rustc_hash::FxHashMap<K, V>;
+
 pub use self::{
     mas_writer::{MasWriter, checks::mas_pre_migration_checks, locking::LockedMasDatabase},
     migration::migrate,

crates/syn2mas/src/lib.rs

@@ -11,7 +11,7 @@
 //! This module does not implement any of the safety checks that should be run
 //! *before* the migration.
 
-use std::{collections::HashMap, pin::pin, time::Instant};
+use std::{pin::pin, time::Instant};
 
 use chrono::{DateTime, Utc};
 use compact_str::CompactString;
@@ -26,7 +26,7 @@ use ulid::Ulid;
 use uuid::Uuid;
 
 use crate::{
-    SynapseReader,
+    HashMap, RandomState, SynapseReader,
     mas_writer::{
         self, MasNewCompatAccessToken, MasNewCompatRefreshToken, MasNewCompatSession,
         MasNewEmailThreepid, MasNewUnsupportedThreepid, MasNewUpstreamOauthLink, MasNewUser,
@@ -116,7 +116,7 @@ struct MigrationState {
 
     /// A mapping of Synapse external ID providers to MAS upstream OAuth 2.0
     /// provider ID
-    provider_id_mapping: HashMap<String, Uuid>,
+    provider_id_mapping: std::collections::HashMap<String, Uuid>,
 }
 
 /// Performs a migration from Synapse's database to MAS' database.
@@ -139,7 +139,7 @@ pub async fn migrate(
     server_name: String,
     clock: &dyn Clock,
     rng: &mut impl RngCore,
-    provider_id_mapping: HashMap<String, Uuid>,
+    provider_id_mapping: std::collections::HashMap<String, Uuid>,
 ) -> Result<(), Error> {
     let span = Span::current();
     span.pb_set_message("counting work");
@@ -150,8 +150,11 @@ pub async fn migrate(
         server_name,
         // We oversize the hashmaps, as the estimates are innaccurate, and we would like to avoid
         // reallocations.
-        users: HashMap::with_capacity(counts.users * 9 / 8),
-        devices_to_compat_sessions: HashMap::with_capacity(counts.devices * 9 / 8),
+        users: HashMap::with_capacity_and_hasher(counts.users * 9 / 8, RandomState::default()),
+        devices_to_compat_sessions: HashMap::with_capacity_and_hasher(
+            counts.devices * 9 / 8,
+            RandomState::default(),
+        ),
         provider_id_mapping,
     };