feat: parse golang test output

Author: not-matthias

src/run/runner/helpers/run_command_with_log_pipe.rs

@@ -5,22 +5,37 @@ use std::future::Future;
 use std::io::{Read, Write};
 use std::process::Command;
 use std::process::ExitStatus;
+use std::sync::{Arc, Mutex};
 use std::thread;
 
+struct CmdRunnerOptions<F> {
+    on_process_spawned: Option<F>,
+    capture_stdout: bool,
+}
+
+impl<F> Default for CmdRunnerOptions<F> {
+    fn default() -> Self {
+        Self {
+            on_process_spawned: None,
+            capture_stdout: false,
+        }
+    }
+}
+
 /// Run a command and log its output to stdout and stderr
 ///
 /// # Arguments
 /// - `cmd`: The command to run.
-/// - `cb`: A callback function that takes the process ID and returns a result.
+/// - `options`: Configuration options for the runner (e.g. capture output, run callback)
 ///
 /// # Returns
-///
-/// The exit status of the command.
-///
-pub async fn run_command_with_log_pipe_and_callback<F, Fut>(
+/// A tuple containing:
+/// - `ExitStatus`: The exit status of the executed command
+/// - `Option<String>`: Captured stdout if `capture_stdout` was true, otherwise None
+async fn run_command_with_log_pipe_and_options<F, Fut>(
     mut cmd: Command,
-    cb: F,
-) -> Result<ExitStatus>
+    options: CmdRunnerOptions<F>,
+) -> Result<(ExitStatus, Option<String>)>
 where
     F: FnOnce(u32) -> Fut,
     Fut: Future<Output = anyhow::Result<()>>,
@@ -29,6 +44,7 @@ where
         mut reader: impl Read,
         mut writer: impl Write,
         log_prefix: Option<&str>,
+        captured_output: Option<Arc<Mutex<Vec<u8>>>>,
     ) -> Result<()> {
         let prefix = log_prefix.unwrap_or("");
         let mut buffer = [0; 1024];
@@ -37,6 +53,12 @@ where
             if bytes_read == 0 {
                 break;
             }
+
+            if let Some(ref capture) = captured_output {
+                let mut output = capture.lock().unwrap();
+                output.extend_from_slice(&buffer[..bytes_read]);
+            }
+
             suspend_progress_bar(|| {
                 writer.write_all(&buffer[..bytes_read]).unwrap();
                 trace!(
@@ -57,19 +79,76 @@ where
         .context("failed to spawn the process")?;
     let stdout = process.stdout.take().expect("unable to get stdout");
     let stderr = process.stderr.take().expect("unable to get stderr");
-    thread::spawn(move || {
-        log_tee(stdout, std::io::stdout(), None).unwrap();
-    });
 
-    thread::spawn(move || {
-        log_tee(stderr, std::io::stderr(), Some("[stderr]")).unwrap();
-    });
+    let captured_stdout = if options.capture_stdout {
+        Some(Arc::new(Mutex::new(Vec::new())))
+    } else {
+        None
+    };
+    let (stdout_handle, stderr_handle) = {
+        let stdout_capture = captured_stdout.clone();
+        let stdout_handle = thread::spawn(move || {
+            log_tee(stdout, std::io::stdout(), None, stdout_capture).unwrap();
+        });
+        let stderr_handle = thread::spawn(move || {
+            log_tee(stderr, std::io::stderr(), Some("[stderr]"), None).unwrap();
+        });
 
-    cb(process.id()).await?;
+        (stdout_handle, stderr_handle)
+    };
 
-    process.wait().context("failed to wait for the process")
+    if let Some(cb) = options.on_process_spawned {
+        cb(process.id()).await?;
+    }
+
+    let exit_status = process.wait().context("failed to wait for the process")?;
+    let _ = (stdout_handle.join().unwrap(), stderr_handle.join().unwrap());
+
+    let stdout_output = captured_stdout
+        .map(|capture| String::from_utf8_lossy(&capture.lock().unwrap()).to_string());
+    Ok((exit_status, stdout_output))
+}
+
+pub async fn run_command_with_log_pipe_and_callback<F, Fut>(
+    cmd: Command,
+    cb: F,
+) -> Result<(ExitStatus, Option<String>)>
+where
+    F: FnOnce(u32) -> Fut,
+    Fut: Future<Output = anyhow::Result<()>>,
+{
+    run_command_with_log_pipe_and_options(
+        cmd,
+        CmdRunnerOptions {
+            on_process_spawned: Some(cb),
+            capture_stdout: false,
+        },
+    )
+    .await
 }
 
 pub async fn run_command_with_log_pipe(cmd: Command) -> Result<ExitStatus> {
-    run_command_with_log_pipe_and_callback(cmd, async |_| Ok(())).await
+    let (exit_status, _) = run_command_with_log_pipe_and_options(
+        cmd,
+        CmdRunnerOptions::<fn(u32) -> futures::future::Ready<anyhow::Result<()>>> {
+            on_process_spawned: None,
+            capture_stdout: false,
+        },
+    )
+    .await?;
+    Ok(exit_status)
+}
+
+pub async fn run_command_with_log_pipe_capture_stdout(
+    cmd: Command,
+) -> Result<(ExitStatus, String)> {
+    let (exit_status, stdout) = run_command_with_log_pipe_and_options(
+        cmd,
+        CmdRunnerOptions::<fn(u32) -> futures::future::Ready<anyhow::Result<()>>> {
+            on_process_spawned: None,
+            capture_stdout: true,
+        },
+    )
+    .await?;
+    Ok((exit_status, stdout.unwrap_or_default()))
 }

src/run/runner/wall_time/executor.rs

@@ -5,7 +5,8 @@ use crate::run::instruments::mongo_tracer::MongoTracer;
 use crate::run::runner::executor::Executor;
 use crate::run::runner::helpers::env::{get_base_injected_env, is_codspeed_debug_enabled};
 use crate::run::runner::helpers::get_bench_command::get_bench_command;
-use crate::run::runner::helpers::run_command_with_log_pipe::run_command_with_log_pipe;
+use crate::run::runner::helpers::run_command_with_log_pipe::run_command_with_log_pipe_capture_stdout;
+use crate::run::runner::wall_time::golang;
 use crate::run::runner::{ExecutorName, RunData};
 use crate::run::{check_system::SystemInfo, config::Config};
 use async_trait::async_trait;
@@ -181,7 +182,16 @@ impl Executor for WallTimeExecutor {
                 cmd.args(["sh", "-c", &bench_cmd]);
                 debug!("cmd: {cmd:?}");
 
-                run_command_with_log_pipe(cmd).await
+                let (status, stdout) = run_command_with_log_pipe_capture_stdout(cmd).await?;
+
+                if config.command.trim().starts_with("go test") {
+                    let results_folder = run_data.profile_folder.join("results");
+                    std::fs::create_dir_all(&results_folder)?;
+
+                    golang::collect_walltime_results(&stdout, &results_folder)?;
+                }
+
+                Ok(status)
             }
         };
 

src/run/runner/wall_time/perf/mod.rs

@@ -133,7 +133,9 @@ impl PerfRunner {
 
             Ok(())
         };
-        run_command_with_log_pipe_and_callback(cmd, on_process_started).await
+        run_command_with_log_pipe_and_callback(cmd, on_process_started)
+            .await
+            .map(|(exit_status, _)| exit_status)
     }
 
     pub async fn save_files_to(&self, profile_folder: &PathBuf) -> anyhow::Result<()> {