perf: use tikv-jemallocator

akasamq · Dec 2, 2023 · 7883ac1 · 7883ac1 · TheWaWaR · Dec 2, 2023
1 parent 2d7bd33
commit 7883ac1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/akasa/Cargo.toml b/akasa/Cargo.toml
@@ -16,3 +16,6 @@ log = "0.4.17"
 dashmap = "5.4.0"
 rpassword = "7.2.0"
 rand = { version = "0.8.5", features = ["getrandom"] }
+
+[target.'cfg(not(target_env = "msvc"))'.dependencies]
+tikv-jemallocator = "0.5.4"
diff --git a/akasa/src/main.rs b/akasa/src/main.rs
@@ -17,6 +17,13 @@ use rand::{rngs::OsRng, RngCore};
 
 use default_hook::DefaultHook;
 
+#[cfg(not(target_env = "msvc"))]
+use tikv_jemallocator::Jemalloc;
+
+#[cfg(not(target_env = "msvc"))]
+#[global_allocator]
+static GLOBAL: Jemalloc = Jemalloc;
+
 #[derive(Parser, Debug)]
 #[clap(author, version, about, long_about = None)]
 /// Akasa MQTT server