Merge pull request #7 from satler-git/custom-alloc

satler-git · Nov 25, 2024 · c166bd3 · c166bd3
2 parents 70b7b40 + f3b8349
commit c166bd3
Show file tree

Hide file tree

Showing 3 changed files with 79 additions and 22 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/etymora/Cargo.toml b/crates/etymora/Cargo.toml
@@ -29,5 +29,13 @@ thiserror.workspace = true
 tracing.workspace  = true
 tracing-subscriber = "0.3.18"
 
+jemallocator = { version = "0.5.4", optional = true }
+mimalloc     = { version = "0.1.43", optional = true }
+
 [build-dependencies]
 shadow-rs = "0.36.0"
+
+[features]
+default  = []
+mimalloc = ["dep:mimalloc"]
+jemalloc = ["dep:jemallocator"]
diff --git a/crates/etymora/src/main.rs b/crates/etymora/src/main.rs
@@ -9,6 +9,14 @@ use tracing::{info, Level};
 use std::error::Error;
 use std::time::Duration;
 
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static ALLOC: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
+#[cfg(all(feature = "jemalloc", not(target_env = "msvc")))]
+#[global_allocator]
+static ALLOC: jemallocator::Jemalloc = jemallocator::Jemalloc;
+
 shadow!(build);
 
 #[derive(Parser, Debug)]