Use Jemalloc when possible

Author: erebe

.github/workflows/release.yaml

@@ -22,31 +22,31 @@ jobs:
           # Linux
           - name: Linux x86_64
             target: x86_64-unknown-linux-musl
-            build-args: "--release"
+            build-args: "--release --features=jemalloc"
 
           - name: Linux x86
             target: i686-unknown-linux-musl
             build-args: "--release"
 
           - name: Linux aarch64
             target: aarch64-unknown-linux-musl
-            build-args: "--release"
+            build-args: "--release --features=jemalloc"
 
           - name: Linux armv7hf
             target: armv7-unknown-linux-musleabihf
             build-args: "--release"
 
           - name: Freebsd x86_64
             target: x86_64-unknown-freebsd
-            build-args: "--release"
+            build-args: "--release --features=jemalloc"
 
           - name: Freebsd x86
             target: i686-unknown-freebsd
             build-args: "--release"
 
           - name: Android aarch64
             target: aarch64-linux-android
-            build-args: "--release"
+            build-args: "--release --features=jemalloc"
 
           - name: Android armv7
             target: armv7-linux-androideabi
@@ -62,12 +62,12 @@ jobs:
           - name: MacOS x86_64
             os: macos-latest
             target: x86_64-apple-darwin
-            build-args: "--release"
+            build-args: "--release --features=jemalloc"
 
           - name: MacOS aarch64
             os: macos-latest
             target: aarch64-apple-darwin
-            build-args: "--release"
+            build-args: "--release --features=jemalloc"
 
           # - name: iOS x86_64
           #   target: x86_64-apple-ios

Cargo.lock

@@ -32,7 +32,7 @@ ARG BIN_TARGET=--bins
 ARG PROFILE=release
 
 #ENV RUSTFLAGS="-C link-arg=-Wl,--compress-debug-sections=zlib -C force-frame-pointers=yes"
-RUN cargo build --package=wstunnel-cli --profile=${PROFILE} ${BIN_TARGET}
+RUN cargo build --package=wstunnel-cli --features=jemalloc --profile=${PROFILE} ${BIN_TARGET}
 
 
 ############################################################

Dockerfile

@@ -17,7 +17,7 @@ make_release $VERSION $FORCE="":
 docker_release $TAG:
   #docker login -u erebe ghcr.io
   #~/.depot/bin/depot build --project v4z5w7md33 --platform linux/arm/v7,linux/arm64,linux/amd64 -t ghcr.io/erebe/wstunnel:$TAG -t ghcr.io/erebe/wstunnel:latest --push .
-  docker buildx create  --name builder  --driver=kubernetes --platform=linux/arm64   '--driver-opt="nodeselector=kubernetes.io/arch=arm64","tolerations=key=kubernetes.io/hostname,value=server"' --node=build-arm64
+  docker buildx create  --name builder  --driver=kubernetes --platform=linux/arm64   '--driver-opt="nodeselector=kubernetes.io/arch=arm64","tolerations=key=kubernetes.io/hostname,value=server","requests.cpu=16","requests.memory=16G"' --node=build-arm64
   docker buildx create --append --name builder  --driver=kubernetes --platform=linux/amd64   '--driver-opt="nodeselector=kubernetes.io/arch=amd64","tolerations=key=kubernetes.io/hostname,value=toybox","requests.cpu=16","requests.memory=16G"' --node=build-amd64
   docker buildx use builder
   docker buildx build \

justfile

@@ -12,6 +12,12 @@ tracing = { version = "0.1.41", features = ["log"] }
 tracing-subscriber = { version = "0.3.19", features = ["env-filter", "fmt", "local-time"] }
 wstunnel = { path = ".." , features = ["clap"] }
 
+tikv-jemallocator = {  version = "0.6", optional = true }
+
+[features]
+default = []
+jemalloc = ["dep:tikv-jemallocator"]
+
 [[bin]]
 name = "wstunnel"
 path = "src/main.rs"

wstunnel-cli/Cargo.toml

@@ -8,6 +8,13 @@ use wstunnel::config::{Client, Server};
 use wstunnel::LocalProtocol;
 use wstunnel::{run_client, run_server};
 
+#[cfg(feature = "jemalloc")]
+use tikv_jemallocator::Jemalloc;
+
+#[cfg(feature = "jemalloc")]
+#[global_allocator]
+static GLOBAL: Jemalloc = Jemalloc;
+
 /// Use Websocket or HTTP2 protocol to tunnel {TCP,UDP} traffic
 /// wsTunnelClient <---> wsTunnelServer <---> RemoteHost
 #[derive(clap::Parser, Debug)]