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
|
//! An example to show how to use TcpStream.
#[cfg(not(target_os = "linux"))]
fn main() {}
#[cfg(target_os = "linux")]
fn main() {
println!("Will run with IoUringDriver(you must be on linux and enable iouring feature)");
run::<monoio::IoUringDriver>();
}
#[cfg(target_os = "linux")]
fn run<D>()
where
D: monoio::Buildable + monoio::Driver,
{
use futures::channel::oneshot;
use monoio::{
io::{AsyncReadRent, AsyncWriteRentExt},
net::{TcpListener, TcpStream},
};
const ADDRESS: &str = "127.0.0.1:50000";
let (mut tx, rx) = oneshot::channel::<()>();
let client_thread = std::thread::spawn(|| {
monoio::start::<D, _>(async move {
println!("[Client] Waiting for server ready");
tx.cancellation().await;
println!("[Client] Server is ready, will connect and send data");
let mut conn = TcpStream::connect(ADDRESS)
.await
.expect("[Client] Unable to connect to server");
let buf: Vec<u8> = vec![97; 10];
let (r, _) = conn.write_all(buf).await;
println!("[Client] Written {} bytes data and leave", r.unwrap());
});
});
let server_thread = std::thread::spawn(|| {
monoio::start::<D, _>(async move {
let listener = TcpListener::bind(ADDRESS)
.unwrap_or_else(|_| panic!("[Server] Unable to bind to {ADDRESS}"));
println!("[Server] Bind ready");
drop(rx);
let (mut conn, _addr) = listener
.accept()
.await
.expect("[Server] Unable to accept connection");
println!("[Server] Accepted a new connection, will read form it");
let buf = vec![0; 64];
let (r, buf) = conn.read(buf).await;
let read_len = r.unwrap();
println!(
"[Server] Read {} bytes data: {:?}",
read_len,
&buf[..read_len]
);
});
});
server_thread.join().unwrap();
client_thread.join().unwrap();
}
|
