delayCollect->clean

Socks5Balancer · Sep 1, 2023 · 699e2da · 699e2da
1 parent 960e6dc
commit 699e2da
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/DelayCollection.h b/src/DelayCollection.h
@@ -135,6 +135,11 @@ namespace DelayCollection {
                 return di.timeClock < time;
             });
         }
+
+        void clean() {
+            std::lock_guard lg{mtx};
+            q.clear();
+        }
     };
 
     class DelayCollect : public std::enable_shared_from_this<DelayCollect> {
@@ -204,6 +209,21 @@ namespace DelayCollection {
             historyRelayFirstDelay.removeBefore(time);
         }
 
+        void cleanTcpPing() {
+            historyTcpPing.clean();
+            lastTcpPing = TimeMsInvalid;
+        }
+
+        void cleanHttpPing() {
+            historyHttpPing.clean();
+            lastHttpPing = TimeMsInvalid;
+        }
+
+        void cleanFirstDelay() {
+            historyRelayFirstDelay.clean();
+            lastRelayFirstDelay = TimeMsInvalid;
+        }
+
     public:
         void pushTcpPing(TimeMs t) {
             lastTcpPing = t;

diff --git a/src/StateMonitorServer.cpp b/src/StateMonitorServer.cpp
@@ -473,6 +473,9 @@ void HttpConnectSession::path_op(HttpConnectSession::QueryPairsType &queryPairs)
                         a->lastOnlinePing = std::chrono::milliseconds{-1};
                         a->lastConnectTime = UpstreamTimePoint{};
                         a->lastConnectPing = std::chrono::milliseconds{-1};
+                        a->delayCollect->cleanTcpPing();
+                        a->delayCollect->cleanHttpPing();
+                        a->delayCollect->cleanFirstDelay();
                     }
                     // recheck
                     upstreamPool->forceCheckNow();