perf: enable jemallocator for musl targets (#715)

* perf: enable jemallocator for musl targets

* chore: add a feature for exposing internal api

* build: fix conflicts between jemalloc & dhat

Author: Itsusinn

.github/workflows/ci.yml

@@ -54,6 +54,10 @@ jobs:
           - os: ubuntu-latest
             target: x86_64-unknown-linux-musl
             tool: cross
+            extra-args: -F "shadowsocks, tuic, onion, jemallocator"
+          - os: ubuntu-latest
+            target: i686-unknown-linux-musl
+            tool: cross
             extra-args: -F "shadowsocks, tuic, onion"
           # Linux x86 gnu static-crt
           - os: ubuntu-latest
@@ -85,7 +89,7 @@ jobs:
           - os: ubuntu-latest
             target: aarch64-unknown-linux-musl
             tool: cross
-            extra-args: -F "shadowsocks, tuic, onion"
+            extra-args: -F "shadowsocks, tuic, onion, jemallocator"
           - os: ubuntu-latest
             target: armv7-unknown-linux-musleabihf
             tool: cross
@@ -186,11 +190,11 @@ jobs:
           - os: ubuntu-latest
             target: x86_64-unknown-freebsd
             tool: cross
-            extra-args: --verbose -F "shadowsocks, tuic"
+            extra-args: -F "shadowsocks, tuic"
           - os: ubuntu-latest
             target: i686-unknown-freebsd
             tool: cross
-            extra-args: --verbose -F "shadowsocks, tuic"
+            extra-args: -F "shadowsocks, tuic"
           # Android
           - os: ubuntu-latest
             target: aarch64-linux-android

Cargo.lock

@@ -6,18 +6,20 @@ edition = { workspace = true }
 
 [features]
 default = []
+
 shadowsocks = ["clash_lib/shadowsocks"]
 tuic = ["clash_lib/tuic"]
-bench = ["clash_lib/bench"]
 onion = ["clash_lib/onion"]
 
+bench = ["clash_lib/bench"]
 dhat-heap = ["dep:dhat"]
-
 tokio-console = ["clash_lib/tokio-console"]
+jemallocator = ["dep:tikv-jemallocator"]
 
 [dependencies]
-clap = { version = "4.5.28", features = ["derive"] }
+clap = { version = "4", features = ["derive"] }
 
 clash_lib = { path = "../clash_lib", default-features = false }
 
-dhat = { version = "0.3.3", optional = true }
+dhat = { version = "0.3", optional = true }
+tikv-jemallocator = { version = "0.6", optional = true }

clash/Cargo.toml

@@ -2,6 +2,13 @@
 #[global_allocator]
 static ALLOC: dhat::Alloc = dhat::Alloc;
 
+#[cfg(all(feature = "jemallocator", not(feature = "dhat-heap")))]
+use tikv_jemallocator::Jemalloc;
+
+#[cfg(all(feature = "jemallocator", not(feature = "dhat-heap")))]
+#[global_allocator]
+static GLOBAL: Jemalloc = Jemalloc;
+
 extern crate clash_lib as clash;
 
 use clap::Parser;

clash/src/main.rs

@@ -5,15 +5,17 @@ version = { workspace = true }
 edition = { workspace = true }
 
 [features]
-default = ["tokio-console", "zero_copy"]
+default = ["zero_copy"]
+
+internal = []
 # Protos
 shadowsocks = ["dep:shadowsocks"]
 tuic = ["dep:tuic", "dep:tuic-quinn", "dep:register-count"]
 ssh = ["dep:russh", "dep:dirs", "dep:totp-rs"]
-bench = ["dep:criterion"]
 onion = ["dep:arti-client", "dep:tor-rtcompat", "arti-client/onion-service-client"]
-zero_copy = []
 
+zero_copy = []
+bench = ["dep:criterion"]
 tokio-console = ["tokio/tracing"]
 
 [dependencies]

clash_lib/Cargo.toml

@@ -36,6 +36,9 @@ use tracing::{debug, error, info};
 
 mod app;
 mod common;
+#[cfg(feature = "internal")]
+pub mod config;
+#[cfg(not(feature = "internal"))]
 mod config;
 mod proxy;
 mod session;