libzrpc 0.1.0

A RPC 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

use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpStream;

use crate::{error::ZRpcError, log};

pub struct TcpTransport {
    stream: TcpStream,
}

impl TcpTransport {
    pub fn new(stream: TcpStream) -> Self {
        Self { stream }
    }

    pub async fn send(&mut self, bytes: &[u8]) -> Result<(), ZRpcError> {
        let len = (bytes.len() as u32).to_be_bytes();
        self.stream.write_all(&len).await.map_err(ZRpcError::Io)?;
        self.stream.write_all(bytes).await.map_err(ZRpcError::Io)?;

        log!(
            "[TcpTransport:{:?}] {} bytes were sent",
            std::thread::current().id(),
            len.len() + bytes.len()
        );

        Ok(())
    }

    pub async fn receive(&mut self) -> Result<Vec<u8>, ZRpcError> {
        let mut len = [0u8; 4];
        self.stream
            .read_exact(&mut len)
            .await
            .map_err(ZRpcError::Io)?;

        let mut buf = vec![0u8; u32::from_be_bytes(len) as usize];
        self.stream
            .read_exact(&mut buf)
            .await
            .map_err(ZRpcError::Io)?;

        log!(
            "[TcpTransport:{:?}] {} bytes were received",
            std::thread::current().id(),
            len.len() + buf.len()
        );

        Ok(buf)
    }
}