habsi/elasticsearch-js
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update ESQL helper types (#2738)

Browse Source
This commit is contained in:
Josh Mock
2025-04-14 12:44:39 -05:00
committed by GitHub
parent 1ab089022e
commit 931a80cacb
1 changed files with 6 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/helpers.ts
View File

@ -132,15 +132,6 @@ export interface EsqlColumn {
type: string
}
export type EsqlValue = any[]
export type EsqlRow = EsqlValue[]
export interface EsqlResponse {
columns: EsqlColumn[]
values: EsqlRow[]
}
export interface EsqlHelper {
toRecords: <TDocument>() => Promise<EsqlToRecords<TDocument>>
toArrowTable: () => Promise<Table<TypeMap>>
@ -963,7 +954,7 @@ export default class Helpers {
esql (params: T.EsqlQueryRequest, reqOptions: TransportRequestOptions = {}): EsqlHelper {
const client = this[kClient]
function toRecords<TDocument> (response: EsqlResponse): TDocument[] {
function toRecords<TDocument> (response: T.EsqlEsqlResult): TDocument[] {
const { columns, values } = response
return values.map(row => {
const doc: Partial<TDocument> = {}
@ -990,8 +981,7 @@ export default class Helpers {
params.format = 'json'
params.columnar = false
// @ts-expect-error it's typed as ArrayBuffer but we know it will be JSON
const response: EsqlResponse = await client.esql.query(params, reqOptions)
const response = await client.esql.query(params, reqOptions)
const records: TDocument[] = toRecords(response)
const { columns } = response
return { records, columns }
@ -1005,7 +995,8 @@ export default class Helpers {
params.format = 'arrow'
const response = await client.esql.query(params, reqOptions)
// @ts-expect-error the return type will be ArrayBuffer when the format is set to 'arrow'
const response: ArrayBuffer = await client.esql.query(params, reqOptions)
return tableFromIPC(response)
},
@ -1018,7 +1009,8 @@ export default class Helpers {
params.format = 'arrow'
const response = await client.esql.query(params, reqOptions)
// @ts-expect-error the return type will be ArrayBuffer when the format is set to 'arrow'
const response: ArrayBuffer = await client.esql.query(params, reqOptions)
return RecordBatchStreamReader.from(response)
}
}