如何用Rust与Tokio实现TCP客户端异步收发消息（修复代码问题）

你的代码核心问题是**std::io::stdin().read_line是同步阻塞操作，不属于Tokio异步生态的Future类型**，无法在tokio::select!中使用。此外代码还存在其他问题：外层无限循环重复生成任务会导致stream被多次所有权转移报错、writer.write_all未异步等待、读取后字符串未清空导致内容累加等。

以下是修正后的完整代码：

use tokio::io::{self, AsyncBufReadExt, AsyncWriteExt, BufReader};
use tokio::net::TcpStream;

#[tokio::main]
async fn main() -> io::Result<()> {
    // 连接TCP服务器
    let stream = TcpStream::connect("localhost:8080").await?;
    let (reader, mut writer) = stream.split();
    let mut reader = BufReader::new(reader);

    // 存储服务器响应内容
    let mut server_line = String::new();
    // 存储用户输入内容
    let mut input_line = String::new();

    loop {
        tokio::select! {
            // 异步读取服务器消息
            result = reader.read_line(&mut server_line) => {
                match result {
                    Ok(0) => {
                        println!("服务器连接已关闭");
                        break;
                    }
                    Ok(_) => {
                        print!("收到服务器消息: {server_line}");
                        server_line.clear(); // 清空缓冲区避免内容累加
                    }
                    Err(e) => {
                        eprintln!("读取服务器消息失败: {e}");
                        break;
                    }
                }
            }
            // 异步读取标准输入
            result = io::stdin().read_line(&mut input_line) => {
                match result {
                    Ok(0) => {
                        println!("标准输入已关闭");
                        break;
                    }
                    Ok(_) => {
                        // 异步写入TCP流
                        if let Err(e) = writer.write_all(input_line.as_bytes()).await {
                            eprintln!("发送消息失败: {e}");
                            break;
                        }
                        println!("消息已发送");
                        input_line.clear(); // 清空输入缓冲区
                    }
                    Err(e) => {
                        eprintln!("读取标准输入失败: {e}");
                        break;
                    }
                }
            }
        }
    }

    Ok(())
}

关键修改说明： #

• 替换为Tokio异步标准输入：用tokio::io::stdin()替代std::io::stdin，它的read_line是异步方法，返回符合要求的Future，可在tokio::select!中使用。
• 移除外层无效循环：原代码外层无限loop反复生成任务会导致stream所有权冲突，单个任务即可持续处理收发逻辑。
• 异步处理写入操作：writer.write_all必须添加await等待异步写入完成，同时处理可能的IO错误。
• 清空缓冲区：每次读取后调用clear()清空字符串，避免旧内容残留导致输出重复。
• 显式错误处理：替换unwrap()为错误匹配逻辑，让程序在遇到异常时能友好提示并退出，而非直接panic。
• 使用Tokio兼容的BufReader：确保读取操作适配Tokio runtime，保持异步一致性。

内容的提问来源于stack exchange，提问作者tailai tailai