add identity layer

1 files changed, 7 insertions, 3 deletions

diff --git a/src/transport/traffic.rs b/src/transport/traffic.rs
index c27f0d7..59acaf2 100644
--- a/src/transport/traffic.rs
+++ b/src/transport/traffic.rs
@@ -40,6 +40,9 @@ pub(crate) struct ByKind {
     pub(crate) ping: Counter,
     pub(crate) pong: Counter,
     pub(crate) blob: Counter,
+    /// WHO and ME (`transport::identity`), counted by the identity layer under the
+    /// address they went to or came from.
+    pub(crate) identity: Counter,
 }
 
 impl ByKind {
@@ -51,6 +54,7 @@ impl ByKind {
             MessageKind::Ping => &mut self.ping,
             MessageKind::Pong => &mut self.pong,
             MessageKind::Blob => &mut self.blob,
+            MessageKind::Who | MessageKind::Me => &mut self.identity,
         }
     }
 }
@@ -92,15 +96,15 @@ impl Traffic {
         }
     }
 
-    fn sent(&self, peer: &str, kind: MessageKind, bytes: usize) {
+    pub(crate) fn sent(&self, peer: &str, kind: MessageKind, bytes: usize) {
         self.count(peer, bytes, |t| t.sent.get_mut(kind));
     }
 
-    fn refused(&self, peer: &str, kind: MessageKind, bytes: usize) {
+    pub(crate) fn refused(&self, peer: &str, kind: MessageKind, bytes: usize) {
         self.count(peer, bytes, |t| t.refused.get_mut(kind));
     }
 
-    fn received(&self, peer: &str, kind: MessageKind, bytes: usize) {
+    pub(crate) fn received(&self, peer: &str, kind: MessageKind, bytes: usize) {
         self.count(peer, bytes, |t| t.received.get_mut(kind));
     }