Intermediate changes

1 files changed, 14 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/Storages/MergeTree/ReplicatedMergeTreeAttachThread.cpp b/contrib/clickhouse/src/Storages/MergeTree/ReplicatedMergeTreeAttachThread.cpp
index 65cf8bbce7..927df3c50e 100644
--- a/contrib/clickhouse/src/Storages/MergeTree/ReplicatedMergeTreeAttachThread.cpp
+++ b/contrib/clickhouse/src/Storages/MergeTree/ReplicatedMergeTreeAttachThread.cpp
@@ -2,6 +2,11 @@
 #include <Storages/StorageReplicatedMergeTree.h>
 #include <Common/ZooKeeper/IKeeper.h>
 
+namespace CurrentMetrics
+{
+    extern const Metric ReadonlyReplica;
+}
+
 namespace DB
 {
 
@@ -67,6 +72,9 @@ void ReplicatedMergeTreeAttachThread::run()
             LOG_ERROR(log, "Initialization failed, table will remain readonly. Error: {}", getCurrentExceptionMessage(/* with_stacktrace */ true));
             storage.initialization_done = true;
         }
+
+        if (!std::exchange(storage.is_readonly_metric_set, true))
+            CurrentMetrics::add(CurrentMetrics::ReadonlyReplica);
     }
 
     if (!first_try_done.exchange(true))
@@ -74,6 +82,12 @@ void ReplicatedMergeTreeAttachThread::run()
 
     if (shutdown_called)
     {
+        if (std::exchange(storage.is_readonly_metric_set, false))
+        {
+            CurrentMetrics::sub(CurrentMetrics::ReadonlyReplica);
+            chassert(CurrentMetrics::get(CurrentMetrics::ReadonlyReplica) >= 0);
+        }
+
         LOG_WARNING(log, "Shutdown called, cancelling initialization");
         return;
     }