From 17b690f7593c5d9a1bc2d9332d360dc8ca5a337c Mon Sep 17 00:00:00 2001
From: Lev Kokotov <lev.kokotov@gmail.com>
Date: Fri, 4 Feb 2022 17:57:54 -0800
Subject: [PATCH] NOW its transactional

---
 src/client.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/client.rs b/src/client.rs
index 01ed605..25f5ab2 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -141,7 +141,9 @@ impl Client {
         loop {
             // Only grab a connection once we have some traffic on the socket
             // TODO: this is not the most optimal way to share servers.
-            match self.read.get_ref().ready(Interest::READABLE).await {
+            let mut peek_buf = vec![0u8; 2];
+
+            match self.read.get_mut().peek(&mut peek_buf).await {
                 Ok(_) => (),
                 Err(_) => return Err(Error::ClientDisconnected),
             };