Add TcpListener::into_incoming

pull/995/head
piegames 3 years ago
parent f4b8c7a108
commit d3133c04be

@ -152,6 +152,45 @@ impl TcpListener {
} }
} }
/// Turn this into a stream over the connections being received on this
/// listener.
///
/// The returned stream is infinite and will also not yield
/// the peer's [`SocketAddr`] structure. Iterating over it is equivalent to
/// calling [`TcpListener::accept`] in a loop.
///
/// ## Examples
///
/// Merge the incoming connections of multiple sockets into one [`Stream`]:
///
/// ```no_run
/// # fn main() -> std::io::Result<()> { async_std::task::block_on(async {
/// #
/// use async_std::net::TcpListener;
///
/// // Our server listens on multiple ports for some reason
/// let listeners = vec![
/// TcpListener::bind("[::0]:8080").await?,
/// TcpListener::bind("[::0]:12345").await?,
/// TcpListener::bind("[::0]:5678").await?,
/// ];
/// // Iterate over all incoming connections
/// let incoming = futures::stream::select_all(
/// listeners.into_iter()
/// .map(TcpListener::into_incoming)
/// .map(Box::pin)
/// );
/// #
/// # Ok(()) }) }
/// ```
#[cfg(feature = "unstable")]
pub fn into_incoming(self) -> impl Stream<Item = io::Result<TcpStream>> + Send {
futures_lite::stream::unfold(self, |listener| async move {
let res = listener.accept().await.map(|(stream, _)| stream);
Some((res, listener))
})
}
/// Returns the local address that this listener is bound to. /// Returns the local address that this listener is bound to.
/// ///
/// This can be useful, for example, to identify when binding to port 0 which port was assigned /// This can be useful, for example, to identify when binding to port 0 which port was assigned

Loading…
Cancel
Save