Skip to content

feat: add JemallocProfCtl::dump_profile #32

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
andylokandy wants to merge 2 commits into
base: main
Choose a base branch
from
Open

feat: add JemallocProfCtl::dump_profile #32

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
3f8e79b
feat: add JemallocProfCtl::dump_stack_profile
andylokandy Mar 13, 2026
55a1fb5
rename dump_stack_profile to dump_profile
andylokandy Mar 14, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 11 additions & 7 deletions src/lib.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,10 @@ use tempfile::NamedTempFile;
use tikv_jemalloc_ctl::raw;
use tokio::sync::Mutex;

use util::parse_jeheap;
#[cfg(feature = "flamegraph")]
pub use util::FlamegraphOptions;
use util::{parse_jeheap, ProfStartTime};
pub use util::{BuildId, Mapping, ProfStartTime, StackProfile, StackProfileIter, WeightedStack};

/// Activate jemalloc profiling.
pub async fn activate_jemalloc_profiling() {
Expand Down Expand Up @@ -159,12 +160,17 @@ impl JemallocProfCtl {
Ok(f.into_file())
}

/// Dump a profile and return the parsed [`StackProfile`].
pub fn dump_profile(&mut self) -> anyhow::Result<StackProfile> {
let f = self.dump()?;
let dump_reader = BufReader::new(f);
parse_jeheap(dump_reader, MAPPINGS.as_deref())
}

/// Dump a profile in pprof format (gzipped protobuf) and
/// return a buffer with its contents.
pub fn dump_pprof(&mut self) -> anyhow::Result<Vec<u8>> {
let f = self.dump()?;
let dump_reader = BufReader::new(f);
let profile = parse_jeheap(dump_reader, MAPPINGS.as_deref())?;
let profile = self.dump_profile()?;
let pprof = profile.to_pprof(("inuse_space", "bytes"), ("space", "bytes"), None);
Ok(pprof)
}
Expand All @@ -184,9 +190,7 @@ impl JemallocProfCtl {
&mut self,
opts: &mut FlamegraphOptions,
) -> anyhow::Result<Vec<u8>> {
let f = self.dump()?;
let dump_reader = BufReader::new(f);
let profile = parse_jeheap(dump_reader, MAPPINGS.as_deref())?;
let profile = self.dump_profile()?;
profile.to_flamegraph(opts)
}
}