test add jemalloc

paulhauner · Feb 10, 2021 · db6fda6 · db6fda6
1 parent e4b6213
commit db6fda6
Show file tree

Hide file tree

Showing 5 changed files with 72 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/beacon_node/http_metrics/Cargo.toml b/beacon_node/http_metrics/Cargo.toml
@@ -20,6 +20,7 @@ lazy_static = "1.4.0"
 eth2 = { path = "../../common/eth2" }
 lighthouse_version = { path = "../../common/lighthouse_version" }
 warp_utils = { path = "../../common/warp_utils" }
+jemalloc-ctl = "0.3.3"
 
 [dev-dependencies]
 tokio = { version = "0.3.2", features = ["sync"] }

diff --git a/beacon_node/http_metrics/src/metrics.rs b/beacon_node/http_metrics/src/metrics.rs
@@ -1,6 +1,7 @@
 use crate::Context;
 use beacon_chain::BeaconChainTypes;
 use lighthouse_metrics::{Encoder, TextEncoder};
+use jemalloc_ctl::{stats, epoch};
 
 pub use lighthouse_metrics::*;
 
@@ -47,3 +48,9 @@ pub fn gather_prometheus_metrics<T: BeaconChainTypes>(
 
     String::from_utf8(buffer).map_err(|e| format!("Failed to encode prometheus info: {:?}", e))
 }
+
+pub fn scrape_jemalloc_metrics() {
+    let e = epoch::mib().unwrap();
+    let allocated = stats::allocated::mib().unwrap();
+    let resident = stats::resident::mib().unwrap();
+}
diff --git a/lighthouse/environment/Cargo.toml b/lighthouse/environment/Cargo.toml
@@ -20,3 +20,4 @@ futures = "0.3.7"
 parking_lot = "0.11.0"
 slog-json = "2.3.0"
 exit-future = "0.2.0"
+jemallocator = "0.3.2"
diff --git a/lighthouse/environment/src/lib.rs b/lighthouse/environment/src/lib.rs
@@ -27,6 +27,10 @@ use task_executor::TaskExecutor;
 use tokio::runtime::{Builder as RuntimeBuilder, Runtime};
 use types::{EthSpec, MainnetEthSpec, MinimalEthSpec, V012LegacyEthSpec};
 
+/// Global allocator
+#[global_allocator]
+pub static ALLOC: jemallocator::Jemalloc = jemallocator::Jemalloc;
+
 pub const ETH2_CONFIG_FILENAME: &str = "eth2-spec.toml";
 const LOG_CHANNEL_SIZE: usize = 2048;
 /// The maximum time in seconds the client will wait for all internal tasks to shutdown.