diff --git a/cpp/velox/operators/writer/VeloxParquetDataSource.cc b/cpp/velox/operators/writer/VeloxParquetDataSource.cc index cd5063a61d4a..5e9f4251368e 100644 --- a/cpp/velox/operators/writer/VeloxParquetDataSource.cc +++ b/cpp/velox/operators/writer/VeloxParquetDataSource.cc @@ -36,7 +36,7 @@ namespace gluten { void VeloxParquetDataSource::initSink(const std::unordered_map& /* sparkConfs */) { if (strncmp(filePath_.c_str(), "file:", 5) == 0) { - sink_ = dwio::common::FileSink::create(filePath_, {.pool = pool_.get()}); + sink_ = dwio::common::FileSink::create(filePath_, {.pool = sinkPool_.get()}); } else { throw std::runtime_error("The file path is not local when writing data with parquet format in velox runtime!"); } diff --git a/cpp/velox/operators/writer/VeloxParquetDataSource.h b/cpp/velox/operators/writer/VeloxParquetDataSource.h index 4dde5c69aa85..dac97168d5a0 100644 --- a/cpp/velox/operators/writer/VeloxParquetDataSource.h +++ b/cpp/velox/operators/writer/VeloxParquetDataSource.h @@ -84,7 +84,11 @@ class VeloxParquetDataSource : public VeloxDataSource { std::shared_ptr veloxPool, std::shared_ptr sinkPool, std::shared_ptr schema) - : VeloxDataSource(filePath, schema), filePath_(filePath), schema_(schema), pool_(std::move(veloxPool)) {} + : VeloxDataSource(filePath, schema), + filePath_(filePath), + sinkPool_(std::move(sinkPool)), + schema_(schema), + pool_(std::move(veloxPool)) {} void init(const std::unordered_map& sparkConfs) override; virtual void initSink(const std::unordered_map& sparkConfs);