Enhanced client stats logging output

1 files changed, 8 insertions, 2 deletions

diff --git a/src/stats/mod.rs b/src/stats/mod.rs
index 126b88b..35fabdf 100644
--- a/src/stats/mod.rs
+++ b/src/stats/mod.rs
@@ -13,6 +13,8 @@
 // limitations under the License.
 
 use hashbrown::HashMap;
+use hashbrown::hash_map::Iter;
+
 use std::net::IpAddr;
 
 /// Maximum number of stats entries to maintain.
@@ -42,6 +44,8 @@ pub trait ClientStats {
 
     fn get_stats(&self, addr: &IpAddr) -> Option<&StatEntry>;
 
+    fn iter(&self) -> Iter<IpAddr, StatEntry>;
+
     fn clear(&mut self);
 }
 
@@ -196,11 +200,13 @@ impl ClientStats for SimpleStats {
     }
 
     fn get_stats(&self, addr: &IpAddr) -> Option<&StatEntry> {
-        self.clients.iter()
-
         self.clients.get(addr)
     }
 
+    fn iter(&self) -> Iter<IpAddr, StatEntry> {
+        self.clients.iter()
+    }
+
     fn clear(&mut self) {
         self.clients.clear();
         self.num_overflows = 0;