mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-01 06:00:49 +00:00
Fix arrow stream termination
This commit is contained in:
parent
4effdeaa0e
commit
ab37791134
@ -39,12 +39,12 @@ Chunk ArrowBlockInputFormat::generate()
|
||||
if (stream)
|
||||
{
|
||||
read_status = stream_reader->ReadNext(&single_batch[0]);
|
||||
if (!single_batch[0])
|
||||
return res;
|
||||
}
|
||||
else
|
||||
{
|
||||
read_status = file_reader->ReadRecordBatch(record_batch_current, &single_batch[0]);
|
||||
if (!single_batch[0])
|
||||
return res;
|
||||
}
|
||||
if (!read_status.ok())
|
||||
throw Exception{"Error while reading batch of Arrow data: " + read_status.ToString(),
|
||||
|
Loading…
Reference in New Issue
Block a user