wehub-resource-sync
40 lines
1.5 KiB
TypeScript
40 lines
1.5 KiB
TypeScript
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}`);
|
|
}
|
|
}
|