mirror of
https://github.com/async-rs/async-std.git
synced 2025-04-09 01:46:42 +00:00
43 lines
1.1 KiB
Rust
43 lines
1.1 KiB
Rust
use std::sync::Arc;
|
|
|
|
use async_std::{
|
|
io::{stdin, BufReader},
|
|
net::{TcpStream, ToSocketAddrs},
|
|
prelude::*,
|
|
task,
|
|
future::select,
|
|
};
|
|
|
|
|
|
type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;
|
|
|
|
pub(crate) fn main() -> Result<()> {
|
|
task::block_on(try_main("127.0.0.1:8080"))
|
|
}
|
|
|
|
async fn try_main(addr: impl ToSocketAddrs) -> Result<()> {
|
|
let stream = Arc::new(TcpStream::connect(addr).await?);
|
|
let (reader, writer) = (stream.clone(), stream.clone());
|
|
|
|
let incoming = task::spawn(async move {
|
|
let mut messages = BufReader::new(&*reader).lines();
|
|
while let Some(message) = messages.next().await {
|
|
let message = message?;
|
|
println!("{}", message);
|
|
}
|
|
Ok(())
|
|
});
|
|
|
|
let outgoing = task::spawn(async move {
|
|
let mut stdin = BufReader::new(stdin()).lines();
|
|
|
|
while let Some(line) = stdin.next().await {
|
|
let line = line?;
|
|
let message = format!("{}\n", line);
|
|
(&*writer).write_all(message.as_bytes()).await?;
|
|
}
|
|
Ok(())
|
|
});
|
|
|
|
select!(incoming, outgoing).await
|
|
}
|