如何在Rust中结合Arc使用TcpStream实现跨线程读写分离

错误原因 #

你的代码存在两个核心设计问题：

• Arc<&mut TcpStream>的写法不符合Rust所有权规则：可变引用本身是独占的，不允许被多线程共享，且引用的生命周期绑定到原变量，放在Arc中很容易出现悬垂引用问题，该设计从根本上不可行。
• Arc默认仅提供内部数据的不可变访问，要修改Arc包裹的内容需要搭配具备内部可变性的容器（如Mutex/RwLock），但对于TcpStream场景有更轻量化的实现方案。

最优实现方案：克隆TcpStream句柄 #

标准库的TcpStream内置try_clone()方法，会生成一个指向同一底层套接字的独立句柄，操作系统本身保证套接字的读写操作线程安全，你可以直接将两个句柄分别分配给读线程和写线程，不需要额外的Arc或锁，性能最好。
实现代码如下：

use std::net::{TcpStream, Shutdown};
use std::io::{Read, Write};
use std::thread;

fn main() {
    // 连接成功后生成两个独立句柄，分别用于读写
    let mut stream_write = TcpStream::connect("127.0.0.1:8080").unwrap();
    let mut stream_read = stream_write.try_clone().unwrap();

    // 启动独立线程负责读取
    let read_handle = thread::spawn(move || {
        read_from_stream(&mut stream_read);
        // 读取完成后主动关闭读端，避免资源泄漏
        stream_read.shutdown(Shutdown::Read).ok();
    });

    // 主线程负责写入逻辑
    stream_write.write_all(b"hello server").unwrap();
    stream_write.flush().unwrap();

    read_handle.join().unwrap();
}

fn read_from_stream(tcp_stream: &mut TcpStream) {
    let mut buffer = String::new();
    // read_to_string会持续读取直到对端关闭连接才返回
    tcp_stream.read_to_string(&mut buffer).unwrap();
    println!("读取到的内容：{}", buffer);
}

通用共享方案：Arc搭配Mutex #

如果你的场景确实需要多线程共享同一个TcpStream实例，可以用Arc<Mutex<TcpStream>>组合，通过Mutex提供内部可变性，代码实现如下：

use std::net::TcpStream;
use std::sync::{Arc, Mutex};
use std::io::{Read, Write};
use std::thread;

fn main() {
    let tcp_stream = TcpStream::connect("127.0.0.1:8080").unwrap();
    let shared_stream = Arc::new(Mutex::new(tcp_stream));

    // 克隆Arc传递给读线程
    let stream_clone = Arc::clone(&shared_stream);
    let read_handle = thread::spawn(move || {
        let mut buffer = String::new();
        // 先获取Mutex锁，拿到可变借用后再读取
        stream_clone.lock().unwrap().read_to_string(&mut buffer).unwrap();
        println!("读取到的内容：{}", buffer);
    });

    // 主线程写入时也需要先获取锁
    shared_stream.lock().unwrap().write_all(b"hello server").unwrap();
    read_handle.join().unwrap();
}

注意：该方案读写操作都需要竞争同一把锁，会有额外的性能开销，对于需要同时读写TcpStream的场景，优先选择克隆句柄的方案。

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