v0.2.6: preserve buffered lines and add request read timeouts

1 files changed, 8 insertions, 0 deletions

diff --git a/src/bin/client/connection.rs b/src/bin/client/connection.rs
index 93a7829..49695dd 100644
--- a/src/bin/client/connection.rs
+++ b/src/bin/client/connection.rs
@@ -66,6 +66,7 @@ pub fn connect(
     }
 
     let mut tls = tls;
+    tls.set_read_timeout(Some(Duration::from_secs(10)))?;
 
     let start = Instant::now();
     tls.write_line("PING")?;
@@ -124,6 +125,13 @@ impl TlsStream {
         self.inner.write_line(msg)
     }
 
+    pub fn set_read_timeout(
+        &self,
+        duration: Option<std::time::Duration>,
+    ) -> std::io::Result<()> {
+        self.inner.set_read_timeout(duration)
+    }
+
     pub fn try_clone(&self) -> std::io::Result<TlsStream> {
         Ok(TlsStream {
             inner: self.inner.try_clone()?,