Updating Task Heartbeat Struct to actually include job_id (#646)

microsoft · Mar 9, 2021 · d2e4baa · d2e4baa
1 parent 0a3812d
commit d2e4baa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/agent/onefuzz-agent/src/tasks/heartbeat.rs b/src/agent/onefuzz-agent/src/tasks/heartbeat.rs
@@ -18,6 +18,7 @@ pub enum HeartbeatData {
 #[derive(Debug, Deserialize, Serialize, Clone)]
 struct Heartbeat {
     task_id: Uuid,
+    job_id: Uuid,
     machine_id: Uuid,
     machine_name: String,
     data: Vec<HeartbeatData>,
@@ -53,13 +54,15 @@ pub async fn init_task_heartbeat(
             let task_id = context.state.task_id;
             let machine_id = context.state.machine_id;
             let machine_name = context.state.machine_name.clone();
+            let job_id = context.state.job_id;
 
             let data = HeartbeatClient::<TaskContext, _>::drain_current_messages(context.clone());
             let _ = context
                 .queue_client
                 .enqueue(Heartbeat {
                     task_id,
                     data,
+                    job_id,
                     machine_id,
                     machine_name,
                 })