malkuth 0.1.0

A composable service-supervision toolkit for Rust
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
42
43
44
45

//! TCP transport via [`tokio::net`] (loopback or remote).

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

use async_trait::async_trait;

use crate::{Transport, WireConn, WireListener, codec::FramedConn};

fn strip(addr: &str) -> &str {
    addr.strip_prefix("tcp://").unwrap_or(addr)
}

/// TCP transport (IPv4/IPv6 loopback or remote).
pub struct TcpTransport;

#[async_trait]
impl Transport for TcpTransport {
    async fn listen(&self, addr: &str) -> io::Result<Box<dyn WireListener>> {
        let listener = TcpListener::bind(strip(addr)).await?;
        Ok(Box::new(TcpWireListener { listener }))
    }
    async fn connect(&self, addr: &str) -> io::Result<Box<dyn WireConn>> {
        let stream = TcpStream::connect(strip(addr)).await?;
        Ok(Box::new(FramedConn::new(stream)))
    }
    fn name(&self) -> &'static str {
        "tcp"
    }
}

pub struct TcpWireListener {
    listener: TcpListener,
}

#[async_trait]
impl WireListener for TcpWireListener {
    async fn accept(&self) -> io::Result<Box<dyn WireConn>> {
        let (stream, _peer) = self.listener.accept().await?;
        Ok(Box::new(FramedConn::new(stream)))
    }
    fn local_addr(&self) -> io::Result<String> {
        Ok(format!("{}", self.listener.local_addr()?))
    }
}