Function claimNextQueuedTaskForRuntimeSync

packages/db/src/task-queue.ts:245–337

(runtimeId: string, workspaceId?: string)

}

export function claimNextQueuedTaskForRuntimeSync(runtimeId: string, workspaceId?: string): QueuedTaskRecord | null {
  const db = getDatabase();
  const now = new Date().toISOString();
  let claimedId: string | null = null;
  let fallbackReason: string | undefined;

  db.exec("BEGIN");
  try {
    let row = selectQueuedTaskForRuntime(db, runtimeId, workspaceId);
    if (!row) {
      const fallback = selectFallbackQueuedTaskForRuntime(db, runtimeId, workspaceId);
      if (fallback) {
        row = fallback.row;
        fallbackReason = fallback.reason;
        db.prepare(
          `UPDATE agent_task_queue
           SET runtime_id = ?,
               updated_at = ?
           WHERE id = ? AND status = 'queued'`,
        ).run(runtimeId, now, row.id);
      }
    }

    if (row && typeof row.id === "string") {
      db.prepare(
        `UPDATE agent_task_queue
         SET status = 'claimed',
             claimed_at = ?,
             updated_at = ?
         WHERE id = ? AND status = 'queued'`,
      ).run(now, now, row.id);
      claimedId = row.id;
    }
    db.exec("COMMIT");
  } catch (error) {
    db.exec("ROLLBACK");
    throw error;
  }

  const task = claimedId ? readQueuedTaskSync(claimedId) : null;
  if (task) {
    const runtime = readAgentRuntimeSync(task.runtimeId);
    const providerSession = chooseProviderSessionForTaskSync({ task });
    const attempt = runtime && task.routerSessionId
      ? createAgentTaskAttemptSync({
          workspaceId: task.workspaceId,
          taskQueueId: task.id,
          routerSessionId: task.routerSessionId,
          runtimeId: task.runtimeId,
          provider: runtime.provider,
          providerSessionId: providerSession?.providerSessionId,
          status: "claimed",
          metadata: {
            routingMode: providerSession ? "same_provider_resume" : fallbackReason ? "cold_rebuild_fallback" : "cold_rebuild",
            fallbackReason,
            previousRuntimeId: fallbackReason ? readStringFromTaskInput(task.inputJson, "__previousRuntimeId") : undefined,
          },
        })