feat: expose a collection entrypoint usable within walltime integrations

Author: art049

Cargo.lock

@@ -8,3 +8,7 @@ members = [
     "crates/divan_compat/macros",
 ]
 resolver = "2"
+
+[workspace.dependencies]
+serde = { version = "1.0.217", features = ["derive"] }
+serde_json = "1.0.138"

Cargo.toml

@@ -20,7 +20,9 @@ keywords = ["codspeed", "benchmark"]
 [dependencies]
 colored = "2.0.0"
 libc = "^0.2"
-serde_json = "1.0.120"
+serde = { workspace = true }
+serde_json = { workspace = true }
+uuid = { version = "1.12.1", features = ["v4"] }
 
 [[bench]]
 name = "native"

crates/codspeed/Cargo.toml

@@ -3,3 +3,4 @@ mod macros;
 mod measurement;
 mod request;
 pub mod utils;
+pub mod walltime;

crates/codspeed/src/lib.rs

@@ -0,0 +1,66 @@
+use std::{
+    io::Write,
+    path::{Path, PathBuf},
+};
+
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct BenchmarkMetadata {
+    pub name: String,
+    pub uri: String,
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct RawWallTimeData {
+    #[serde(flatten)]
+    pub metadata: BenchmarkMetadata,
+    pub iter_per_round: u32,
+    pub times_ps: Vec<u128>,
+}
+
+impl RawWallTimeData {
+    fn from_runtime_data(
+        name: String,
+        uri: String,
+        iter_per_round: u32,
+        times_ps: Vec<u128>,
+    ) -> Self {
+        RawWallTimeData {
+            metadata: BenchmarkMetadata { name, uri },
+            iter_per_round,
+            times_ps,
+        }
+    }
+
+    fn dump_to_results(&self, workspace_root: &Path, scope: &str) {
+        let output_dir = workspace_root
+            .join("target")
+            .join("codspeed")
+            .join("raw_results")
+            .join(scope);
+        std::fs::create_dir_all(&output_dir).unwrap();
+        let bench_id = uuid::Uuid::new_v4().to_string();
+        let output_path = output_dir.join(format!("{}.json", bench_id));
+        let mut writer = std::fs::File::create(&output_path).expect("Failed to create the file");
+        serde_json::to_writer_pretty(&mut writer, self).expect("Failed to write the data");
+        writer.flush().expect("Failed to flush the writer");
+    }
+}
+
+/// Entry point called in patched integration to harvest raw walltime data
+pub fn collect_raw_walltime_results(
+    scope: &str,
+    name: String,
+    uri: String,
+    iter_per_round: u32,
+    times_ps: Vec<u128>,
+) {
+    let workspace_root = std::env::var("CODSPEED_CARGO_WORKSPACE_ROOT").map(PathBuf::from);
+    let Ok(workspace_root) = workspace_root else {
+        eprintln!("codspeed failed to get workspace root. skipping");
+        return;
+    };
+    let data = RawWallTimeData::from_runtime_data(name, uri, iter_per_round, times_ps);
+    data.dump_to_results(&workspace_root, scope);
+}