qorb 0.4.1

Connection Pooling
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

use std::env;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpListener;

#[tokio::main]
async fn main() {
    let args: Vec<String> = env::args().collect();

    let bind_address = if args.len() == 1 {
        "[::1]:0"
    } else if args.len() == 2 {
        &args[1]
    } else {
        eprintln!("Usage: {} <optional bind address>", args[0]);
        return;
    };

    let listener = TcpListener::bind(bind_address).await.unwrap();
    let addr = listener.local_addr().unwrap();

    println!("listening started on {}, ready to accept", addr);
    while let Ok((mut stream, _)) = listener.accept().await {
        tokio::task::spawn(async move {
            let mut buf = vec![0; 1024];
            loop {
                let Ok(n) = stream.read(&mut buf[0..]).await else {
                    return;
                };

                if n == 0 {
                    return;
                }

                let Ok(_) = stream.write_all(&buf[0..n]).await else {
                    return;
                };
            }
        });
    }
}