make supervisor heartbeat interval configurable

Author: nicktrn

apps/supervisor/src/env.ts

@@ -6,6 +6,7 @@ import { AdditionalEnvVars, BoolEnv } from "./envUtil.js";
 const Env = z.object({
   // This will come from `spec.nodeName` in k8s
   TRIGGER_WORKER_INSTANCE_NAME: z.string().default(randomUUID()),
+  TRIGGER_WORKER_HEARTBEAT_INTERVAL_SECONDS: z.coerce.number().default(30),
 
   // Required settings
   TRIGGER_API_URL: z.string().url(),

apps/supervisor/src/index.ts

@@ -121,6 +121,7 @@ class ManagedSupervisor {
       maxRunCount: env.TRIGGER_DEQUEUE_MAX_RUN_COUNT,
       maxConsumerCount: env.TRIGGER_DEQUEUE_MAX_CONSUMER_COUNT,
       runNotificationsEnabled: env.TRIGGER_WORKLOAD_API_ENABLED,
+      heartbeatIntervalSeconds: env.TRIGGER_WORKER_HEARTBEAT_INTERVAL_SECONDS,
       preDequeue: async () => {
         if (this.isKubernetes) {
           // Not used in k8s for now

packages/core/src/v3/runEngineWorker/supervisor/session.ts

@@ -13,7 +13,7 @@ import { IntervalService } from "../../utils/interval.js";
 type SupervisorSessionOptions = SupervisorClientCommonOptions & {
   queueConsumerEnabled?: boolean;
   runNotificationsEnabled?: boolean;
-  heartbeatIntervalSeconds?: number;
+  heartbeatIntervalSeconds: number;
   dequeueIntervalMs: number;
   dequeueIdleIntervalMs: number;
   preDequeue?: PreDequeueFn;
@@ -32,7 +32,6 @@ export class SupervisorSession extends EventEmitter<WorkerEvents> {
   private readonly queueConsumers: RunQueueConsumer[];
 
   private readonly heartbeat: IntervalService;
-  private readonly heartbeatIntervalSeconds: number;
 
   constructor(private opts: SupervisorSessionOptions) {
     super();
@@ -53,7 +52,6 @@ export class SupervisorSession extends EventEmitter<WorkerEvents> {
       });
     });
 
-    this.heartbeatIntervalSeconds = opts.heartbeatIntervalSeconds || 30;
     this.heartbeat = new IntervalService({
       onInterval: async () => {
         console.debug("[SupervisorSession] Sending heartbeat");
@@ -65,7 +63,7 @@ export class SupervisorSession extends EventEmitter<WorkerEvents> {
           console.error("[SupervisorSession] Heartbeat failed", { error: response.error });
         }
       },
-      intervalMs: this.heartbeatIntervalSeconds * 1000,
+      intervalMs: opts.heartbeatIntervalSeconds * 1000,
       leadingEdge: false,
       onError: async (error) => {
         console.error("[SupervisorSession] Failed to send heartbeat", { error });