add traffic counter

This introduces a new wrapper that counts messages and payload bytes per
peer, split by kind (hello, helloReply, sync, ping, pong), each sent,
received and refused.

The traffic counter goes around the firewall, which means messages
dropped outbound are still counted as sent, and messages dropped inbound
don't count as received.

1 files changed, 2 insertions, 2 deletions

diff --git a/src/transport/firewall.rs b/src/transport/firewall.rs
index 84fdc59..b8e05cd 100644
--- a/src/transport/firewall.rs
+++ b/src/transport/firewall.rs
@@ -89,12 +89,12 @@ impl Transport for FirewallTransport {
         let firewall = Arc::clone(&self.firewall);
 
         self.inner
-            .start_consumer(Box::new(move |sender, message| {
+            .start_consumer(Box::new(move |sender, message, payload_len| {
                 if firewall.blocks_inbound(&sender) {
                     tracing::trace!(peer = %sender, "Firewall dropped inbound message");
                     return Box::pin(async { Ok(()) });
                 }
-                handler(sender, message)
+                handler(sender, message, payload_len)
             }))
             .await
     }