broker_tokio/net/tcp/
incoming.rs

1use crate::net::tcp::{TcpListener, TcpStream};
2
3use std::io;
4use std::pin::Pin;
5use std::task::{Context, Poll};
6
7/// Stream returned by the `TcpListener::incoming` function representing the
8/// stream of sockets received from a listener.
9#[must_use = "streams do nothing unless polled"]
10#[derive(Debug)]
11pub struct Incoming<'a> {
12    inner: &'a mut TcpListener,
13}
14
15impl Incoming<'_> {
16    pub(crate) fn new(listener: &mut TcpListener) -> Incoming<'_> {
17        Incoming { inner: listener }
18    }
19
20    #[doc(hidden)] // TODO: dox
21    pub fn poll_accept(
22        mut self: Pin<&mut Self>,
23        cx: &mut Context<'_>,
24    ) -> Poll<io::Result<TcpStream>> {
25        let (socket, _) = ready!(self.inner.poll_accept(cx))?;
26        Poll::Ready(Ok(socket))
27    }
28}
29
30#[cfg(feature = "stream")]
31impl crate::stream::Stream for Incoming<'_> {
32    type Item = io::Result<TcpStream>;
33
34    fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
35        let (socket, _) = ready!(self.inner.poll_accept(cx))?;
36        Poll::Ready(Some(Ok(socket)))
37    }
38}
broker_tokio/net/tcp/incoming.rs

broker_tokio/net/tcp/
incoming.rs
