From a4daf2b454c44e1891a61eaddf3a2fd965e5f880 Mon Sep 17 00:00:00 2001
From: kssenii <sumarokovakseniia@mail.ru>
Date: Sat, 20 Apr 2024 14:46:32 +0100
Subject: [PATCH] Fix hdfs race

---
 src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.cpp | 7 ++++++-
 src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.h   | 4 ++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.cpp b/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.cpp
index fc7d49324c7..ed63795cb05 100644
--- a/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.cpp
+++ b/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.cpp
@@ -33,11 +33,16 @@ void HDFSObjectStorage::startup()
 
 void HDFSObjectStorage::initializeHDFS() const
 {
-    if (hdfs_fs)
+    if (initialized)
+        return;
+
+    std::lock_guard lock(init_mutex);
+    if (initialized)
         return;
 
     hdfs_builder = createHDFSBuilder(url, config);
     hdfs_fs = createHDFSFS(hdfs_builder.get());
+    initialized = true;
 }
 
 ObjectStorageKey HDFSObjectStorage::generateObjectKeyForPath(const std::string & /* path */) const
diff --git a/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.h b/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.h
index f57b7e1fda8..b626d3dc779 100644
--- a/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.h
+++ b/src/Disks/ObjectStorages/HDFS/HDFSObjectStorage.h
@@ -121,6 +121,10 @@ private:
 
     mutable HDFSBuilderWrapper hdfs_builder;
     mutable HDFSFSPtr hdfs_fs;
+
+    mutable std::mutex init_mutex;
+    mutable std::atomic_bool initialized{false};
+
     SettingsPtr settings;
     std::string url;
     std::string url_without_path;