ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-17 20:38:06 +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)]
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,
})