feat: add local_data to upload metadata

GuillaumeLagrange · claude · GuillaumeLagrange · commit 4449281ece4f · 2026-03-05T15:12:48.000+01:00
Add a `local_data` field to `RunEnvironmentMetadata` to carry
local-run specific information (expected run parts count).
Also populate `run_part` for local runs with a stable run_id
and per-executor run_part_id.

Co-Authored-By: Claude Sonnet 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -71,6 +71,7 @@ open = "5.3.2"
 tabled = { version = "0.20.0", features = ["ansi"] }
 shell-words = "1.1.0"
 rmp-serde = "1.3.0"
+uuid = { version = "1.21.0", features = ["v4"] }
 
 [target.'cfg(target_os = "linux")'.dependencies]
 procfs = "0.17.0"
diff --git a/src/run_environment/buildkite/provider.rs b/src/run_environment/buildkite/provider.rs
@@ -142,6 +142,7 @@ impl RunEnvironmentProvider for BuildkiteProvider {
             repository_root_path: self.repository_root_path.clone(),
             gh_data: None,
             gl_data: None,
+            local_data: None,
             sender: None,
         })
     }
diff --git a/src/run_environment/buildkite/snapshots/codspeed_runner__run_environment__buildkite__provider__tests__pull_request_run_environment_metadata.snap b/src/run_environment/buildkite/snapshots/codspeed_runner__run_environment__buildkite__provider__tests__pull_request_run_environment_metadata.snap
@@ -12,5 +12,6 @@ expression: run_environment_metadata
   "sender": null,
   "ghData": null,
   "glData": null,
+  "localData": null,
   "repositoryRootPath": "/buildkite/builds/7b10eca7600b-1/my-org/buildkite-test/"
 }
diff --git a/src/run_environment/github_actions/provider.rs b/src/run_environment/github_actions/provider.rs
@@ -179,6 +179,7 @@ impl RunEnvironmentProvider for GitHubActionsProvider {
             event: self.event.clone(),
             gh_data: Some(self.gh_data.clone()),
             gl_data: None,
+            local_data: None,
             sender: self.sender.clone(),
             owner: self.owner.clone(),
             repository: self.repository.clone(),
diff --git a/src/run_environment/github_actions/snapshots/codspeed_runner__run_environment__github_actions__provider__tests__fork_pull_request_run_environment_metadata.snap b/src/run_environment/github_actions/snapshots/codspeed_runner__run_environment__github_actions__provider__tests__fork_pull_request_run_environment_metadata.snap
@@ -18,5 +18,6 @@ expression: run_environment_metadata
     "job": "log-env"
   },
   "glData": null,
+  "localData": null,
   "repositoryRootPath": "/home/runner/work/adrien-python-test/adrien-python-test/"
 }
diff --git a/src/run_environment/github_actions/snapshots/codspeed_runner__run_environment__github_actions__provider__tests__matrix_job_run_environment_metadata.snap b/src/run_environment/github_actions/snapshots/codspeed_runner__run_environment__github_actions__provider__tests__matrix_job_run_environment_metadata.snap
@@ -18,5 +18,6 @@ expression: run_environment_metadata
     "job": "log-env"
   },
   "glData": null,
+  "localData": null,
   "repositoryRootPath": "/home/runner/work/adrien-python-test/adrien-python-test/"
 }
diff --git a/src/run_environment/github_actions/snapshots/codspeed_runner__run_environment__github_actions__provider__tests__pull_request_run_environment_metadata.snap b/src/run_environment/github_actions/snapshots/codspeed_runner__run_environment__github_actions__provider__tests__pull_request_run_environment_metadata.snap
@@ -18,5 +18,6 @@ expression: run_environment_metadata
     "job": "log-env"
   },
   "glData": null,
+  "localData": null,
   "repositoryRootPath": "/home/runner/work/adrien-python-test/adrien-python-test/"
 }
diff --git a/src/run_environment/gitlab_ci/provider.rs b/src/run_environment/gitlab_ci/provider.rs
@@ -161,6 +161,7 @@ impl RunEnvironmentProvider for GitLabCIProvider {
             event: self.event.clone(),
             gh_data: None,
             gl_data: Some(self.gl_data.clone()),
+            local_data: None,
             sender: Some(self.sender.clone()),
             owner: self.owner.clone(),
             repository: self.repository.clone(),
diff --git a/src/run_environment/gitlab_ci/snapshots/codspeed_runner__run_environment__gitlab_ci__provider__tests__fork_merge_request_run_environment_metadata.snap b/src/run_environment/gitlab_ci/snapshots/codspeed_runner__run_environment__gitlab_ci__provider__tests__fork_merge_request_run_environment_metadata.snap
@@ -18,5 +18,6 @@ expression: run_environment_metadata
     "runId": "6957110437",
     "job": "build-job"
   },
