chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,39 @@
+import { legacyRunEngineWorker } from "../legacyRunEngineWorker.server";
+import { TaskRunHeartbeatFailedService } from "../taskRunHeartbeatFailed.server";
+import type { VisibilityTimeoutStrategy } from "./types";
+
+export class V3GraphileVisibilityTimeout implements VisibilityTimeoutStrategy {
+  async startHeartbeat(messageId: string, timeoutInMs: number): Promise<void> {
+    await TaskRunHeartbeatFailedService.enqueue(messageId, new Date(Date.now() + timeoutInMs));
+  }
+
+  async heartbeat(messageId: string, timeoutInMs: number): Promise<void> {
+    await TaskRunHeartbeatFailedService.enqueue(messageId, new Date(Date.now() + timeoutInMs));
+  }
+
+  async cancelHeartbeat(messageId: string): Promise<void> {
+    await TaskRunHeartbeatFailedService.dequeue(messageId);
+  }
+}
+
+export class V3LegacyRunEngineWorkerVisibilityTimeout implements VisibilityTimeoutStrategy {
+  async startHeartbeat(messageId: string, timeoutInMs: number): Promise<void> {
+    await legacyRunEngineWorker.enqueue({
+      id: `heartbeat:${messageId}`,
+      job: "runHeartbeat",
+      payload: { runId: messageId },
+      availableAt: new Date(Date.now() + timeoutInMs),
+    });
+  }
+
+  async heartbeat(messageId: string, timeoutInMs: number): Promise<void> {
+    await legacyRunEngineWorker.reschedule(
+      `heartbeat:${messageId}`,
+      new Date(Date.now() + timeoutInMs)
+    );
+  }
+
+  async cancelHeartbeat(messageId: string): Promise<void> {
+    await legacyRunEngineWorker.ack(`heartbeat:${messageId}`);
+  }
+}