Add benchmark whitelist flag

alacritty · Oct 20, 2024 · 9db239f · 9db239f
1 parent b6f91be
commit 9db239f
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 7 deletions.
diff --git a/src/bench.rs b/src/bench.rs
@@ -31,7 +31,7 @@ impl Display for Error {
 }
 
 /// Find all benchmarks recursively in the specified location.
-pub fn find_benchmarks(paths: &[PathBuf]) -> Vec<BenchmarkLoader> {
+pub fn find_benchmarks(paths: &[PathBuf], names: &Vec<String>) -> Vec<BenchmarkLoader> {
     let mut benchmarks: HashMap<String, (Option<PathBuf>, Option<PathBuf>)> = HashMap::new();
 
     for path in paths {
@@ -47,11 +47,14 @@ pub fn find_benchmarks(paths: &[PathBuf]) -> Vec<BenchmarkLoader> {
 
             if let Some(name) = path.parent().and_then(|path| path.file_name()) {
                 let name = name.to_string_lossy().to_string();
-                let bench = benchmarks.entry(name).or_default();
-                match file_name.as_str() {
-                    "setup" => bench.0 = Some(path),
-                    "benchmark" => bench.1 = Some(path),
-                    _ => unreachable!(),
+
+                if names.len() == 0 || names.contains(&name) {
+                    let bench = benchmarks.entry(name).or_default();
+                    match file_name.as_str() {
+                        "setup" => bench.0 = Some(path),
+                        "benchmark" => bench.1 = Some(path),
+                        _ => unreachable!(),
+                    }
                 }
             }
         }

diff --git a/src/cli.rs b/src/cli.rs
@@ -1,6 +1,7 @@
 //! CLI argument parsing.
 
 use std::path::PathBuf;
+use std::string::String;
 
 use clap::Parser;
 
@@ -15,6 +16,10 @@ pub struct Config {
     #[clap(short, long, value_name = "DIRECTORY", default_value = "./benchmarks")]
     pub benchmarks: Vec<PathBuf>,
 
+    /// Benchmark name.
+    #[clap(short, long, value_name = "WHITELIST")]
+    pub whitelist: Vec<String>,
+
     /// Number of warmup iterations.
     #[clap(long, value_name = "NUM", default_value = "1")]
     pub warmup: usize,

diff --git a/src/main.rs b/src/main.rs
@@ -15,7 +15,7 @@ fn main() {
     let config = Config::parse();
 
     // Find all available benchmarks.
-    let mut loaders = bench::find_benchmarks(&config.benchmarks);
+    let mut loaders = bench::find_benchmarks(&config.benchmarks, &config.whitelist);
 
     // Run all benchmarks one at a time.
     let results: Vec<Results> = loaders