set dequeued at on new runners

Author: nicktrn

apps/supervisor/src/index.ts

@@ -224,6 +224,7 @@ class ManagedSupervisor {
 
       try {
         await this.workloadManager.create({
+          dequeuedAt: message.dequeuedAt,
           envId: message.environment.id,
           envType: message.environment.type,
           image: message.image,

apps/supervisor/src/workloadManager/docker.ts

@@ -28,6 +28,7 @@ export class DockerWorkloadManager implements WorkloadManager {
       "run",
       "--detach",
       `--network=${env.DOCKER_NETWORK}`,
+      `--env=TRIGGER_DEQUEUED_AT_MS=${opts.dequeuedAt.getTime()}`,
       `--env=TRIGGER_POD_SCHEDULED_AT_MS=${Date.now()}`,
       `--env=TRIGGER_ENV_ID=${opts.envId}`,
       `--env=TRIGGER_RUN_ID=${opts.runFriendlyId}`,

apps/supervisor/src/workloadManager/kubernetes.ts

@@ -61,6 +61,10 @@ export class KubernetesWorkloadManager implements WorkloadManager {
                 ],
                 resources: this.#getResourcesForMachine(opts.machine),
                 env: [
+                  {
+                    name: "TRIGGER_DEQUEUED_AT_MS",
+                    value: opts.dequeuedAt.getTime().toString(),
+                  },
                   {
                     name: "TRIGGER_POD_SCHEDULED_AT_MS",
                     value: Date.now().toString(),

apps/supervisor/src/workloadManager/types.ts

@@ -21,6 +21,7 @@ export interface WorkloadManagerCreateOptions {
   machine: MachinePreset;
   version: string;
   nextAttemptNumber?: number;
+  dequeuedAt: Date;
   // identifiers
   envId: string;
   envType: EnvironmentType;

packages/cli-v3/src/entryPoints/managed-run-controller.ts

@@ -26,6 +26,11 @@ import { assertExhaustive } from "../utilities/assertExhaustive.js";
 import { setTimeout as sleep } from "timers/promises";
 import { io, type Socket } from "socket.io-client";
 
+const DateEnv = z
+  .string()
+  .transform((val) => new Date(parseInt(val, 10)))
+  .pipe(z.date());
+
 // All IDs are friendly IDs
 const Env = z.object({
   // Set at build time
@@ -52,7 +57,8 @@ const Env = z.object({
   TRIGGER_METADATA_URL: z.string().optional(),
 
   // Timeline metrics
-  TRIGGER_POD_SCHEDULED_AT_MS: z.coerce.date(),
+  TRIGGER_POD_SCHEDULED_AT_MS: DateEnv,
+  TRIGGER_DEQUEUED_AT_MS: DateEnv,
 
   // May be overridden
   TRIGGER_SUPERVISOR_API_PROTOCOL: z.enum(["http", "https"]),
@@ -1295,7 +1301,7 @@ class ManagedRunController {
       this.startAndExecuteRunAttempt({
         runFriendlyId: env.TRIGGER_RUN_ID,
         snapshotFriendlyId: env.TRIGGER_SNAPSHOT_ID,
-        dequeuedAt: new Date(),
+        dequeuedAt: env.TRIGGER_DEQUEUED_AT_MS,
         podScheduledAt: env.TRIGGER_POD_SCHEDULED_AT_MS,
       }).finally(() => {});
       return;