net: Use ToConn and ToListener traits for objects that can be converted

to Conn and Listener.

1 files changed, 20 insertions, 2 deletions

diff --git a/net/src/transports/tcp.rs b/net/src/transports/tcp.rs
index 7cd7127..99243b5 100644
--- a/net/src/transports/tcp.rs
+++ b/net/src/transports/tcp.rs
@@ -7,9 +7,9 @@ use smol::{
 };
 
 use crate::{
-    connection::Connection,
+    connection::{Connection, ToConn},
     endpoint::{Addr, Endpoint, Port},
-    listener::ConnListener,
+    listener::{ConnListener, ToListener},
     Error, Result,
 };
 
@@ -95,3 +95,21 @@ impl From<TcpListener> for Box<dyn ConnListener> {
         Box::new(listener)
     }
 }
+
+impl ToConn for TcpStream {
+    fn to_conn(self) -> Box<dyn Connection> {
+        self.into()
+    }
+}
+
+impl ToConn for TcpConn {
+    fn to_conn(self) -> Box<dyn Connection> {
+        Box::new(self)
+    }
+}
+
+impl ToListener for TcpListener {
+    fn to_listener(self) -> Box<dyn ConnListener> {
+        self.into()
+    }
+}