viam-rust-utils 0.5.1

Utilities designed for use with Viamrobotics's SDKs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use hyper::server::accept::Accept;
use std::io::Error;
use std::pin::Pin;
use std::task::{Context, Poll};

use tokio::net::{TcpListener, TcpStream};

pub struct Connector {
    inner: TcpListener,
    addr: String,
}

impl Connector {
    pub fn new() -> Result<Self, Error> {
        let tcp = std::net::TcpListener::bind("127.0.0.1:0")?;
        tcp.set_nonblocking(true)?;
        Ok(Connector {
            addr: tcp.local_addr()?.to_string(),
            inner: TcpListener::from_std(tcp)?,
        })
    }
    pub fn get_path(&self) -> &str {
        &self.addr
    }
}

impl Accept for Connector {
    type Conn = TcpStream;
    type Error = Error;

    fn poll_accept(
        self: Pin<&mut Self>,
        cx: &mut Context<'_>,
    ) -> Poll<Option<Result<Self::Conn, Self::Error>>> {
        match self.inner.poll_accept(cx) {
            Poll::Pending => Poll::Pending,
            Poll::Ready(Ok((stream, _port))) => Poll::Ready(Some(Ok(stream))),
            Poll::Ready(Err(err)) => Poll::Ready(Some(Err(err))),
        }
    }
}