Fix parquet directory have empty file (#1593)
This commit is contained in:
@ -55,6 +55,9 @@ Status ParquetReaderWrap::init_parquet_reader(const std::vector<SlotDescriptor*>
|
||||
_file_metadata = _reader->parquet_reader()->metadata();
|
||||
// initial members
|
||||
_total_groups = _file_metadata->num_row_groups();
|
||||
if (_total_groups == 0) {
|
||||
return Status::EndOfFile("Empty Parquet File");
|
||||
}
|
||||
_rows_of_group = _file_metadata->RowGroup(0)->num_rows();
|
||||
|
||||
// map
|
||||
|
||||
@ -142,7 +142,12 @@ Status ParquetScanner::open_next_reader() {
|
||||
continue;
|
||||
}
|
||||
_cur_file_reader = new ParquetReaderWrap(file_reader.release());
|
||||
return _cur_file_reader->init_parquet_reader(_src_slot_descs);
|
||||
Status status = _cur_file_reader->init_parquet_reader(_src_slot_descs);
|
||||
if (status.is_end_of_file()) {
|
||||
continue;
|
||||
} else {
|
||||
return status;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user