fix TPCH benchmark reader crash

parquet and dwrf reader are registered in TPCH benchmark initialize
facebookincubator · Dec 11, 2024 · 19d33e6 · 19d33e6
1 parent b44ffc9
commit 19d33e6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/velox/benchmarks/QueryBenchmarkBase.cpp b/velox/benchmarks/QueryBenchmarkBase.cpp
@@ -193,6 +193,8 @@ void QueryBenchmarkBase::initialize() {
           connector::hive::HiveConnectorFactory::kHiveConnectorName)
           ->newConnector(kHiveConnectorId, properties, ioExecutor_.get());
   connector::registerConnector(hiveConnector);
+  parquet::registerParquetReaderFactory();
+  dwrf::registerDwrfReaderFactory();
 }
 
 std::vector<std::shared_ptr<connector::ConnectorSplit>>

diff --git a/velox/benchmarks/QueryBenchmarkBase.h b/velox/benchmarks/QueryBenchmarkBase.h
@@ -33,6 +33,8 @@
 #include "velox/connectors/hive/HiveConfig.h"
 #include "velox/connectors/hive/HiveConnector.h"
 #include "velox/dwio/common/Options.h"
+#include "velox/dwio/dwrf/RegisterDwrfReader.h"
+#include "velox/dwio/parquet/RegisterParquetReader.h"
 #include "velox/exec/PlanNodeStats.h"
 #include "velox/exec/Split.h"
 #include "velox/exec/tests/utils/HiveConnectorTestBase.h"