feat: add perf support in walltime

Author: not-matthias

Cargo.lock

@@ -46,6 +46,8 @@ sysinfo = { version = "0.33.1", features = ["serde"] }
 indicatif = "0.17.8"
 console = "0.15.8"
 async-trait = "0.1.82"
+codspeed = { git = "https://github.com/CodspeedHQ/codspeed-rust", branch = "cod-674-collect-profiles-while-benchmarks-are-running", version = "=2.9.1" }
+libc = "0.2.171"
 
 [dev-dependencies]
 temp-env = { version = "0.3.6", features = ["async_closure"] }

Cargo.toml

@@ -2,3 +2,4 @@ pub mod env;
 pub mod get_bench_command;
 pub mod profile_folder;
 pub mod run_command_with_log_pipe;
+pub mod setup;

src/run/runner/helpers/mod.rs

@@ -6,7 +6,10 @@ use std::process::Command;
 use std::process::ExitStatus;
 use std::thread;
 
-pub fn run_command_with_log_pipe(mut cmd: Command) -> Result<ExitStatus> {
+pub fn run_command_with_log_pipe_and_callback<F: FnOnce(u32)>(
+    mut cmd: Command,
+    cb: F,
+) -> Result<ExitStatus> {
     fn log_tee(
         mut reader: impl Read,
         mut writer: impl Write,
@@ -46,5 +49,13 @@ pub fn run_command_with_log_pipe(mut cmd: Command) -> Result<ExitStatus> {
     thread::spawn(move || {
         log_tee(stderr, std::io::stderr(), Some("[stderr]")).unwrap();
     });
+
+    let perf_pid = process.id();
+    cb(perf_pid);
+
     process.wait().context("failed to wait for the process")
 }
+
+pub fn run_command_with_log_pipe(cmd: Command) -> Result<ExitStatus> {
+    run_command_with_log_pipe_and_callback(cmd, |_| {})
+}

src/run/runner/helpers/run_command_with_log_pipe.rs

@@ -0,0 +1,33 @@
+use crate::prelude::*;
+use anyhow::{anyhow, bail, Result};
+use log::{debug, info};
+use std::process::{Command, Stdio};
+
+/// Run a command with sudo if available
+pub fn run_with_sudo(command_args: &[&str]) -> Result<()> {
+    let use_sudo = Command::new("sudo")
+        // `sudo true` will fail if sudo does not exist or the current user does not have sudo privileges
+        .arg("true")
+        .stdout(Stdio::null())
+        .status()
+        .is_ok_and(|status| status.success());
+    let mut command_args: Vec<&str> = command_args.into();
+    if use_sudo {
+        command_args.insert(0, "sudo");
+    }
+
+    debug!("Running command: {}", command_args.join(" "));
+    let output = Command::new(command_args[0])
+        .args(&command_args[1..])
+        .stdout(Stdio::piped())
+        .output()
+        .map_err(|_| anyhow!("Failed to execute command: {}", command_args.join(" ")))?;
+
+    if !output.status.success() {
+        info!("stdout: {}", String::from_utf8_lossy(&output.stdout));
+        error!("stderr: {}", String::from_utf8_lossy(&output.stderr));
+        bail!("Failed to execute command: {}", command_args.join(" "));
+    }
+
+    Ok(())
+}

src/run/runner/helpers/setup.rs

@@ -5,7 +5,7 @@ use crate::prelude::*;
 use super::RunnerMode;
 
 mod executor;
-mod helpers;
+pub(crate) mod helpers;
 mod interfaces;
 mod valgrind;
 mod wall_time;