*: Add mimalloc feature to tikv_alloc

Cargo.toml

@@ -15,6 +15,7 @@ publish = false
 default = ["tikv_alloc/default"]
 tcmalloc = ["tikv_alloc/tcmalloc"]
 jemalloc = ["tikv_alloc/jemalloc"]
+mimalloc = ["tikv_alloc/mimalloc"]
 portable = ["engine/portable"]
 sse = ["engine/sse"]
 mem-profiling = ["tikv_alloc/mem-profiling"]

Makefile

@@ -37,6 +37,8 @@ ENABLE_FEATURES ?=
 # Pick an allocator
 ifeq ($(TCMALLOC),1)
 ENABLE_FEATURES += tcmalloc
+else ifeq ($(MIMALLOC),1)
+ENABLE_FEATURES += mimalloc
 else ifeq ($(SYSTEM_ALLOC),1)
 # no feature needed for system allocator
 else

components/tikv_alloc/Cargo.toml

@@ -12,6 +12,7 @@ jemalloc = ["jemallocator", "jemalloc-sys", "jemalloc-ctl"]
 # Build jemalloc's profiling features. Without this
 # certain profile functions will return nothing.
 mem-profiling = ["jemallocator/profiling"]
+mimalloc = ["mimallocator"]
 
 [dependencies]
 libc = "0.2"
@@ -44,3 +45,7 @@ optional = true
 version = "0.3.0"
 optional = true
 features = ["bundled"]
+
+[dependencies.mimallocator]
+version = "0.1.3"
+optional = true

components/tikv_alloc/src/lib.rs

@@ -96,7 +96,14 @@ mod imp;
 #[cfg(all(unix, not(fuzzing), feature = "tcmalloc"))]
 #[path = "tcmalloc.rs"]
 mod imp;
-#[cfg(not(all(unix, not(fuzzing), any(feature = "jemalloc", feature = "tcmalloc"))))]
+#[cfg(all(unix, not(fuzzing), feature = "mimalloc"))]
+#[path = "mimalloc.rs"]
+mod imp;
+#[cfg(not(all(
+    unix,
+    not(fuzzing),
+    any(feature = "jemalloc", feature = "tcmalloc", feature = "mimalloc")
+)))]
 #[path = "system.rs"]
 mod imp;
 

components/tikv_alloc/src/mimalloc.rs

@@ -0,0 +1,7 @@
+pub use crate::default::*;
+
+pub type Allocator = mimallocator::Mimalloc;
+
+pub const fn allocator() -> Allocator {
+    mimallocator::Mimalloc
+}