pprof_server 0.2.0

A simple pprof server in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use axum::{extract::Query, http::StatusCode, response::{IntoResponse, Response}, routing::get, Router};
use crate::profiling::{generate_profile, ProfileParams};

async fn pprof_profile_axum(Query(params): Query<ProfileParams>) -> Result<Response, StatusCode> {
    let duration = params.seconds.unwrap_or(30);
    match generate_profile(duration).await {
        Ok(body) => Ok((
            StatusCode::OK,
            [("Content-Type", "application/octet-stream")],
            [("Content-Disposition", "attachment; filename=\"profile.pb.gz\"")],
            body,
        ).into_response()),
        Err(_) => Err(StatusCode::INTERNAL_SERVER_ERROR),
    }
}

pub fn app() -> Router {
    Router::new().route("/debug/pprof/profile", get(pprof_profile_axum))
}