wehub-resource-sync
37 lines
1.2 KiB
TypeScript
37 lines
1.2 KiB
TypeScript
import type { RescheduleRunRequestBody } from "@trigger.dev/core/v3";
|
|
import type { TaskRun } from "@trigger.dev/database";
|
|
import { parseDelay } from "~/utils/delays";
|
|
import { BaseService, ServiceValidationError } from "./baseService.server";
|
|
import { EnqueueDelayedRunService } from "./enqueueDelayedRun.server";
|
|
import { engine } from "../runEngine.server";
|
|
|
|
export class RescheduleTaskRunService extends BaseService {
|
|
public async call(taskRun: TaskRun, body: RescheduleRunRequestBody) {
|
|
if (taskRun.status !== "DELAYED") {
|
|
throw new ServiceValidationError("Cannot reschedule a run that is not delayed");
|
|
}
|
|
|
|
const delay = await parseDelay(body.delay);
|
|
|
|
if (!delay) {
|
|
throw new ServiceValidationError(`Invalid delay: ${body.delay}`);
|
|
}
|
|
|
|
const updatedRun = await this.runStore.rescheduleRun(
|
|
taskRun.id,
|
|
{
|
|
delayUntil: delay,
|
|
queueTimestamp: delay,
|
|
},
|
|
this._prisma
|
|
);
|
|
|
|
if (updatedRun.engine === "V1") {
|
|
await EnqueueDelayedRunService.reschedule(taskRun.id, delay);
|
|
return updatedRun;
|
|
} else {
|
|
return engine.rescheduleDelayedRun({ runId: taskRun.id, delayUntil: delay });
|
|
}
|
|
}
|
|
}
|