tcplane 8.1.0

tcplane is a lightweight and high-performance Rust TCP server library designed to simplify network service development. It supports TCP communication, data stream management, and connection handling, focusing on providing efficient low-level network connections and data transmission capabilities, making it ideal for building modern network services.
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use crate::*;

#[derive(Clone, Copy, Debug)]
struct GreetingHandler;

impl ServerHook for GreetingHandler {
    async fn new(_: &Context) -> Self {
        Self
    }

    async fn handle(self, ctx: &Context) {
        let _ = ctx.send("Hello from GreetingHandler!").await;
    }
}

#[derive(Clone, Copy, Debug)]
struct EchoHandler;

impl ServerHook for EchoHandler {
    async fn new(_: &Context) -> Self {
        Self
    }

    async fn handle(self, ctx: &Context) {
        let request: Request = ctx.get_request().await;
        let response: String = format!("Echo: {request:?}");
        let _ = ctx.send(response).await;
    }
}

#[derive(Clone, Copy, Debug)]
struct PanicHandler;

impl ServerHook for PanicHandler {
    async fn new(_: &Context) -> Self {
        Self
    }

    async fn handle(self, ctx: &Context) {
        let _ = ctx.send("Panic occurred!").await;
    }
}

#[derive(Clone, Copy, Debug)]
struct ErrorHandler;

impl ServerHook for ErrorHandler {
    async fn new(_: &Context) -> Self {
        Self
    }

    async fn handle(self, ctx: &Context) {
        if let Some(error) = ctx.try_get_data::<String, _>("error").await {
            eprintln!("{error}");
            let _ = std::io::Write::flush(&mut std::io::stderr());
        }
    }
}

#[tokio::test]
async fn test_server_basic_usage() {
    let server_config: ServerConfig = ServerConfig::new();
    server_config.host("0.0.0.0").await;
    server_config.port(60000).await;
    server_config.buffer_size(100_024_000).await;
    let server: Server = Server::new();
    server.task_panic::<PanicHandler>().await;
    server.hook::<GreetingHandler>().await;
    server.hook::<EchoHandler>().await;
    server.hook::<DefaultHook>().await;
    server.read_error::<ErrorHandler>().await;
    let server_control_hook_1: ServerControlHook = server.run().await.unwrap_or_default();
    let server_control_hook_2: ServerControlHook = server_control_hook_1.clone();
    tokio::spawn(async move {
        tokio::time::sleep(std::time::Duration::from_secs(60)).await;
        server_control_hook_2.shutdown().await;
    });
    server_control_hook_1.wait().await;
}

#[tokio::test]
async fn test_server_handler_trait() {
    #[derive(Clone, Copy, Debug)]
    struct CustomHandler;
    impl ServerHook for CustomHandler {
        async fn new(_: &Context) -> Self {
            Self
        }

        async fn handle(self, ctx: &Context) {
            let _ = ctx.send("CustomHandler executed").await;
        }
    }
    let server: Server = Server::new();
    server.hook::<CustomHandler>().await;
    let hook_count: usize = server.read().await.get_hook().len();
    assert_eq!(hook_count, 1);
}

#[tokio::test]
async fn test_server_task_panic() {
    #[derive(Clone, Copy, Debug)]
    struct TestPanicHandler;
    impl ServerHook for TestPanicHandler {
        async fn new(_: &Context) -> Self {
            Self
        }

        async fn handle(self, _: &Context) {}
    }
    let server: Server = Server::new();
    server.task_panic::<TestPanicHandler>().await;
    let panic_count: usize = server.read().await.get_task_panic().len();
    assert_eq!(panic_count, 1);
}