如何在Rust结构体中使用TcpStream并在多函数间共享

原代码存在几个核心问题：

• 使用&TcpStream引用但未持有所有权，无法在connect方法中为其赋值
• &self是不可变引用，无法修改结构体字段
• conn字段没有初始值，违反Rust的安全初始化规则

Rust中没有null，我们可以用Option<TcpStream>来表示连接的"未初始化/已初始化"状态，配合可变引用实现延迟初始化逻辑。以下是修正后的实现：

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

pub struct Connection {
    url: String,
    conn: Option<TcpStream>,
}

impl Connection {
    // 初始化Connection，此时连接未建立
    pub fn new(url: &str) -> Self {
        Connection {
            url: url.to_string(),
            conn: None,
        }
    }

    // 建立连接，使用&mut self来修改conn字段
    pub fn connect(&mut self) -> Result<(), io::Error> {
        let stream = TcpStream::connect(&self.url)?;
        self.conn = Some(stream);
        Ok(())
    }

    // 写入数据，需要先检查连接是否存在，同时需要&mut self因为TcpStream的write需要可变引用
    pub fn write(&mut self, cmd: &str) -> Result<usize, io::Error> {
        match &mut self.conn {
            Some(stream) => stream.write(cmd.as_bytes()),
            None => Err(io::Error::new(io::ErrorKind::NotConnected, "Connection not established")),
        }
    }

    // 读取数据，同理需要检查连接状态和可变引用
    pub fn read(&mut self, buf: &mut [u8]) -> Result<usize, io::Error> {
        match &mut self.conn {
            Some(stream) => stream.read(buf),
            None => Err(io::Error::new(io::ErrorKind::NotConnected, "Connection not established")),
        }
    }
}

// 使用示例
fn main() -> Result<(), io::Error> {
    let mut my_connection = Connection::new("localhost:23");
    my_connection.connect()?;
    
    let bytes_written = my_connection.write("Hello\r\n")?;
    println!("Wrote {} bytes", bytes_written);

    let mut buf = [0; 1024];
    let bytes_read = my_connection.read(&mut buf)?;
    println!("Read: {}", String::from_utf8_lossy(&buf[..bytes_read]));

    Ok(())
}

关键说明 #

• Option<TcpStream>完美替代了其他语言中"null"的作用，明确标记连接的状态
• connect方法使用&mut self，允许修改结构体内部的conn字段
• write和read方法需要&mut self，因为TcpStream的读写操作都需要可变引用（保证线程安全和状态一致性）
• 加入了错误处理，符合Rust的错误处理范式，比单纯返回bool更实用

你之前尝试的once_cell和lazy_static是用于全局静态变量的延迟初始化，并不适用于每个实例独立持有的连接场景，所以这里用Option是最合理的方案。

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