Skip to content

Commit fba7d2a

Browse files
committed
oops, still need to check for valid schema!
1 parent 73eaa85 commit fba7d2a

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

src/main/capi/arrow-c.cpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -432,6 +432,16 @@ duckdb_state Ingest(duckdb_connection connection, const char *table_name, struct
432432

433433
duckdb_state duckdb_arrow_scan(duckdb_connection connection, const char *table_name, duckdb_arrow_stream arrow) {
434434
auto stream = reinterpret_cast<ArrowArrayStream *>(arrow);
435+
436+
ArrowSchema schema;
437+
if (stream->get_schema(stream, &schema) == DuckDBError) {
438+
return DuckDBError;
439+
}
440+
441+
if (schema.release) {
442+
schema.release(&schema);
443+
}
444+
435445
return arrow_array_stream_wrapper::Ingest(connection, table_name, stream);
436446
}
437447

0 commit comments

Comments
 (0)