ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-18 04:38:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
nharper285
2021-03-09 07:24:04 -08:00
committed by GitHub
parent 0a3812d8bc
commit d2e4baa48d
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/agent/onefuzz-agent/src/tasks/heartbeat.rs
View File

@ -18,6 +18,7 @@ pub enum HeartbeatData {
#[derive(Debug, Deserialize, Serialize, Clone)] #[derive(Debug, Deserialize, Serialize, Clone)]
struct Heartbeat { struct Heartbeat {
task_id: Uuid, task_id: Uuid,
job_id: Uuid,
machine_id: Uuid, machine_id: Uuid,
machine_name: String, machine_name: String,
data: Vec<HeartbeatData>, data: Vec<HeartbeatData>,
@ -53,6 +54,7 @@ pub async fn init_task_heartbeat(
let task_id = context.state.task_id; let task_id = context.state.task_id;
let machine_id = context.state.machine_id; let machine_id = context.state.machine_id;
let machine_name = context.state.machine_name.clone(); 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 data = HeartbeatClient::<TaskContext, _>::drain_current_messages(context.clone());
let _ = context let _ = context
@ -60,6 +62,7 @@ pub async fn init_task_heartbeat(
.enqueue(Heartbeat { .enqueue(Heartbeat {
task_id, task_id,
data, data,
job_id,
machine_id, machine_id,
machine_name, machine_name,
}) })