refactor: use helper to check if running with codspeed runner

Author: not-matthias

crates/codspeed/src/utils.rs

@@ -38,6 +38,10 @@ pub fn get_formated_function_path(function_path: impl Into<String>) -> String {
     function_path.replace(" :: ", "::")
 }
 
+pub fn running_with_codspeed_runner() -> bool {
+    std::env::var("CODSPEED_ENV").is_ok()
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

crates/codspeed/src/walltime.rs

@@ -58,7 +58,7 @@ pub fn collect_raw_walltime_results(
     max_time_ns: Option<u128>,
     times_ns: Vec<u128>,
 ) {
-    if std::env::var("CODSPEED_ENV").is_err() {
+    if !crate::utils::running_with_codspeed_runner() {
         return;
     }
     let workspace_root = std::env::var("CODSPEED_CARGO_WORKSPACE_ROOT").map(PathBuf::from);

crates/criterion_compat/criterion_fork/src/analysis/mod.rs

@@ -257,7 +257,7 @@ pub(crate) fn common<M: Measurement, T: ?Sized>(
         }
     }
 
-    if criterion.should_save_baseline() && std::env::var("CODSPEED_ENV").is_ok() {
+    if criterion.should_save_baseline() && ::codspeed::utils::running_with_codspeed_runner() {
         codspeed::collect_walltime_results(id, criterion, &iters, avg_times);
     }
 }
@@ -301,7 +301,7 @@ mod codspeed {
             pid: std::process::id(),
             uri: uri.clone(),
         }) {
-            if std::env::var("CODSPEED_ENV").is_ok() {
+            if codspeed::utils::running_with_codspeed_runner() {
                 eprintln!("Failed to send benchmark URI to runner: {error:?}");
             }
         }

crates/criterion_compat/criterion_fork/src/macros.rs

@@ -68,7 +68,7 @@ macro_rules! criterion_group {
             let mut criterion: $crate::Criterion<_> = $config
                 .configure_from_args();
             $(
-                if std::env::var("CODSPEED_ENV").is_ok() {
+                if codspeed::utils::running_with_codspeed_runner()  {
                     criterion.set_current_file($crate::abs_file!());
                     criterion.set_macro_group(format!("{}::{}", stringify!($name), stringify!($target)));
                 }

crates/divan_compat/divan_fork/src/divan.rs

@@ -432,7 +432,7 @@ mod codspeed {
             pid: std::process::id(),
             uri: uri.clone(),
         }) {
-            if std::env::var("CODSPEED_ENV").is_ok() {
+            if codspeed::utils::running_with_codspeed_runner() {
                 eprintln!("Failed to send benchmark URI to runner: {error:?}");
             }
         }