From 5111a3d5749625c3d8e26a24a5a32c4da58f18d3 Mon Sep 17 00:00:00 2001
From: hozan23 <hozan23@proton.me>
Date: Thu, 30 Nov 2023 23:51:18 +0300
Subject: net: Use ToConn and ToListener traits for objects that can be
 converted to Conn and Listener.

---
 net/src/transports/udp.rs | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'net/src/transports/udp.rs')

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)
+    }
+}
-- 
cgit v1.2.3