mimalloc-pprof 0.10.0

mimalloc global allocator with pprof-compatible sampled heap profiling
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

#![allow(dead_code)]

use mimalloc_pprof::{prof, MiMalloc};
use regex::Regex;

#[global_allocator]
static ALLOCATOR: MiMalloc = MiMalloc;

pub fn start(rate: usize, seed: u64) {
    if prof::is_enabled() {
        prof::stop();
    }
    assert!(prof::start_seeded(rate, seed));
}

pub fn dump() -> String {
    String::from_utf8(prof::dump_to_vec()).expect("heap profile is UTF-8")
}

pub fn header(text: &str) -> (u64, u64, u64, u64, u64) {
    let re = Regex::new(
        r"^heap profile: +([0-9]+): ([0-9]+) \[ *([0-9]+): ([0-9]+)\] @ heap_v2/([0-9]+)$",
    )
    .unwrap();
    let caps = re
        .captures(text.lines().next().unwrap())
        .expect("heap_v2 header");
    (
        caps[1].parse().unwrap(),
        caps[2].parse().unwrap(),
        caps[3].parse().unwrap(),
        caps[4].parse().unwrap(),
        caps[5].parse().unwrap(),
    )
}

pub fn sample_lines(text: &str) -> Vec<&str> {
    let re = Regex::new(r"^ +[0-9]+: [0-9]+ \[ *[0-9]+: [0-9]+\] @( 0x[0-9a-fA-F]+)+$").unwrap();
    text.lines().filter(|line| re.is_match(line)).collect()
}

pub fn stop() {
    prof::stop();
}