add StarvingInRowForNotEnoughCpu for tcp sessions (#625)

Co-authored-by: Aleksandr Kriukov <kruall@ydb.ru>
ydb-platform · Dec 22, 2023 · c289f6b · c289f6b
1 parent 1459018
commit c289f6b
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/ydb/library/actors/interconnect/interconnect_tcp_input_session.cpp b/ydb/library/actors/interconnect/interconnect_tcp_input_session.cpp
@@ -260,7 +260,12 @@ namespace NActors {
             }
         }
 
-        SetEnoughCpu(enoughCpu);
+        if (enoughCpu) {
+            SetEnoughCpu(true);
+            StarvingInRow = 0;
+        } else {
+            SetEnoughCpu(++StarvingInRow < StarvingInRowForNotEnoughCpu);
+        }
 
         // calculate ping time
         auto it = std::min_element(PingQ.begin(), PingQ.end());

diff --git a/ydb/library/actors/interconnect/interconnect_tcp_session.cpp b/ydb/library/actors/interconnect/interconnect_tcp_session.cpp
@@ -402,9 +402,10 @@ namespace NActors {
 
             if (!canProducePackets && !canWriteData) {
                 SetEnoughCpu(true); // we do not starve
+                StarvingInRow = 0;
                 break;
             } else if (TimeLimit->CheckExceeded()) {
-                SetEnoughCpu(false);
+                SetEnoughCpu(++StarvingInRow < StarvingInRowForNotEnoughCpu);
                 IssueRam(false);
                 break;
             }

diff --git a/ydb/library/actors/interconnect/interconnect_tcp_session.h b/ydb/library/actors/interconnect/interconnect_tcp_session.h
@@ -33,6 +33,9 @@
 #include <unordered_map>
 
 namespace NActors {
+
+    static constexpr ui64 StarvingInRowForNotEnoughCpu = 32;
+
     class TSlowPathChecker {
         using TTraceCallback = std::function<void(double)>;
         TTraceCallback Callback;
@@ -299,6 +302,8 @@ namespace NActors {
         std::array<ui32, 16> InputTrafficArray;
         THashMap<ui16, ui32> InputTrafficMap;
 
+        ui64 StarvingInRow = 0;
+
         bool CloseInputSessionRequested = false;
 
         void CloseInputSession();
@@ -634,6 +639,8 @@ namespace NActors {
         bool StartHandshakeOnSessionClose = false;
 
         ui64 EqualizeCounter = 0;
+
+        ui64 StarvingInRow = 0;
     };
 
     class TInterconnectSessionKiller