resolve unused imports if feature flags are disabled

ChainSafe · Dec 18, 2024 · 25e2376 · 25e2376
1 parent 40cce8c
commit 25e2376
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/cli_shared/mod.rs b/src/cli_shared/mod.rs
@@ -11,10 +11,14 @@ use crate::networks::NetworkChain;
 use crate::utils::io::read_toml;
 use std::path::PathBuf;
 
-#[cfg(feature = "mimalloc")]
-pub use mimalloc;
-#[cfg(feature = "jemalloc")]
-pub use tikv_jemallocator;
+cfg_if::cfg_if! {
+    if #[cfg(feature = "rustalloc")] {
+    } else if #[cfg(feature = "mimalloc")] {
+        pub use mimalloc;
+    } else if #[cfg(feature = "jemalloc")] {
+        pub use tikv_jemallocator;
+    }
+}
 
 /// Gets chain data directory
 pub fn chain_path(config: &Config) -> PathBuf {