+  "localData": null,
   "repositoryRootPath": "/builds/owner/repository"
 }
diff --git a/src/run_environment/gitlab_ci/snapshots/codspeed_runner__run_environment__gitlab_ci__provider__tests__merge_request_run_environment_metadata.snap b/src/run_environment/gitlab_ci/snapshots/codspeed_runner__run_environment__gitlab_ci__provider__tests__merge_request_run_environment_metadata.snap
@@ -18,5 +18,6 @@ expression: run_environment_metadata
     "runId": "6957110437",
     "job": "build-job"
   },
+  "localData": null,
   "repositoryRootPath": "/builds/owner/repository"
 }
diff --git a/src/run_environment/gitlab_ci/snapshots/codspeed_runner__run_environment__gitlab_ci__provider__tests__push_main_run_environment_metadata.snap b/src/run_environment/gitlab_ci/snapshots/codspeed_runner__run_environment__gitlab_ci__provider__tests__push_main_run_environment_metadata.snap
@@ -18,5 +18,6 @@ expression: run_environment_metadata
     "runId": "1234567890",
     "job": "job"
   },
+  "localData": null,
   "repositoryRootPath": "/builds/owner/repository"
 }
diff --git a/src/run_environment/interfaces.rs b/src/run_environment/interfaces.rs
@@ -33,6 +33,7 @@ pub struct RunEnvironmentMetadata {
     pub sender: Option<Sender>,
     pub gh_data: Option<GhData>,
     pub gl_data: Option<GlData>,
+    pub local_data: Option<LocalData>,
     pub repository_root_path: String,
 }
 
@@ -60,6 +61,12 @@ pub struct GlData {
     pub job: String,
 }
 
+#[derive(Deserialize, Serialize, Debug, Clone)]
+#[serde(rename_all = "camelCase")]
+pub struct LocalData {
+    pub expected_run_parts_count: u32,
+}
+
 /// Each execution of the CLI maps to a `RunPart`.
 ///
 /// Several `RunParts` can be aggregated in a single `Run` thanks to this data.
diff --git a/src/run_environment/local/provider.rs b/src/run_environment/local/provider.rs
diff --git a/src/upload/interfaces.rs b/src/upload/interfaces.rs
diff --git a/src/upload/snapshots/codspeed_runner__upload__upload_metadata__tests__get_local_metadata_hash-2.snap b/src/upload/snapshots/codspeed_runner__upload__upload_metadata__tests__get_local_metadata_hash-2.snap
diff --git a/src/upload/snapshots/codspeed_runner__upload__upload_metadata__tests__get_metadata_hash-2.snap b/src/upload/snapshots/codspeed_runner__upload__upload_metadata__tests__get_metadata_hash-2.snap
diff --git a/src/upload/upload_metadata.rs b/src/upload/upload_metadata.rs

Original file line number	Diff line number	Diff line change
`@@ -142,6 +142,7 @@ impl RunEnvironmentProvider for BuildkiteProvider {`
`142`	`142`	`repository_root_path: self.repository_root_path.clone(),`
`143`	`143`	`gh_data: None,`
`144`	`144`	`gl_data: None,`
	`145`	`+ local_data: None,`
`145`	`146`	`sender: None,`
`146`	`147`	`})`
`147`	`148`	`}`
Original file line number	Diff line number	Diff line change
`@@ -12,5 +12,6 @@ expression: run_environment_metadata`
`12`	`12`	`"sender": null,`
`13`	`13`	`"ghData": null,`
`14`	`14`	`"glData": null,`
	`15`	`+ "localData": null,`
`15`	`16`	`"repositoryRootPath": "/buildkite/builds/7b10eca7600b-1/my-org/buildkite-test/"`
`16`	`17`	`}`
Original file line number	Diff line number	Diff line change
`@@ -18,5 +18,6 @@ expression: run_environment_metadata`
`18`	`18`	`"job": "log-env"`
`19`	`19`	`},`
`20`	`20`	`"glData": null,`
	`21`	`+ "localData": null,`
`21`	`22`	`"repositoryRootPath": "/home/runner/work/adrien-python-test/adrien-python-test/"`
`22`	`23`	`}`