From 54699222f8b69e69d16724c1a505b9d324af96e6 Mon Sep 17 00:00:00 2001
From: Lev Kokotov <levkk@users.noreply.github.com>
Date: Sat, 14 May 2022 21:35:33 -0700
Subject: [PATCH] Possible fix for clients waiting stat leak (#68)

---
 src/pool.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/pool.rs b/src/pool.rs
index 0da8278..cc8b688 100644
--- a/src/pool.rs
+++ b/src/pool.rs
@@ -215,12 +215,14 @@ impl ConnectionPool {
             // as per request. If no specific role is requested, the first
             // available will be chosen.
             if address.role != role {
+                self.stats.client_disconnecting(process_id, address.id);
                 continue;
             }
 
             allowed_attempts -= 1;
 
             if self.is_banned(address, shard, role) {
+                self.stats.client_disconnecting(process_id, address.id);
                 continue;
             }