Merge d34bd25 into d4eaec4

Author: alexvru

ydb/library/actors/interconnect/interconnect_tcp_proxy.cpp

@@ -942,4 +942,39 @@ namespace NActors {
         // TODO: unregister actor mon page
         TActor::PassAway();
     }
+
+    void TInterconnectProxyTCP::RegisterDisconnect() {
+        const TMonotonic now = TActivationContext::Monotonic();
+        ShiftDisconnectWindow(now);
+        ++NumDisconnectsInLastHour;
+        if (NumDisconnects.empty()) {
+            NumDisconnects.push_back(1);
+            FirstDisconnectWindowMinutes = now.Minutes();
+        } else {
+            ++NumDisconnects.back();
+        }
+    }
+
+    ui32 TInterconnectProxyTCP::GetDisconnectCountInLastHour() {
+        ShiftDisconnectWindow(TMonotonic::Now());
+        return NumDisconnectsInLastHour;
+    }
+
+    void TInterconnectProxyTCP::ShiftDisconnectWindow(TMonotonic now) {
+        if (!NumDisconnects.empty()) {
+            if (const ui32 steps = now.Minutes() - FirstDisconnectWindowMinutes; steps < NumDisconnects.size()) {
+                auto it = NumDisconnects.begin();
+                for (ui32 i = 0; i < steps; ++i, ++it) {
+                    NumDisconnectsInLastHour -= *it;
+                }
+                NumDisconnects.erase(NumDisconnects.begin(), it);
+                FirstDisconnectWindowMinutes += steps;
+            } else {
+                NumDisconnects.clear();
+                NumDisconnectsInLastHour = 0;
+                FirstDisconnectWindowMinutes = 0;
+            }
+        }
+    }
+
 }

ydb/library/actors/interconnect/interconnect_tcp_proxy.h

@@ -565,6 +565,17 @@ namespace NActors {
         void HandleTerminate();
 
         void PassAway() override;
+
+        ////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+        // Disconnection management
+
+        std::deque<ui32> NumDisconnects;
+        ui32 NumDisconnectsInLastHour = 0;
+        ui64 FirstDisconnectWindowMinutes = 0;
+
+        void RegisterDisconnect();
+        ui32 GetDisconnectCountInLastHour();
+        void ShiftDisconnectWindow(TMonotonic now);
     };
 
 }

ydb/library/actors/interconnect/interconnect_tcp_session.cpp

@@ -542,6 +542,7 @@ namespace NActors {
             CloseOnIdleWatchdog.Disarm();
             LostConnectionWatchdog.Rearm(SelfId());
             Proxy->Metrics->SetConnected(0);
+            Proxy->RegisterDisconnect();
             LOG_INFO(*TlsActivationContext, NActorsServices::INTERCONNECT_STATUS, "[%u] disconnected", Proxy->PeerNodeId);
         }
         if (XdcSocket) {