don't block in poll_x fns, fixes conn closing issues

[connect-rs.git] / src / lib.rs

diff --git a/src/lib.rs b/src/lib.rs
index 9239db5379336efb356c753c0a3c2bbf03384337..c85c2cd8d99a5b50f05a257ec447454b7bd77f92 100644 (file)
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -6,7 +6,7 @@ mod writer;
 
 pub use crate::reader::ConnectionReader;
 pub use crate::writer::ConnectionWriter;
-use async_std::net::SocketAddr;
+use async_std::{net::SocketAddr, pin::Pin};
 use futures::{AsyncRead, AsyncWrite};
 pub use futures::{SinkExt, StreamExt};
 
@@ -22,8 +22,8 @@ impl Connection {
     pub(crate) fn new(
         local_addr: SocketAddr,
         peer_addr: SocketAddr,
-        read_stream: Box<dyn AsyncRead + Send + Sync + Unpin>,
-        write_stream: Box<dyn AsyncWrite + Send + Sync + Unpin>,
+        read_stream: Pin<Box<dyn AsyncRead + Send + Sync>>,
+        write_stream: Pin<Box<dyn AsyncWrite + Send + Sync>>,
     ) -> Self {
         Self {
             local_addr,