Merge pull request #72 from Fishrock123/h1-pooling-fix-recycle

fix: only recycle if conn is still open

Author: Fishrock123

src/h1/tcp.rs

@@ -58,10 +58,12 @@ impl AsyncWrite for TcpConnWrapper {
 #[async_trait]
 impl Manager<TcpStream, std::io::Error> for TcpConnection {
     async fn create(&self) -> Result<TcpStream, std::io::Error> {
-        Ok(TcpStream::connect(self.addr).await?)
+        TcpStream::connect(self.addr).await
     }
 
-    async fn recycle(&self, _conn: &mut TcpStream) -> RecycleResult<std::io::Error> {
+    async fn recycle(&self, conn: &mut TcpStream) -> RecycleResult<std::io::Error> {
+        let mut buf = [0; 4];
+        conn.peek(&mut buf[..]).await?;
         Ok(())
     }
 }

src/h1/tls.rs

@@ -74,7 +74,12 @@ impl Manager<TlsStream<TcpStream>, Error> for TlsConnection {
         Ok(tls_stream)
     }
 
-    async fn recycle(&self, _conn: &mut TlsStream<TcpStream>) -> RecycleResult<Error> {
+    async fn recycle(&self, conn: &mut TlsStream<TcpStream>) -> RecycleResult<Error> {
+        let mut buf = [0; 4];
+        conn.get_ref()
+            .peek(&mut buf[..])
+            .await
+            .map_err(Error::from)?;
         Ok(())
     }
 }