tokio_stream/wrappers/
tcp_listener.rs

1use crate::Stream;
2use std::io;
3use std::pin::Pin;
4use std::task::{Context, Poll};
5use tokio::net::{TcpListener, TcpStream};
6
7/// A wrapper around [`TcpListener`] that implements [`Stream`].
8///
9/// [`TcpListener`]: struct@tokio::net::TcpListener
10/// [`Stream`]: trait@crate::Stream
11#[derive(Debug)]
12#[cfg_attr(docsrs, doc(cfg(feature = "net")))]
13pub struct TcpListenerStream {
14    inner: TcpListener,
15}
16
17impl TcpListenerStream {
18    /// Create a new `TcpListenerStream`.
19    pub fn new(listener: TcpListener) -> Self {
20        Self { inner: listener }
21    }
22
23    /// Get back the inner `TcpListener`.
24    pub fn into_inner(self) -> TcpListener {
25        self.inner
26    }
27}
28
29impl Stream for TcpListenerStream {
30    type Item = io::Result<TcpStream>;
31
32    fn poll_next(
33        self: Pin<&mut Self>,
34        cx: &mut Context<'_>,
35    ) -> Poll<Option<io::Result<TcpStream>>> {
36        match self.inner.poll_accept(cx) {
37            Poll::Ready(Ok((stream, _))) => Poll::Ready(Some(Ok(stream))),
38            Poll::Ready(Err(err)) => Poll::Ready(Some(Err(err))),
39            Poll::Pending => Poll::Pending,
40        }
41    }
42}
43
44impl AsRef<TcpListener> for TcpListenerStream {
45    fn as_ref(&self) -> &TcpListener {
46        &self.inner
47    }
48}
49
50impl AsMut<TcpListener> for TcpListenerStream {
51    fn as_mut(&mut self) -> &mut TcpListener {
52        &mut self.inner
53    }
54}
tokio_stream/wrappers/tcp_listener.rs

tokio_stream/wrappers/
tcp_listener.rs
