fix(perf): switch global allocator to jemalloc

rust-lang · Oct 22, 2024 · adab712 · adab712
1 parent 42f4143
commit adab712
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -99,6 +99,7 @@ snapbox = { version = "0.6.18", features = ["diff", "dir", "term-svg", "regex",
 tar = { version = "0.4.42", default-features = false }
 tempfile = "3.10.1"
 thiserror = "1.0.63"
+tikv-jemallocator = "0.6"
 time = { version = "0.3.36", features = ["parsing", "formatting", "serde"] }
 toml = "0.8.19"
 toml_edit = { version = "0.22.20", features = ["serde"] }
@@ -243,6 +244,9 @@ features = [
   "Win32_System_Threading",
 ]
 
+[target.'cfg(not(target_env = "msvc"))'.dependencies]
+tikv-jemallocator.workspace = true
+
 [dev-dependencies]
 annotate-snippets = { workspace = true, features = ["testing-colors"] }
 cargo-test-support.workspace = true

diff --git a/src/bin/cargo/main.rs b/src/bin/cargo/main.rs
@@ -1,5 +1,9 @@
 #![allow(clippy::self_named_module_files)] // false positive in `commands/build.rs`
 
+#[cfg(not(target_env = "msvc"))]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 use cargo::core::features;
 use cargo::core::shell::Shell;
 use cargo::util::network::http::http_handle;