Rust Tokio Chat Server 用 Rust 写一个聊天服务器有多简单

Creating a Chat Server with async Rust and Tokio

Simple Chat Room

---

use tokio::{
    io::{AsyncBufReadExt, AsyncWriteExt, BufReader},
    net::{TcpListener, TcpSocket},
    sync::broadcast::{self, error::RecvError},
};

#[tokio::main]
async fn main() {

    // let listener = TcpListener::bind("127.0.0.1:8888").unwrap();
    let listener = {
        let addr = "127.0.0.1:8888";
        let backlog = 1024;

        let socket = TcpSocket::new_v4().unwrap();
        socket.bind(addr.parse().unwrap()).unwrap();
        socket.set_reuseaddr(true).unwrap();
        socket.set_reuseport(true).unwrap();
        let listener = socket.listen(backlog).unwrap();
        listener
    };

    let (tx, _rx) = broadcast::channel(50);

    loop {
        let (mut stream, addr) = listener.accept().await.unwrap();
        stream.set_nodelay(true).unwrap();

        let tx = tx.clone();
        let mut rx = tx.subscribe();

        println!("new client: {}", addr);

        tokio::spawn(async move {
            let (reader, mut writer) = stream.split();

            let mut reader = BufReader::new(reader);

            let mut line = String::new();

            loop {
                tokio::select! {
                    result = reader.read_line(&mut line) => {
                        if result.unwrap() == 0 {
                            break;
                        }
                        println!("msg received: {}", line);
                        if line.trim() == "quit" { break; }
                        tx.send((line.clone(), addr)).unwrap();
                        line.clear();
                    }
                    result = rx.recv() => match result{
                        Ok((line, addr_other)) => {
                            if addr != addr_other {
                                writer.write_all(line.as_bytes()).await.unwrap();
                            }
                        },
                        Err(RecvError::Lagged(_num)) => continue,
                        Err(_) => break,
                    }
                };
            }

            println!("client leave: {}", addr);
        });
    }
}

tokio tutorial