echosrv 0.3.0

A high-performance set of async echo server library built with Tokio for testing and development environments
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

use crate::EchoError;
use crate::stream::{StreamConfig, StreamProtocol};
use async_trait::async_trait;
use std::net::SocketAddr;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::{TcpListener, TcpStream};

/// TCP protocol implementation
pub struct TcpProtocol;

#[async_trait]
impl StreamProtocol for TcpProtocol {
    type Error = EchoError;
    type Listener = TcpListener;
    type Stream = TcpStream;

    async fn bind(config: &StreamConfig) -> std::result::Result<TcpListener, EchoError> {
        TcpListener::bind(config.bind_addr)
            .await
            .map_err(|e| EchoError::Config(format!("Failed to bind TCP listener: {e}")))
    }

    async fn accept(
        listener: &mut TcpListener,
    ) -> std::result::Result<(TcpStream, SocketAddr), EchoError> {
        listener.accept().await.map_err(EchoError::Tcp)
    }

    async fn connect(addr: SocketAddr) -> std::result::Result<TcpStream, EchoError> {
        TcpStream::connect(addr)
            .await
            .map_err(|e| EchoError::Config(format!("Failed to connect to {addr}: {e}")))
    }

    async fn read(
        stream: &mut TcpStream,
        buffer: &mut [u8],
    ) -> std::result::Result<usize, EchoError> {
        stream.read(buffer).await.map_err(EchoError::Tcp)
    }

    async fn write(stream: &mut TcpStream, data: &[u8]) -> std::result::Result<(), EchoError> {
        stream.write_all(data).await.map_err(EchoError::Tcp)
    }

    async fn flush(stream: &mut TcpStream) -> std::result::Result<(), EchoError> {
        stream.flush().await.map_err(EchoError::Tcp)
    }

    fn map_io_error(err: std::io::Error) -> EchoError {
        EchoError::Tcp(err)
    }
}