Function pollQueryReady

packages/query-sdk/src/apiTransport.ts:559–597

(
    fetchFn: FetchAdapter,
    projectUuid: string,
    queryUuid: string,
    pageSize: number = PAGE_SIZE,
)

}

async function pollQueryReady(
    fetchFn: FetchAdapter,
    projectUuid: string,
    queryUuid: string,
    pageSize: number = PAGE_SIZE,
): Promise<Extract<PollResponse, { status: 'ready' }>> {
    const pollUrl = (page: number) =>
        `/api/v2/projects/${projectUuid}/query/${queryUuid}?page=${page}&pageSize=${pageSize}`;

    // The poll endpoint returns status PENDING/QUEUED/EXECUTING while
    // the warehouse is still running, and READY once results are available.
    const deadline = Date.now() + MAX_POLL_DURATION_MS;
    let backoffMs = INITIAL_BACKOFF_MS;

    while (Date.now() < deadline) {
        const pollResult = await fetchFn<PollResponse>('GET', pollUrl(1));

        if (pollResult.status === 'ready') {
            return pollResult;
        }

        if (pollResult.status === 'error' || pollResult.status === 'expired') {
            throw new Error(
                `Query failed: ${pollResult.error ?? 'unknown error'}`,
            );
        }

        if (pollResult.status === 'cancelled') {
            throw new Error('Query was cancelled');
        }

        // Still running — wait with exponential backoff and retry.
        const sleepMs = backoffMs;
        await sleep(sleepMs);
        backoffMs = Math.min(backoffMs * 2, MAX_BACKOFF_MS);
    }

    throw new Error('Query timed out waiting for results');
}

async function pollQueryRows(
    fetchFn: FetchAdapter,