feat: Add AsyncListener trait

Which allows us to expose the local address

Author: tmccombs

examples/axum.rs

@@ -47,7 +47,7 @@ impl axum::serve::Listener for Listener {
         }
     }
     fn local_addr(&self) -> io::Result<Self::Addr> {
-        self.0.listener().local_addr()
+        self.0.local_addr()
     }
 }
 

src/accept.rs

@@ -90,3 +90,9 @@ where
         current: first_future,
     }
 }
+
+///An AsyncListener that can also report its local address
+pub trait AsyncListener: AsyncAccept {
+    /// The local address of the listener, if available.
+    fn local_addr(&self) -> Result<Self::Address, Self::Error>;
+}

src/lib.rs

@@ -247,6 +247,11 @@ where
     pub fn listener(&self) -> &A {
         &self.listener
     }
+
+    /// Get the local address of the underlying listener
+    pub fn local_addr(&self) -> Result<A::Address, A::Error> where A: AsyncListener {
+        self.listener.local_addr()
+    }
 }
 
 impl<A, T> Stream for TlsListener<A, T>

src/net.rs

@@ -1,4 +1,4 @@
-use super::AsyncAccept;
+use super::{AsyncAccept, AsyncListener};
 use std::io;
 use std::pin::Pin;
 use std::task::{Context, Poll};
@@ -24,6 +24,14 @@ impl AsyncAccept for TcpListener {
     }
 }
 
+#[cfg_attr(docsrs, doc(cfg(feature = "tokio-net")))]
+impl AsyncListener for TcpListener {
+    #[inline]
+    fn local_addr(&self) -> Result<Self::Address, Self::Error> {
+        TcpListener::local_addr(self)
+    }
+}
+
 #[cfg(unix)]
 #[cfg_attr(docsrs, doc(cfg(feature = "tokio-net")))]
 impl AsyncAccept for UnixListener {
@@ -42,3 +50,11 @@ impl AsyncAccept for UnixListener {
         }
     }
 }
+
+#[cfg_attr(docsrs, doc(cfg(feature = "tokio-net")))]
+impl AsyncListener for UnixListener {
+    #[inline]
+    fn local_addr(&self) -> Result<Self::Address, Self::Error> {
+        UnixListener::local_addr(self)
+    }
+}