Add tikv-jemallocator

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "rmqtt-bench"
-version = "0.1.1"
+version = "0.1.2"
 authors = ["rmqtt <[email protected]>"]
 edition = "2021"
 license = "MIT OR Apache-2.0"
@@ -12,6 +12,9 @@ description = "rmqtt-bench - MQTT benchmark tool"
 name = "rmqtt-bench"
 path = "src/main.rs"
 
+[target.'cfg(target_os = "linux")'.dependencies]
+tikv-jemallocator = "0.5"
+
 [dependencies]
 ntex-mqtt = "0.9"
 ntex = { version = "0.5.16", features = ["tokio", "rustls"] }

src/main.rs

@@ -25,6 +25,10 @@ mod script;
 mod stats;
 mod v3;
 
+#[cfg(target_os = "linux")]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 #[ntex::main]
 async fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
     std::env::set_var("RUST_LOG", "rmqtt_bench=info");