From d2e4baa48d964c06034b0a79b33c62054a9ebcce Mon Sep 17 00:00:00 2001
From: nharper285 <74685766+nharper285@users.noreply.github.com>
Date: Tue, 9 Mar 2021 07:24:04 -0800
Subject: [PATCH] Updating Task Heartbeat Struct to actually include job_id
 (#646)

---
 src/agent/onefuzz-agent/src/tasks/heartbeat.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/agent/onefuzz-agent/src/tasks/heartbeat.rs b/src/agent/onefuzz-agent/src/tasks/heartbeat.rs
index 634ecada6..4d57021b6 100644
--- 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,6 +54,7 @@ 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
@@ -60,6 +62,7 @@ pub async fn init_task_heartbeat(
                 .enqueue(Heartbeat {
                     task_id,
                     data,
+                    job_id,
                     machine_id,
                     machine_name,
                 })