Function cancelQuery

console/src/api/materialize/cancelQuery.ts:36–70

({
  params,
  getConnectionIdFromSessionIdQueryKey,
  cancelQueryQueryKey,
}: {
  params: CancelQueryParams;
  getConnectionIdFromSessionIdQueryKey: QueryKey;
  cancelQueryQueryKey: QueryKey;
})

  | { sessionId: string };

export async function cancelQuery({
  params,
  getConnectionIdFromSessionIdQueryKey,
  cancelQueryQueryKey,
}: {
  params: CancelQueryParams;
  getConnectionIdFromSessionIdQueryKey: QueryKey;
  cancelQueryQueryKey: QueryKey;
}) {
  let connectionId;
  if ("sessionId" in params) {
    const { sessionId } = params;

    const { rows } = await executeSqlV2({
      queries: buildGetConnectionIdFromSessionIdQuery(sessionId).compile(),
      queryKey: getConnectionIdFromSessionIdQueryKey,
    });

    assertAtLeastOneRow(rows.length);

    connectionId = rows[0].connection_id;
  } else {
    connectionId = params.connectionId;
  }

  const { rows } = await executeSqlV2({
    queries: buildCancelQuery(connectionId).compile(queryBuilder),
    queryKey: cancelQueryQueryKey,
  });
  assertAtLeastOneRow(rows.length);

  if (!rows[0].pg_cancel_backend) {
    throw new Error("Cancel not successful");
  }
}