add TcpStream from std conversion (#78)

* add TcpStream from std conversion Signed-off-by: Yoshua Wuyts <[email protected]> * add TcpStream from SocketAddr Signed-off-by: Yoshua Wuyts <[email protected]>
withoutboats · Mar 26, 2019 · fab9432 · fab9432
1 parent 2d8389a
commit fab9432
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/tcp/stream.rs b/src/tcp/stream.rs
@@ -610,6 +610,24 @@ impl Future for ConnectFuture {
     }
 }
 
+impl std::convert::TryFrom<std::net::TcpStream> for TcpStream {
+    type Error = io::Error;
+
+    fn try_from(stream: std::net::TcpStream) -> Result<Self, Self::Error> {
+        let tcp = mio::net::TcpStream::from_stream(stream)?;
+        Ok(TcpStream::new(tcp))
+    }
+}
+
+impl std::convert::TryFrom<&std::net::SocketAddr> for TcpStream {
+    type Error = io::Error;
+
+    fn try_from(addr: &std::net::SocketAddr) -> Result<Self, Self::Error> {
+        let tcp = mio::net::TcpStream::connect(&addr)?;
+        Ok(TcpStream::new(tcp))
+    }
+}
+
 #[cfg(unix)]
 mod sys {
     use super::TcpStream;