fixup: add benches arg for run cmd

Author: not-matthias

go-runner/src/lib.rs

@@ -11,7 +11,6 @@ mod results;
 /// Builds the specified go project and writes the benchmarks to the .codspeed folder.
 pub fn build_benchmarks(project_dir: &Path) -> anyhow::Result<Vec<PathBuf>> {
     // 1. Benchmark and package discovery
-    //
     let benchmark_packages = builder::discovery::run(project_dir)?;
     info!("Discovered {} packages", benchmark_packages.len());
     debug!("Packages: {:?}", benchmark_packages.keys());
@@ -49,8 +48,7 @@ pub fn build_benchmarks(project_dir: &Path) -> anyhow::Result<Vec<PathBuf>> {
     Ok(binaries)
 }
 
-/// This will be invoked by `codspeed run -- run_benchmarks`
-pub fn run_benchmarks() -> anyhow::Result<()> {
+pub fn run_benchmarks(bench: &str) -> anyhow::Result<()> {
     // 1. Discover the benchmarks
     let codspeed_dir = std::env::current_dir()?.join(".codspeed").join("walltime");
     let benchmark_packages = std::fs::read_dir(&codspeed_dir)?
@@ -64,7 +62,7 @@ pub fn run_benchmarks() -> anyhow::Result<()> {
         info!("Running: {}", bench_path.display());
 
         let cmd = std::process::Command::new(bench_path)
-            .arg("-test.bench=.") // TODO: Pass the args
+            .arg(format!("-test.bench={bench}"))
             .output()?;
 
         // Check if the command was successful
@@ -83,6 +81,7 @@ pub fn run_benchmarks() -> anyhow::Result<()> {
         }
     }
 
+    // 3. Collect the results
     collect_walltime_results()?;
 
     Ok(())

go-runner/src/main.rs

@@ -17,7 +17,19 @@ enum Commands {
         path: PathBuf,
     },
     /// Run the already built benchmarks
-    Run,
+    Run {
+        /// Equivalent to the `-bench regexp` flag in `go test`.
+        ///
+        /// Run only those benchmarks matching a regular expression.
+        /// By default, no benchmarks are run.
+        /// To run all benchmarks, use '-bench .' or '-bench=.'.
+        /// The regular expression is split by unbracketed slash (/)
+        /// characters into a sequence of regular expressions, and each
+        /// part of a benchmark's identifier must match the corresponding
+        /// element in the sequence, if any.
+        #[arg(long, default_value = ".")]
+        bench: String,
+    },
 }
 
 fn main() -> anyhow::Result<()> {
@@ -29,8 +41,8 @@ fn main() -> anyhow::Result<()> {
         Commands::Build { path } => {
             go_runner::build_benchmarks(path)?;
         }
-        Commands::Run => {
-            go_runner::run_benchmarks()?;
+        Commands::Run { bench } => {
+            go_runner::run_benchmarks(bench)?;
         }
     }