Function selectFallbackQueuedTaskForRuntime

packages/db/src/task-queue.ts:852–914

(
  db: ReturnType<typeof getDatabase>,
  runtimeId: string,
  workspaceId?: string,
)

}

function selectFallbackQueuedTaskForRuntime(
  db: ReturnType<typeof getDatabase>,
  runtimeId: string,
  workspaceId?: string,
): { row: Record<string, unknown>; reason: string } | null {
  const runtime = readAgentRuntimeSync(runtimeId);
  if (!runtime || runtime.status !== "online") {
    return null;
  }
  const rows = db.prepare(
    `SELECT
       q.id,
       q.runtime_id AS runtimeId,
       q.workspace_id AS workspaceId,
       q.requested_by_user_id AS requestedByUserId,
       r.status AS selectedRuntimeStatus,
       r.provider AS selectedProvider
     FROM agent_task_queue q
     JOIN agent_runtime r ON r.id = q.runtime_id
     WHERE q.status = 'queued'
       AND q.runtime_id <> ?
       ${typeof workspaceId === "string" ? "AND q.workspace_id = ?" : ""}
     ORDER BY q.priority DESC, q.created_at ASC
     LIMIT 20`,
  ).all(...(typeof workspaceId === "string" ? [runtimeId, workspaceId] : [runtimeId])) as Array<Record<string, unknown>>;

  for (const row of rows) {
    if (row.workspaceId !== runtime.workspaceId) {
      continue;
    }
    if (row.selectedRuntimeStatus === "online") {
      continue;
    }
    const requesterUserId = typeof row.requestedByUserId === "string" ? row.requestedByUserId : undefined;
    if (requesterUserId && !canUserUseRuntimeSync(runtime.workspaceId, runtime.id, requesterUserId)) {
      continue;
    }
    const selectedProvider = typeof row.selectedProvider === "string" ? row.selectedProvider : undefined;
    if (
      selectedProvider &&
      runtime.provider !== selectedProvider &&
      existsUsableOnlineRuntimeForProvider({
        db,
        workspaceId: runtime.workspaceId,
        provider: selectedProvider,
        requesterUserId,
      })
    ) {
      continue;
    }
    return {
      row,
      reason:
        selectedProvider && runtime.provider !== selectedProvider
          ? `preferred runtime ${String(row.runtimeId)} is offline and no usable ${selectedProvider} runtime is online`
          : requesterUserId
            ? `preferred runtime ${String(row.runtimeId)} is offline and requester can use ${runtime.id}`
            : `preferred runtime ${String(row.runtimeId)} is offline`,