fix(runner): immediate poll to decrease restore time

Author: nicktrn

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

@@ -514,7 +514,7 @@ export class ManagedRunController {
           supervisorApiUrl: this.env.TRIGGER_SUPERVISOR_API_URL,
         };
 
-        await this.currentExecution.processEnvOverrides("socket disconnected");
+        await this.currentExecution.processEnvOverrides("socket disconnected", true);
 
         const newEnv = {
           workerInstanceName: this.env.TRIGGER_WORKER_INSTANCE_NAME,

packages/cli-v3/src/entryPoints/managed/execution.ts

@@ -896,7 +896,10 @@ export class RunExecution {
   /**
    * Processes env overrides from the metadata service. Generally called when we're resuming from a suspended state.
    */
-  public async processEnvOverrides(reason?: string): Promise<{ overrides: Metadata } | null> {
+  public async processEnvOverrides(
+    reason?: string,
+    shouldPollForSnapshotChanges?: boolean
+  ): Promise<{ overrides: Metadata } | null> {
     if (!this.metadataClient) {
       return null;
     }
@@ -943,6 +946,12 @@ export class RunExecution {
       this.httpClient.updateRunnerId(this.env.TRIGGER_RUNNER_ID);
     }
 
+    // Poll for snapshot changes immediately
+    if (shouldPollForSnapshotChanges) {
+      this.sendDebugLog("[override] polling for snapshot changes", { reason });
+      this.fetchAndProcessSnapshotChanges("restore").catch(() => {});
+    }
+
     return {
       overrides,
     };