fix potential GetStatistics hanging during rolling update (#9942)

Author: alexd65536

ydb/core/statistics/service/service_impl.cpp

@@ -664,7 +664,7 @@ class TStatService : public TActorBootstrapped<TStatService> {
         request.StatType = ev->Get()->StatType;
         request.StatRequests.swap(ev->Get()->StatRequests);
 
-        if (!EnableStatistics) {
+        if (!EnableStatistics || IsStatisticsDisabledInSA) {
             ReplyFailed(requestId, true);
             return;
         }
@@ -842,6 +842,8 @@ class TStatService : public TActorBootstrapped<TStatService> {
 
         Send(ev->Sender, new TEvStatistics::TEvPropagateStatisticsResponse);
 
+        IsStatisticsDisabledInSA = false;
+
         auto* record = ev->Get()->MutableRecord();
         for (const auto& entry : record->GetEntries()) {
             ui64 schemeShardId = entry.GetSchemeShardId();
@@ -994,6 +996,7 @@ class TStatService : public TActorBootstrapped<TStatService> {
             << ", status = " << ev->Get()->Status);
 
         if (clientId == SAPipeClientId) {
+            IsStatisticsDisabledInSA = false;
             if (ev->Get()->Status != NKikimrProto::OK) {
                 SAPipeClientId = TActorId();
                 ConnectToSA();
@@ -1030,6 +1033,7 @@ class TStatService : public TActorBootstrapped<TStatService> {
             << ", tablet id = " << tabletId);
 
         if (clientId == SAPipeClientId) {
+            IsStatisticsDisabledInSA = false;
             SAPipeClientId = TActorId();
             ConnectToSA();
             SyncNode();
@@ -1049,6 +1053,7 @@ class TStatService : public TActorBootstrapped<TStatService> {
     }
 
     void Handle(TEvStatistics::TEvStatisticsIsDisabled::TPtr&) {
+        IsStatisticsDisabledInSA = true;
         ReplyAllFailed();
     }
 
@@ -1527,6 +1532,7 @@ class TStatService : public TActorBootstrapped<TStatService> {
 
     bool EnableStatistics = false;
     bool EnableColumnStatistics = false;
+    bool IsStatisticsDisabledInSA = false;
 
     static constexpr size_t StatFanOut = 10;