risingwave_common_heap_profiling/
jeprof.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::path::Path;
16use std::process::Command;
17use std::{env, fs};
18
19/// Error type for running `jeprof`.
20#[derive(thiserror::Error, Debug, thiserror_ext::ContextInto)]
21pub enum JeprofError {
22    #[error(transparent)]
23    IoError(#[from] std::io::Error),
24
25    #[error("jeprof exit with an error (stdout: {stdout}, stderr: {stderr}): {inner}")]
26    ExitError {
27        #[source]
28        inner: std::process::ExitStatusError,
29        stdout: String,
30        stderr: String,
31    },
32}
33
34/// Run `jeprof --collapsed` on the given profile.
35pub async fn run(profile_path: String, collapsed_path: String) -> Result<(), JeprofError> {
36    let executable_path = env::current_exe()?;
37
38    let prof_cmd = move || {
39        Command::new("jeprof")
40            .arg("--collapsed")
41            .arg(executable_path)
42            .arg(Path::new(&profile_path))
43            .output()
44    };
45
46    let output = tokio::task::spawn_blocking(prof_cmd).await.unwrap()?;
47
48    output.status.exit_ok().into_exit_error(
49        String::from_utf8_lossy(&output.stdout),
50        String::from_utf8_lossy(&output.stderr),
51    )?;
52
53    fs::write(Path::new(&collapsed_path), &output.stdout)?;
54
55    Ok(())
56}
risingwave_common_heap_profiling/jeprof.rs

risingwave_common_heap_profiling/
jeprof.rs
