fix(capture-replay): try using jemalloc (#24778)

PostHog · Sep 4, 2024 · 74420aa · 74420aa
1 parent 43555bf
commit 74420aa
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/rust/Cargo.lock b/rust/Cargo.lock
diff --git a/rust/capture/Cargo.toml b/rust/capture/Cargo.toml
@@ -7,6 +7,10 @@ edition = "2021"
 workspace = true
 
 [dependencies]
+
+[target.'cfg(not(target_env = "msvc"))'.dependencies]
+tikv-jemallocator = "0.6"
+
 anyhow = { workspace = true }
 async-trait = { workspace = true }
 axum = { workspace = true }

diff --git a/rust/capture/src/main.rs b/rust/capture/src/main.rs
@@ -16,6 +16,13 @@ use tracing_subscriber::{EnvFilter, Layer};
 use capture::config::Config;
 use capture::server::serve;
 
+#[cfg(not(target_env = "msvc"))]
+use tikv_jemallocator::Jemalloc;
+
+#[cfg(not(target_env = "msvc"))]
+#[global_allocator]
+static GLOBAL: Jemalloc = Jemalloc;
+
 async fn shutdown() {
     let mut term = signal::unix::signal(signal::unix::SignalKind::terminate())
         .expect("failed to register SIGTERM handler");