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,36 @@
+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 });
+    }
+  }
+}