From c02a22d6282fb43ac2efb913fec103de10b6dad6 Mon Sep 17 00:00:00 2001
From: Yuan <yuanzhou@apache.org>
Date: Wed, 24 Sep 2025 11:42:23 +0100
Subject: [PATCH] [GLUTEN-10794][VL] Register AzureClientProviderFactory on
 ABFS write

Signed-off-by: Yuan <yuanzhou@apache.org>
---
 cpp/velox/operators/writer/VeloxParquetDataSourceABFS.h | 2 ++
 1 file changed, 2 insertions(+)

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<facebook::velox::config::ConfigBase>(std::unordered_map<std::string, std::string>(sparkConfs)),
         FileSystemType::kAbfs);
+    facebook::velox::filesystems::registerAzureClientProvider(*hiveConf);
     auto fileSystem = filesystems::getFileSystem(filePath_, hiveConf);
     auto* abfsFileSystem = dynamic_cast<filesystems::AbfsFileSystem*>(fileSystem.get());
     sink_ = std::make_unique<dwio::common::WriteFileSink>(