feat: parallelize imports patching

Author: not-matthias

go-runner/src/builder/patcher.rs

@@ -1,9 +1,11 @@
 //! Patches the imports to use codspeed rather than the official "testing" package.
 
 use crate::prelude::*;
+use rayon::prelude::*;
 use std::fs;
 use std::path::Path;
 use std::process::Command;
+use std::sync::atomic::AtomicU64;
 
 pub fn replace_pkg<P: AsRef<Path>>(folder: P) -> anyhow::Result<()> {
     let codspeed_root = Path::new(env!("CARGO_MANIFEST_DIR")).parent().unwrap();
@@ -34,28 +36,37 @@ pub fn patch_imports<P: AsRef<Path>>(folder: P) -> anyhow::Result<()> {
     debug!("Patching imports in folder: {folder:?}");
 
     // 1. Find all imports that match "testing" and replace them with codspeed equivalent
-    let mut patched_files = 0;
-
+    let patched_files = AtomicU64::new(0);
     let pattern = folder.join("**/*.go");
-    for go_file in glob::glob(pattern.to_str().unwrap())?.filter_map(Result::ok) {
-        // Skip directories - glob can match directories ending in .go (e.g., vendor/github.com/nats-io/nats.go)
-        if !go_file.is_file() {
-            continue;
-        }
-
-        let content =
-            fs::read_to_string(&go_file).context(format!("Failed to read Go file: {go_file:?}"))?;
-
-        let patched_content = patch_imports_for_source(&content);
-        if patched_content != content {
-            fs::write(&go_file, patched_content)
-                .context(format!("Failed to write patched Go file: {go_file:?}"))?;
-
-            debug!("Patched imports in: {go_file:?}");
-            patched_files += 1;
-        }
-    }
-    debug!("Patched {patched_files} files");
+    glob::glob(pattern.to_str().unwrap())?
+        .filter_map(Result::ok)
+        .par_bridge()
+        .for_each(|go_file| {
+            // Skip directories - glob can match directories ending in .go (e.g., vendor/github.com/nats-io/nats.go)
+            if !go_file.is_file() {
+                return;
+            }
+
+            let Ok(content) = fs::read_to_string(&go_file) else {
+                error!("Failed to read Go file: {go_file:?}");
+                return;
+            };
+
+            let patched_content = patch_imports_for_source(&content);
+            if patched_content != content {
+                let Ok(_) = fs::write(&go_file, patched_content) else {
+                    error!("Failed to write patched Go file: {go_file:?}");
+                    return;
+                };
+
+                debug!("Patched imports in: {go_file:?}");
+                patched_files.fetch_add(1, std::sync::atomic::Ordering::SeqCst);
+            }
+        });
+    debug!(
+        "Patched {} files",
+        patched_files.load(std::sync::atomic::Ordering::SeqCst)
+    );
 
     Ok(())
 }