add throughput stats

1 files changed, 5 insertions, 0 deletions

diff --git a/src/swarm/peer.rs b/src/swarm/peer.rs
index 7a3eed8..3b85539 100644
--- a/src/swarm/peer.rs
+++ b/src/swarm/peer.rs
@@ -496,6 +496,11 @@ pub(crate) async fn run_peer_worker(state: AppState) {
             {
                 let known = state.peer_states.read().await;
                 state.traffic.retain(|peer| known.contains_key(peer));
+                state
+                    .throughput
+                    .lock()
+                    .unwrap_or_else(|poisoned| poisoned.into_inner())
+                    .retain(|peer| known.contains_key(peer));
             }
             if let Some(change) = topology::refresh(&state).await {
                 apply_neighbour_change(&state, change).await;