rtt table pruned

2 files changed, 9 insertions, 0 deletions

diff --git a/src/swarm/peer/worker.rs b/src/swarm/peer/worker.rs
index b5c533a..7a0b502 100644
--- a/src/swarm/peer/worker.rs
+++ b/src/swarm/peer/worker.rs
@@ -201,6 +201,7 @@ pub(crate) async fn run_peer_worker(state: AppState) {
                     .throughput
                     .lock_or_recover()
                     .retain(|peer| known.contains_key(peer));
+                state.lock_rtt().retain(|peer| known.contains_key(peer));
             }
             // Fallback for what doesn't notify (`topology::run_worker`).
             state.topology_changed.notify_one();
diff --git a/src/swarm/rtt.rs b/src/swarm/rtt.rs
index 370754c..6dfe84a 100644
--- a/src/swarm/rtt.rs
+++ b/src/swarm/rtt.rs
@@ -39,6 +39,14 @@ struct Outstanding {
 }
 
 impl RttTable {
+    /// Keeps only the peers `keep` says (those still known: else every peer ever
+    /// greeted or measured would stay).
+    pub(crate) fn retain(&mut self, mut keep: impl FnMut(&str) -> bool) {
+        self.hellos.retain(|peer, _| keep(peer));
+        self.from_paths.retain(|peer, _| keep(peer));
+        self.from_hellos.retain(|peer, _| keep(peer));
+    }
+
     /// The paths layer's round-trip time to `peer` (`Transport::rtt`), as a PONG from
     /// it arrived.
     pub(crate) fn measured(&mut self, peer: &str, rtt: Duration, now: Instant) {