rtt table pruned

1 files changed, 8 insertions, 0 deletions

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) {