From 2a9f60d57401bf9d6165e44af30c2f21ada79320 Mon Sep 17 00:00:00 2001
From: Codex <codex@noreply.local>
Date: Sun, 17 May 2026 00:47:02 +0000
Subject: [PATCH] fix: avoid stale task recovery on read pods

---
 src/components/microservices/code-queue/src/index.ts | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/components/microservices/code-queue/src/index.ts b/src/components/microservices/code-queue/src/index.ts
index cfe92641..d34b0789 100644
--- a/src/components/microservices/code-queue/src/index.ts
+++ b/src/components/microservices/code-queue/src/index.ts
@@ -2890,10 +2890,18 @@ async function sleepForRetryBackoff(task: QueueTask, delayMs: number): Promise<v
   }
 }
 
-function queueActiveTasksForRestartRetry(reason: string, method: string): number {
+function taskHasLocalExecutionRecoveryClaim(task: QueueTask): boolean {
+  if (task.status === "judging") return true;
+  if (activeRunForTask(task) !== null) return true;
+  return activeRunSlotReservations.has(queueIdOf(task));
+}
+
+function queueActiveTasksForRestartRetry(reason: string, method: string, options: { onlyActiveRuns?: boolean } = {}): number {
+  if (!config.schedulerEnabled || !serviceRoleAllowsScheduler(config.serviceRole)) return 0;
   let recovered = 0;
   for (const task of state.tasks) {
     if (task.status !== "running" && task.status !== "judging") continue;
+    if (options.onlyActiveRuns === true && !taskHasLocalExecutionRecoveryClaim(task)) continue;
     task.status = "retry_wait";
     task.finishedAt = null;
     task.readAt = null;
@@ -3443,7 +3451,7 @@ function installShutdownHandlers(): void {
   const stop = (signal: NodeJS.Signals): void => {
     if (shutdownRequested) process.exit(0);
     shutdownRequested = true;
-    const recovered = queueActiveTasksForRestartRetry("Service stopping while task was active", "shutdown");
+    const recovered = queueActiveTasksForRestartRetry("Service stopping while task was active", "shutdown", { onlyActiveRuns: true });
     for (const run of activeRuns.values()) run.app.stop();
     activeRuns.clear();
     activeRunSlotReservations.clear();