cargo-prof 0.1.3

Tools to make profiling Rust programs easier
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

use clap::Parser;
use color_eyre::Result;
use prof::{check_commands, heap, leak, Commands, Prof};
use tracing_subscriber::{prelude::*, EnvFilter, Registry};

use tracing_subscriber::filter::LevelFilter;

/// Tools to profile your Rust code
#[derive(Parser)]
#[clap(name = "cargo")]
#[clap(bin_name = "cargo")]
enum Cargo {
    Prof(Prof),
}

fn main() -> Result<()> {
    color_eyre::config::HookBuilder::default()
        .display_location_section(false)
        .display_env_section(false)
        .install()?;

    let env_filter = EnvFilter::builder()
        .with_default_directive(LevelFilter::INFO.into())
        .from_env_lossy();

    let subscriber = Registry::default()
        .with(env_filter)
        .with(tracing_subscriber::fmt::layer());

    tracing::subscriber::set_global_default(subscriber)?;
    let Cargo::Prof(prof) = Cargo::parse();

    match &prof.command {
        Commands::Heap(x) => heap(&prof, x, Some(cargo_build)),
        Commands::Leak(x) => leak(&prof, x, Some(cargo_build)),
    }
}

pub fn cargo_build(bin: &Option<String>) -> Result<Option<String>> {
    check_commands(&["cargo"])?;
    cargo_prof::cargo_build(bin)?;
    Ok(Some(cargo_prof::get_bin()?))
}