perf: use specific allocator on different platform (#912)

* perf: use specific allocator on different platform

* fix: move allocation declare to lib entry

Cargo.toml

@@ -3,7 +3,9 @@ resolver = "2"
 members  = ["crates/*"]
 
 [workspace.dependencies]
-serde      = { version = "1.0.171", features = ["derive"] }
-serde_json = "1.0.100"
-cached     = "0.46.1"
-swc_core   = { version = "=0.83.19", default-features = false }
+mimalloc-rust     = { version = "=0.2.1" }
+serde             = { version = "1.0.171", features = ["derive"] }
+serde_json        = "1.0.100"
+cached            = "0.46.1"
+swc_core          = { version = "=0.83.19", default-features = false }
+tikv-jemallocator = { version = "=0.5.4", features = ["disable_initial_exec_tls"] }

crates/mako/Cargo.toml

@@ -19,6 +19,12 @@ swc_core   = { workspace = true, features = ["swc_ecma_quote_macros"] }
 miette     = { version = "5.10.0", features = ["fancy"] }
 glob-match = "0.2.1"
 
+[target.'cfg(not(target_os = "linux"))'.dependencies]
+mimalloc-rust = { workspace = true }
+
+[target.'cfg(all(target_os = "linux", target_env = "gnu", any(target_arch = "x86_64", target_arch = "aarch64")))'.dependencies]
+tikv-jemallocator = { workspace = true }
+
 [dev-dependencies]
 insta                       = { version = "1.30.0", features = ["yaml"] }
 maplit                      = "1.0.2"

crates/mako/src/main.rs

@@ -59,6 +59,18 @@ mod update;
 mod util;
 mod watch;
 
+#[cfg(not(target_os = "linux"))]
+#[global_allocator]
+static GLOBAL: mimalloc_rust::GlobalMiMalloc = mimalloc_rust::GlobalMiMalloc;
+
+#[cfg(all(
+    target_os = "linux",
+    target_env = "gnu",
+    any(target_arch = "x86_64", target_arch = "aarch64")
+))]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 #[tokio::main]
 async fn main() -> Result<()> {
     // logger

crates/node/Cargo.toml

@@ -15,6 +15,12 @@ mako_core   = { path = "../core" }
 serde_json  = { workspace = true }
 cached      = { workspace = true }
 
+[target.'cfg(not(target_os = "linux"))'.dependencies]
+mimalloc-rust = { workspace = true }
+
+[target.'cfg(all(target_os = "linux", target_env = "gnu", any(target_arch = "x86_64", target_arch = "aarch64")))'.dependencies]
+tikv-jemallocator = { workspace = true }
+
 [build-dependencies]
 napi-build = "2.0.1"
 

crates/node/package.json

@@ -42,5 +42,6 @@
     "@okamjs/okam-darwin-arm64": "0.4.3",
     "@okamjs/okam-darwin-x64": "0.4.3",
     "@okamjs/okam-linux-x64-gnu": "0.4.3"
-  }
+  },
+  "repository": "[email protected]:umijs/mako.git"
 }

crates/node/src/lib.rs

@@ -20,6 +20,18 @@ mod tsfn;
 
 pub(crate) mod threadsafe_function;
 
+#[cfg(not(target_os = "linux"))]
+#[global_allocator]
+static GLOBAL: mimalloc_rust::GlobalMiMalloc = mimalloc_rust::GlobalMiMalloc;
+
+#[cfg(all(
+    target_os = "linux",
+    target_env = "gnu",
+    any(target_arch = "x86_64", target_arch = "aarch64")
+))]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 static LOG_INIT: Once = Once::new();
 
 #[napi(object)]