diff --git a/cpp/velox/operators/writer/VeloxParquetDataSourceABFS.h b/cpp/velox/operators/writer/VeloxParquetDataSourceABFS.h index 6dd7027cea9..7b8b6d2cabf 100644 --- a/cpp/velox/operators/writer/VeloxParquetDataSourceABFS.h +++ b/cpp/velox/operators/writer/VeloxParquetDataSourceABFS.h @@ -27,6 +27,7 @@ #include "compute/VeloxRuntime.h" #include "velox/common/compression/Compression.h" +#include "velox/connectors/hive/storage_adapters/abfs/RegisterAbfsFileSystem.h" #include "velox/core/QueryConfig.h" #include "velox/core/QueryCtx.h" #include "velox/dwio/common/Options.h" @@ -46,6 +47,7 @@ class VeloxParquetDataSourceABFS final : public VeloxParquetDataSource { auto hiveConf = getHiveConfig( std::make_shared(std::unordered_map(sparkConfs)), FileSystemType::kAbfs); + facebook::velox::filesystems::registerAzureClientProvider(*hiveConf); auto fileSystem = filesystems::getFileSystem(filePath_, hiveConf); auto* abfsFileSystem = dynamic_cast(fileSystem.get()); sink_ = std::make_unique(