Pass UnitLoaderFactory to NimbleReader

Summary: As said Differential Revision: D56739290
facebookincubator · May 1, 2024 · bafadc7 · bafadc7
1 parent 51953e6
commit bafadc7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dwio/nimble/velox/VeloxReader.h b/dwio/nimble/velox/VeloxReader.h
@@ -29,6 +29,7 @@
 #include "velox/dwio/common/ColumnSelector.h"
 #include "velox/dwio/common/ExecutorBarrier.h"
 #include "velox/dwio/common/FlatMapHelper.h"
+#include "velox/dwio/common/UnitLoader.h"
 #include "velox/type/Type.h"
 #include "velox/vector/BaseVector.h"
 
@@ -60,6 +61,8 @@ struct VeloxReadParams : public FieldReaderParams {
   // Report the Wall time (us) that we spend decoding.
   std::function<void(std::chrono::high_resolution_clock::duration)>
       decodingTimeCallback;
+
+  std::shared_ptr<velox::dwio::common::UnitLoaderFactory> unitLoaderFactory;
 };
 
 class VeloxReader {