Use jemalloc

rust/Cargo.toml

@@ -63,8 +63,10 @@ futures = "0.3.31"
 tokio = { version = "1.44.2", features = ["rt", "rt-multi-thread"] }
 once_cell = "1.18.0"
 itertools = "0.14.0"
-tikv-jemallocator = { version = "0.6.1", features = ["disable_initial_exec_tls", "unprefixed_malloc_on_supported_platforms"] }
-tikv-jemallocator-global = "0.6.0"
+tikv-jemallocator = { version = "0.6.1", features = [
+    "disable_initial_exec_tls",
+    "unprefixed_malloc_on_supported_platforms",
+] }
 
 [build-dependencies]
 blake2 = "0.10.4"

rust/src/lib.rs

@@ -3,6 +3,7 @@ use std::convert::Infallible;
 use lazy_static::lazy_static;
 use pyo3::prelude::*;
 use pyo3_log::ResetHandle;
+use tikv_jemallocator::Jemalloc;
 
 pub mod acl;
 pub mod canonical_json;
@@ -19,6 +20,9 @@ pub mod rendezvous;
 pub mod room_versions;
 pub mod segmenter;
 
+#[global_allocator]
+static GLOBAL: Jemalloc = Jemalloc;
+
 lazy_static! {
     static ref LOGGING_HANDLE: ResetHandle = pyo3_log::init();
 }