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

to Conn and Listener.

1 files changed, 13 insertions, 1 deletions

diff --git a/net/src/transports/udp.rs b/net/src/transports/udp.rs
index 9576876..a8b505c 100644
--- a/net/src/transports/udp.rs
+++ b/net/src/transports/udp.rs
@@ -4,7 +4,7 @@ use async_trait::async_trait;
 use smol::net::UdpSocket;
 
 use crate::{
-    connection::Connection,
+    connection::{Connection, ToConn},
     endpoint::{Addr, Endpoint, Port},
     Error, Result,
 };
@@ -79,3 +79,15 @@ impl From<UdpSocket> for Box<dyn Connection> {
         Box::new(UdpConn::new(conn))
     }
 }
+
+impl ToConn for UdpSocket {
+    fn to_conn(self) -> Box<dyn Connection> {
+        self.into()
+    }
+}
+
+impl ToConn for UdpConn {
+    fn to_conn(self) -> Box<dyn Connection> {
+        Box::new(self)
+    }
+}