perf: use mimalloc for bindings_napi (#5201)

rollup · Oct 15, 2023 · f3eaa28 · f3eaa28
1 parent cb144b2
commit f3eaa28
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 0 deletions.
diff --git a/rust/Cargo.lock b/rust/Cargo.lock
diff --git a/rust/bindings_napi/Cargo.toml b/rust/bindings_napi/Cargo.toml
@@ -15,5 +15,11 @@ napi-derive = "2.13.0"
 parse_ast = { path = "../parse_ast" }
 xxhash = { path = "../xxhash" }
 
+[target.'cfg(not(target_os = "linux"))'.dependencies]
+mimalloc-rust = { version = "0.2" }
+
+[target.'cfg(all(target_os = "linux", not(all(target_env = "musl", target_arch = "aarch64"))))'.dependencies]
+mimalloc-rust = { version = "0.2", features = ["local-dynamic-tls"] }
+
 [build-dependencies]
 napi-build = "2.0.1"
diff --git a/rust/bindings_napi/src/lib.rs b/rust/bindings_napi/src/lib.rs
@@ -2,6 +2,13 @@ use napi::bindgen_prelude::*;
 use napi_derive::napi;
 use parse_ast::parse_ast;
 
+#[cfg(all(
+  not(all(target_os = "linux", target_env = "musl", target_arch = "aarch64")),
+  not(debug_assertions)
+))]
+#[global_allocator]
+static ALLOC: mimalloc_rust::GlobalMiMalloc = mimalloc_rust::GlobalMiMalloc;
+
 #[napi]
 pub fn parse(code: String, allow_return_outside_function: bool) -> Buffer {
   parse_ast(code, allow_return_outside_function).into()