Get max duration working on deployed runs

Author: ericallam

apps/webapp/app/v3/marqs/sharedQueueConsumer.server.ts

@@ -403,6 +403,8 @@ export class SharedQueueConsumer {
             startedAt: existingTaskRun.startedAt ?? new Date(),
             baseCostInCents: env.CENTS_PER_RUN,
             machinePreset: machinePresetFromConfig(backgroundTask.machineConfig ?? {}).name,
+            maxDurationInSeconds:
+              existingTaskRun.maxDurationInSeconds ?? backgroundTask.maxDurationInSeconds,
           },
           include: {
             runtimeEnvironment: true,
@@ -1067,6 +1069,7 @@ class SharedQueueTasks {
         costInCents: taskRun.costInCents,
         baseCostInCents: taskRun.baseCostInCents,
         metadata,
+        maxDuration: taskRun.maxDurationInSeconds ?? undefined,
       },
       queue: {
         id: queue.friendlyId,

packages/cli-v3/src/entryPoints/deploy-run-worker.ts

@@ -338,7 +338,13 @@ const zodIpc = new ZodIpcConnection({
             }
           });
 
-          const { result } = await executor.execute(execution, metadata, traceContext, measurement);
+          const { result } = await executor.execute(
+            execution,
+            metadata,
+            traceContext,
+            measurement,
+            signal
+          );
 
           const usageSample = usage.stop(measurement);
 

references/hello-world/src/trigger/example.ts

@@ -1,4 +1,4 @@
-import { logger, task, wait } from "@trigger.dev/sdk/v3";
+import { logger, task, usage, wait } from "@trigger.dev/sdk/v3";
 import { setTimeout } from "timers/promises";
 
 export const helloWorldTask = task({
@@ -42,3 +42,24 @@ export const childTask = task({
     }
   },
 });
+
+export const maxDurationTask = task({
+  id: "max-duration",
+  run: async (payload: { sleepFor: number }, { signal }) => {
+    await setTimeout(payload.sleepFor * 1000, { signal });
+
+    return usage.getCurrent();
+  },
+});
+
+export const maxDurationParentTask = task({
+  id: "max-duration-parent",
+  run: async (payload: { sleepFor?: number; maxDuration?: number }, { ctx, signal }) => {
+    const result = await maxDurationTask.triggerAndWait(
+      { sleepFor: payload.sleepFor ?? 10 },
+      { maxDuration: payload.maxDuration ?? 600 }
+    );
+
+    return result;
+  },
+});

references/hello-world/trigger.config.ts

@@ -3,6 +3,7 @@ import { defineConfig } from "@trigger.dev/sdk/v3";
 export default defineConfig({
   project: "proj_rrkpdguyagvsoktglnod",
   logLevel: "log",
+  maxDuration: 60,
   retries: {
     enabledInDev: true,
     default: {