From f9df0cb1305c08fc76144874b458356aa3e32e49 Mon Sep 17 00:00:00 2001
From: Daniel Munoz <munozdaniel@meta.com>
Date: Wed, 1 May 2024 10:43:57 -0700
Subject: [PATCH] Pass UnitLoaderFactory to NimbleReader

Summary: As said

Reviewed By: helfman

Differential Revision: D56739290
---
 dwio/nimble/velox/VeloxReader.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/dwio/nimble/velox/VeloxReader.h b/dwio/nimble/velox/VeloxReader.h
index 474696a..5b1b972 100644
--- 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,10 @@ struct VeloxReadParams : public FieldReaderParams {
   // Report the Wall time (us) that we spend decoding.
   std::function<void(std::chrono::high_resolution_clock::duration)>
       decodingTimeCallback;
+
+  // Factory with the algorithm to load stripes (units).
+  // If nullptr we'll use the default one, that doesn't pre-load stripes.
+  std::shared_ptr<velox::dwio::common::UnitLoaderFactory> unitLoaderFactory;
 };
 
 class VeloxReader {