@databricks/sql
Version:
Driver for connection to Databricks SQL via Thrift API.
18 lines (17 loc) • 833 B
TypeScript
/// <reference types="node" />
import { TGetResultSetMetadataResp, TRowSet } from '../../thrift/TCLIService_types';
import IClientContext from '../contracts/IClientContext';
import IResultsProvider, { ResultsProviderFetchNextOptions } from './IResultsProvider';
import { ArrowBatch } from './utils';
export default class ArrowResultHandler implements IResultsProvider<ArrowBatch> {
private readonly context;
private readonly source;
private readonly arrowSchema?;
private readonly isLZ4Compressed;
constructor(context: IClientContext, source: IResultsProvider<TRowSet | undefined>, { schema, arrowSchema, lz4Compressed }: TGetResultSetMetadataResp);
hasMore(): Promise<boolean>;
fetchNext(options: ResultsProviderFetchNextOptions): Promise<{
batches: Buffer[];
rowCount: number;
}>;
}