trellis-net 0.1.0

A simple networking library 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63


// ===== Imports =====
use std::{io::Error, net::SocketAddr};
use bytes::BytesMut;
use tokio::{net::TcpStream, io::{AsyncReadExt, AsyncWriteExt}};

use crate::{
  Address,
  rw::{ReadPreReqs, WritePreReqs, Reader, Writer},
};
// ===================

/// # Connection
/// I/O Object for managing Tcp connections.
/// A new connection can be created by connecting to an endpoint using `Connection::new()` constructor.
pub struct Connection {
  pub local_addr: Address,
  pub peer_addr: Address,
  stream: TcpStream,
}

impl Connection {
  /// ## Constructor
  /// Constructs a new `Connection` to the provided endpoint.
  pub async fn new(to: Address) -> Result<Self, Error> {
    let to: SocketAddr = to.into();
    let stream = TcpStream::connect(to).await?;
    let local_addr = stream.local_addr()?.into();
    let peer_addr = stream.peer_addr()?.into();
    Ok(Self { local_addr, peer_addr, stream })
  }
}

impl TryFrom<TcpStream> for Connection {
  type Error = Error;

  fn try_from(stream: TcpStream) -> Result<Self, Self::Error> {
    let local_addr = stream.local_addr()?.into();
    let peer_addr = stream.peer_addr()?.into();
    Ok(Self { local_addr, peer_addr, stream })
  }
}

#[async_trait]
impl ReadPreReqs for Connection {
  async fn read(&mut self, byts: &mut BytesMut) -> Result<(), Error> {
    self.stream.read(byts).await?;
    Ok(())
  }
}

#[async_trait]
impl WritePreReqs for Connection {
  async fn write(&mut self, byts: &mut BytesMut) -> Result<(), Error> {
    self.stream.write(&byts[..]).await?;
    Ok(())
  }
}

#[async_trait]
impl Reader for Connection {}
#[async_trait]
impl Writer for Connection {}