Skip to content

Commit 99f3a88

Browse files
committed
Added error checking in case an error occurs
1 parent 93ab8ea commit 99f3a88

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

cpp/src/parquet/arrow/reader_writer_benchmark.cc

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -297,8 +297,9 @@ static void BenchmarkReadTable(::benchmark::State& state, const Table& table,
297297
auto reader =
298298
ParquetFileReader::Open(std::make_shared<::arrow::io::BufferReader>(buffer));
299299
std::unique_ptr<FileReader> arrow_reader;
300-
EXIT_NOT_OK(FileReader::Make(::arrow::default_memory_pool(), std::move(reader),
301-
&arrow_reader));
300+
auto reader_result =
301+
FileReader::Make(::arrow::default_memory_pool(), std::move(reader));
302+
EXIT_NOT_OK(result.status);
302303

303304
std::shared_ptr<Table> table;
304305
EXIT_NOT_OK(arrow_reader->ReadTable(&table));
@@ -736,8 +737,9 @@ static void BM_ReadIndividualRowGroups(::benchmark::State& state) {
736737
auto reader =
737738
ParquetFileReader::Open(std::make_shared<::arrow::io::BufferReader>(buffer));
738739
std::unique_ptr<FileReader> arrow_reader;
739-
EXIT_NOT_OK(FileReader::Make(::arrow::default_memory_pool(), std::move(reader),
740-
&arrow_reader));
740+
auto reader_result =
741+
FileReader::Make(::arrow::default_memory_pool(), std::move(reader));
742+
EXIT_NOT_OK(reader_result.status());
741743

742744
std::vector<std::shared_ptr<Table>> tables;
743745
for (int i = 0; i < arrow_reader->num_row_groups(); i++) {
@@ -771,8 +773,10 @@ static void BM_ReadMultipleRowGroups(::benchmark::State& state) {
771773
auto reader =
772774
ParquetFileReader::Open(std::make_shared<::arrow::io::BufferReader>(buffer));
773775
std::unique_ptr<FileReader> arrow_reader;
774-
EXIT_NOT_OK(FileReader::Make(::arrow::default_memory_pool(), std::move(reader),
775-
&arrow_reader));
776+
auto reader_result =
777+
FileReader::Make(::arrow::default_memory_pool(), std::move(reader));
778+
EXIT_NOT_OK(reader_result.status());
779+
776780
std::shared_ptr<Table> table;
777781
EXIT_NOT_OK(arrow_reader->ReadRowGroups(rgs, &table));
778782
}

0 commit comments

Comments
 (0)