uni-stream 0.3.4

Provides universal stream for TCP and UDP traffic and custom DNS resolution service
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

use std::time::Instant;

use tokio::io::{AsyncReadExt, AsyncWriteExt};
use uni_stream::addr::ToSocketAddrs;
use uni_stream::stream::{StreamProvider, TcpStreamProvider, UdpStreamProvider};

struct TimerTickGurad {
    ins: Instant,
}

impl TimerTickGurad {
    fn new() -> Self {
        Self {
            ins: Instant::now(),
        }
    }
}

impl Drop for TimerTickGurad {
    fn drop(&mut self) {
        let end = Instant::now();
        let duration = end - self.ins;
        println!("duration:{duration:?}");
    }
}

/// test echo delay
async fn echo_delay<P: StreamProvider, A: ToSocketAddrs + 'static + Send>(addr: A) {
    let mut stream = P::from_addr(addr).await.unwrap();
    let mut buf = [0; 1024];
    let expected = b"abc";
    for _ in 0..10 {
        let n = {
            let _guard = TimerTickGurad::new();
            stream.write_all(expected).await.unwrap();
            stream.read(&mut buf).await.unwrap()
        };

        assert_eq!(expected, &buf[..n]);
    }
}

#[tokio::main]
async fn main() {
    let is_udp = true;
    if is_udp {
        echo_delay::<UdpStreamProvider, _>("127.0.0.1:8080").await
    } else {
        echo_delay::<TcpStreamProvider, _>("127.0.0.1:8080").await
    }
}