support mimalloc

Author: zh-jq

Cargo.lock

@@ -44,6 +44,7 @@ members = [
     "lib/vey-journal",
     "lib/vey-json",
     "lib/vey-macros",
+    "lib/vey-mimalloc",
     "lib/vey-msgpack",
     "lib/vey-openssl",
     "lib/vey-redis-client",
@@ -227,6 +228,7 @@ vey-jemalloc = { version = "0.1", path = "lib/vey-jemalloc" }
 vey-journal = { version = "0.4", path = "lib/vey-journal" }
 vey-json = { version = "0.5", path = "lib/vey-json" }
 vey-macros = { version = "0.1", path = "lib/vey-macros" }
+vey-mimalloc = { version = "0.1", path = "lib/vey-mimalloc" }
 vey-msgpack = { version = "0.4", path = "lib/vey-msgpack" }
 vey-openssl = { version = "0.4", path = "lib/vey-openssl" }
 vey-redis-client = { version = "0.3", path = "lib/vey-redis-client" }

Cargo.toml

@@ -0,0 +1,9 @@
+[package]
+name = "vey-mimalloc"
+version = "0.1.0"
+license.workspace = true
+edition.workspace = true
+rust-version.workspace = true
+
+[dependencies]
+mimalloc-sys = { package = "vey-mimalloc-sys", version = "0.1" }

lib/vey-mimalloc/Cargo.toml

@@ -0,0 +1,29 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * SPDX-FileCopyrightText: 2026 VEY-OSS Developers.
+ */
+
+use core::alloc::{GlobalAlloc, Layout};
+
+mod version;
+pub use version::lib_version;
+
+pub struct Mimalloc;
+
+unsafe impl GlobalAlloc for Mimalloc {
+    unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
+        unsafe { mimalloc_sys::mi_aligned_alloc(layout.align(), layout.size()) as _ }
+    }
+
+    unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
+        unsafe { mimalloc_sys::mi_free_size_aligned(ptr as _, layout.size(), layout.align()) as _ }
+    }
+
+    unsafe fn alloc_zeroed(&self, layout: Layout) -> *mut u8 {
+        unsafe { mimalloc_sys::mi_zalloc_aligned(layout.size(), layout.align()) as _ }
+    }
+
+    unsafe fn realloc(&self, ptr: *mut u8, layout: Layout, new_size: usize) -> *mut u8 {
+        unsafe { mimalloc_sys::mi_realloc_aligned(ptr as _, new_size, layout.align()) as _ }
+    }
+}

lib/vey-mimalloc/src/lib.rs

@@ -0,0 +1,8 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * SPDX-FileCopyrightText: 2026 VEY-OSS Developers.
+ */
+
+pub fn lib_version() -> i32 {
+    unsafe { mimalloc_sys::mi_version() }
+}

lib/vey-mimalloc/src/version.rs

@@ -56,13 +56,15 @@ vey-openssl.workspace = true
 vey-hickory-client.workspace = true
 vey-rustls-provider.workspace = true
 vey-jemalloc = { workspace = true, optional = true }
+vey-mimalloc = { workspace = true, optional = true }
 
 [build-dependencies]
 vey-build-env.workspace = true
 
 [features]
 default = ["quic", "rustls-ring"]
 jemalloc = ["dep:vey-jemalloc"]
+mimalloc = ["dep:vey-mimalloc"]
 quic = ["vey-types/quic", "vey-socks/quic", "vey-io-ext/quic", "vey-hickory-client/quic", "dep:quinn", "dep:h3", "dep:h3-quinn"]
 rustls-ring = ["vey-rustls-provider/rustls-ring", "vey-types/rustls-ring", "quinn?/rustls-ring"]
 rustls-aws-lc = ["vey-rustls-provider/rustls-aws-lc", "vey-types/rustls-aws-lc", "quinn?/rustls-aws-lc-rs"]

vey-bench/Cargo.toml

@@ -26,6 +26,8 @@ pub fn print_version() {
     if let Some(version) = vey_jemalloc::lib_version() {
         print!(" jemalloc({})", version.to_string_lossy());
     }
+    #[cfg(feature = "mimalloc")]
+    println!(" mimalloc({})", vey_mimalloc::lib_version());
     if let Some(quic) = QUIC_FEATURE {
         print!(" {quic}");
     }

vey-bench/src/build.rs

@@ -12,15 +12,16 @@ use anyhow::{Context, anyhow};
 use clap::{Arg, ArgMatches, Command, value_parser};
 use clap_complete::Shell;
 
-#[cfg(feature = "jemalloc")]
-use vey_jemalloc::Jemalloc;
-
 const COMMAND_VERSION: &str = "version";
 const COMMAND_COMPLETION: &str = "completion";
 
 #[cfg(feature = "jemalloc")]
 #[global_allocator]
-static GLOBAL: Jemalloc = Jemalloc;
+static GLOBAL: vey_jemalloc::Jemalloc = vey_jemalloc::Jemalloc;
+
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static GLOBAL: vey_mimalloc::Mimalloc = vey_mimalloc::Mimalloc;
 
 fn build_cli_args() -> Command {
     vey_bench::add_global_args(Command::new(vey_bench::build::PKG_NAME))

vey-bench/src/main.rs

@@ -27,13 +27,15 @@ vey-tls-cert.workspace = true
 vey-cert-agent.workspace = true
 vey-socket.workspace = true
 vey-jemalloc = { workspace = true, optional = true }
+vey-mimalloc = { workspace = true, optional = true }
 
 [build-dependencies]
 vey-build-env.workspace = true
 
 [features]
 default = []
 jemalloc = ["dep:vey-jemalloc", "vey-daemon/jemalloc"]
+mimalloc = ["dep:vey-mimalloc"]
 vendored-openssl = ["openssl/vendored"]
 vendored-tongsuo = ["openssl/tongsuo", "vey-cert-agent/tongsuo"]
 vendored-boringssl = ["openssl/boringssl"]

vey-dcgen/Cargo.toml

@@ -27,6 +27,8 @@ pub fn print_version(verbose_level: u8) {
         if let Some(version) = vey_jemalloc::lib_version() {
             print!(" jemalloc({})", version.to_string_lossy());
         }
+        #[cfg(feature = "mimalloc")]
+        println!(" mimalloc({})", vey_mimalloc::lib_version());
         println!();
         if let Some(variant) = OPENSSL_VARIANT {
             println!("OpenSSL Variant: {variant}");