refactor: simplify iterating over Arrow's batches (#437)

* refactor: simplify iterating over Arrow's batches * docs: update CHANGELOG.md
InfluxCommunity · Oct 14, 2024 · d2b1d0f · d2b1d0f
1 parent 4d369bd
commit d2b1d0f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 ## 0.12.0 [unreleased]
 
+### Bugfix
+
+1. [437](https://github.com/InfluxCommunity/influxdb3-js/pull/437): Simplify iterating over Arrow's batches in `QueryAPI`
+
 ## 0.11.0 [2024-09-13]
 
 ### Features

diff --git a/packages/client/src/impl/QueryApiImpl.ts b/packages/client/src/impl/QueryApiImpl.ts
@@ -114,14 +114,11 @@ export default class QueryApiImpl implements QueryApi {
  const batches = this._queryRawBatches(query, database, options)
 
  for await (const batch of batches) {
- for (let rowIndex = 0; rowIndex < batch.numRows; rowIndex++) {
- const row: Record<string, any> = {}
- for (let columnIndex = 0; columnIndex < batch.numCols; columnIndex++) {
- const name = batch.schema.fields[columnIndex].name
- const value = batch.getChildAt(columnIndex)?.get(rowIndex)
- row[name] = value
+ const row: Record<string, any> = {}
+ for (const batchRow of batch) {
+ for (const column of batch.schema.fields) {
+ row[column.name] = batchRow[column.name]
  }
-
  yield row
  }
  }