mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-16 11:22:56 +00:00
d297d08c6b
#5244 Pass `JEMALLOC_SYS_WITH_LG_PAGE=16` env to aarch64 cross-compilation to support systems with up to 64-KiB page sizes. This is backwards-compatible for the current (most usual) 4-KiB systems.
70 lines
2.7 KiB
Rust
70 lines
2.7 KiB
Rust
//! Set the allocator to `jemalloc`.
|
|
//!
|
|
//! Due to `jemalloc` requiring configuration at compile time or immediately upon runtime
|
|
//! initialisation it is configured via a Cargo config file in `.cargo/config.toml`.
|
|
//!
|
|
//! The `jemalloc` tuning can be overridden by:
|
|
//!
|
|
//! A) `JEMALLOC_SYS_WITH_MALLOC_CONF` at compile-time.
|
|
//! B) `_RJEM_MALLOC_CONF` at runtime.
|
|
use metrics::{set_gauge, try_create_int_gauge, IntGauge};
|
|
use std::sync::LazyLock;
|
|
use tikv_jemalloc_ctl::{arenas, epoch, stats, Access, AsName, Error};
|
|
|
|
#[global_allocator]
|
|
static ALLOC: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
|
|
|
|
// Metrics for jemalloc.
|
|
pub static NUM_ARENAS: LazyLock<metrics::Result<IntGauge>> =
|
|
LazyLock::new(|| try_create_int_gauge("jemalloc_num_arenas", "The number of arenas in use"));
|
|
pub static BYTES_ALLOCATED: LazyLock<metrics::Result<IntGauge>> = LazyLock::new(|| {
|
|
try_create_int_gauge("jemalloc_bytes_allocated", "Equivalent to stats.allocated")
|
|
});
|
|
pub static BYTES_ACTIVE: LazyLock<metrics::Result<IntGauge>> =
|
|
LazyLock::new(|| try_create_int_gauge("jemalloc_bytes_active", "Equivalent to stats.active"));
|
|
pub static BYTES_MAPPED: LazyLock<metrics::Result<IntGauge>> =
|
|
LazyLock::new(|| try_create_int_gauge("jemalloc_bytes_mapped", "Equivalent to stats.mapped"));
|
|
pub static BYTES_METADATA: LazyLock<metrics::Result<IntGauge>> = LazyLock::new(|| {
|
|
try_create_int_gauge("jemalloc_bytes_metadata", "Equivalent to stats.metadata")
|
|
});
|
|
pub static BYTES_RESIDENT: LazyLock<metrics::Result<IntGauge>> = LazyLock::new(|| {
|
|
try_create_int_gauge("jemalloc_bytes_resident", "Equivalent to stats.resident")
|
|
});
|
|
pub static BYTES_RETAINED: LazyLock<metrics::Result<IntGauge>> = LazyLock::new(|| {
|
|
try_create_int_gauge("jemalloc_bytes_retained", "Equivalent to stats.retained")
|
|
});
|
|
|
|
pub fn scrape_jemalloc_metrics() {
|
|
scrape_jemalloc_metrics_fallible().unwrap()
|
|
}
|
|
|
|
pub fn scrape_jemalloc_metrics_fallible() -> Result<(), Error> {
|
|
// Advance the epoch so that the underlying statistics are updated.
|
|
epoch::advance()?;
|
|
|
|
set_gauge(&NUM_ARENAS, arenas::narenas::read()? as i64);
|
|
set_gauge(&BYTES_ALLOCATED, stats::allocated::read()? as i64);
|
|
set_gauge(&BYTES_ACTIVE, stats::active::read()? as i64);
|
|
set_gauge(&BYTES_MAPPED, stats::mapped::read()? as i64);
|
|
set_gauge(&BYTES_METADATA, stats::metadata::read()? as i64);
|
|
set_gauge(&BYTES_RESIDENT, stats::resident::read()? as i64);
|
|
set_gauge(&BYTES_RETAINED, stats::retained::read()? as i64);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn page_size() -> Result<usize, Error> {
|
|
// Full list of keys: https://jemalloc.net/jemalloc.3.html
|
|
"arenas.page\0".name().read()
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn page_size_ok() {
|
|
assert!(page_size().is_ok());
|
|
}
|
|
}
|