From 5e25b2f11d8270d6c6274c54b8a4a77513080691 Mon Sep 17 00:00:00 2001
From: asonix <asonix@asonix.dog>
Date: Wed, 10 Jan 2024 15:09:49 -0600
Subject: [PATCH] jobs-actix: use configured heartbeat interval

---
 jobs-actix/src/worker.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/jobs-actix/src/worker.rs b/jobs-actix/src/worker.rs
index 9b390af..62c77a2 100644
--- a/jobs-actix/src/worker.rs
+++ b/jobs-actix/src/worker.rs
@@ -55,8 +55,10 @@ async fn heartbeat_job<F: Future>(
     future: F,
     job_id: Uuid,
     runner_id: Uuid,
+    heartbeat_interval: u64,
 ) -> F::Output {
-    let mut interval = actix_rt::time::interval(std::time::Duration::from_secs(5));
+    let mut interval =
+        actix_rt::time::interval(std::time::Duration::from_millis(heartbeat_interval));
 
     let mut future = std::pin::pin!(future);
 
@@ -173,6 +175,7 @@ pub(crate) async fn local_worker<State, Extras>(
 
         let process_span = make_span(id, &queue, "process");
         let job_id = job.id;
+        let heartbeat_interval = job.heartbeat_interval;
         let return_job = process_span
             .in_scope(|| {
                 heartbeat_job(
@@ -180,6 +183,7 @@ pub(crate) async fn local_worker<State, Extras>(
                     time_job(processors.process(job), job_id),
                     job_id,
                     id,
+                    heartbeat_interval,
                 )
             })
             .instrument(process_span)