Skip to content

Commit 6b16362

Browse files
authored
add getcolumn to parquetdataframe (#266)
1 parent bde9155 commit 6b16362

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

src/lib/tableProvider.ts

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,19 @@ export function parquetDataFrame(from: AsyncBufferFrom, metadata: FileMetaData):
188188
}
189189
},
190190
sortable: true,
191-
// TODO(SL): implement getColumn({column, start, end}): any[]
191+
getColumn({ column, start, end }) {
192+
if (!header.includes(column)) {
193+
return Promise.reject(new Error(`Column "${column}" not found in header`))
194+
}
195+
196+
return parquetQueryWorker({
197+
from,
198+
metadata,
199+
rowStart: start,
200+
rowEnd: end,
201+
}).then(rows => {
202+
return rows.map(row => row[column])
203+
})
204+
},
192205
}
193206
}

0 commit comments

Comments
 (0)