feat: turn balance system allocator;

mycrl · Dec 31, 2023 · 7afb693 · 7afb693
1 parent 52da044
commit 7afb693
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/turn-balance/server/Cargo.toml b/turn-balance/server/Cargo.toml
@@ -18,3 +18,12 @@ num_cpus = "1.15"
 
 [build-dependencies]
 prost-build = "0.12"
+
+[target.'cfg(not(target_os = "windows"))'.dependencies]
+jemallocator = "0.5.4"
+
+[target.'cfg(target_os = "windows")'.dependencies]
+mimalloc = "0.1.37"
+
+[features]
+system_allocator = []
diff --git a/turn-balance/server/src/main.rs b/turn-balance/server/src/main.rs
@@ -1,3 +1,11 @@
+#[global_allocator]
+#[cfg(all(target_os = "windows", not(feature = "system_allocator")))]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
+#[global_allocator]
+#[cfg(all(not(target_os = "windows"), not(feature = "system_allocator")))]
+static GLOBAL: jemallocator::Jemalloc = jemallocator::Jemalloc;
+
 mod cluster;
 mod config;