@kb-labs/workflow-daemon 2.119.0 → 2.119.1-canary.b53bbee1d

package/dist/index.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 import { makeAssemblyHook } from '@kb-labs/plugin-runtime';
-import { WorkflowEngine, WorkflowService } from '@kb-labs/workflow-engine';
+import { WorkflowEngine, DAEMON_LEASE_HEARTBEAT_INTERVAL_MS, WorkflowService } from '@kb-labs/workflow-engine';
 import { getListenOptions, HttpObservabilityCollector, createDaemonServer, createServiceReadyResponse, metricLine } from '@kb-labs/shared-http';
 import { runService } from '@kb-labs/shared-daemon';
 import { logDiagnosticEvent, createContextLogger } from '@kb-labs/core-platform';
@@ -3272,6 +3272,14 @@ async function bootstrap(_cwd = process.cwd()) {
         "Cleaning up stale runs from previous daemon process"
       );
       await engine.cleanupStaleRuns();
+      const leaseHeartbeat = setInterval(() => {
+        engine.renewDaemonLease().catch((error) => {
+          bootstrapLogger.warn("Failed to renew daemon liveness lease", {
+            err: error instanceof Error ? error.message : String(error)
+          });
+        });
+      }, DAEMON_LEASE_HEARTBEAT_INTERVAL_MS);
+      leaseHeartbeat.unref?.();
       bootstrapLogger.info("Resuming interrupted jobs");
       await engine.resumeInterruptedJobs();
       bootstrapLogger.info("Reconciling persisted child workflow invocations");
@@ -3363,6 +3371,7 @@ async function bootstrap(_cwd = process.cwd()) {
       bootstrapLogger.info("Workflow daemon started successfully", { port });
       return async () => {
         bootstrapLogger.warn("Stopping workflow daemon components");
+        clearInterval(leaseHeartbeat);
         fileWatcher.close();
         await cronScheduler.stop();
         await worker.stop();