From 857b7766985cb4d60ba3b5797d3a8cb43f6a53bb Mon Sep 17 00:00:00 2001
From: tv <tv@krebsco.de>
Date: Mon, 2 Nov 2026 16:42:25 +0100
Subject: add per-node traffic counts

---
 src/swarm/peer/worker.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'src/swarm/peer/worker.rs')

diff --git a/src/swarm/peer/worker.rs b/src/swarm/peer/worker.rs
index e8e6f8b..6196571 100644
--- a/src/swarm/peer/worker.rs
+++ b/src/swarm/peer/worker.rs
@@ -170,7 +170,15 @@ pub(crate) async fn run_peer_worker(state: AppState) {
             crate::swarm::state_file::update(&state, false).await;
             {
                 let known = state.peer_states.read().await;
-                state.traffic.retain(|peer| known.contains_key(peer));
+                // And their addresses, where the layers below the node count
+                // (`Traffic::node`).
+                let addresses: std::collections::HashSet<String> = known
+                    .keys()
+                    .flat_map(|peer| state.transport.identities_of(peer))
+                    .collect();
+                state
+                    .traffic
+                    .retain(|peer| known.contains_key(peer) || addresses.contains(peer));
                 state
                     .throughput
                     .lock_or_recover()
-- 
cgit v1.3.1