advance-practice/io

[giscus[bot]]

advance-practice/io

https://course.rs/advance-practice/io.html

[howz97]

“在堆上分配缓冲区” 翻译的很赞，感觉英文版都没有讲清楚

[clee01]

不太懂 这儿为啥要优先“在堆上分配缓冲区”
是因为 堆上 只需共享指针即可？还是

[madaoCN]

文章里面已经写了

当任务因为调度在线程间移动时，存储在栈上的数据需要进行保存和恢复，过大的栈上变量会带来不小的数据拷贝开销

因此，存储大量数据的变量最好放到堆上

[ttxs25830]

这种固定长度因为其他原因需要储存在堆上的用Box::new([0; 1024])会不会更清晰一点？

[buchenglei]

求教各位大佬, 标准库里的TcpStream可以实现split这样的操作么, 找了一圈都没找到咋实现的

[Redskaber]

tokio::net::TcpStream 怎么实现的，源码都摆在那了，你说怎么办？

[yuzheng14]

Server 代码：

use tokio::{
    io::{self, AsyncReadExt, AsyncWriteExt},
    net::TcpListener,
};

#[tokio::main]
async fn main() -> io::Result<()> {
    let listener = TcpListener::bind("127.0.0.1:6142").await?;

    loop {
        let (mut socket, _) = listener.accept().await?;

        tokio::spawn(async move {
            let mut buf = vec![0; 1024];

            loop {
                match socket.read(&mut buf).await {
                    Ok(0) => return,
                    Ok(n) => {
                        socket.write_all(&buf[..n]).await.unwrap();
                    }
                    Err(_) => return,
                }
            }
        });
    }
}

客户端代码：

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

#[tokio::main]
async fn main() -> io::Result<()> {
    let socket = TcpStream::connect("127.0.0.1:6142").await?;
    let (mut rd, mut wr) = io::split(socket);

    tokio::spawn(async move {
        wr.write_all(b"hello\r\n").await?;
        wr.write_all(b"world\r\n").await?;

        Ok::<_, io::Error>(())
    });

    let mut buf: Vec<u8> = vec![0; 128];

    loop {
        let n = rd.read(&mut buf).await?;

        if n == 0 {
            break;
        }

        println!("GOT {:?}", String::from_utf8(buf[..n].to_vec()).unwrap());
    }

    Ok(())
}

运行客户端后一直没有退出进程，打日志后发现是一直阻塞在 rd.read 这里。服务端打日志后发现也是卡在了 socket.read。尝试客户端 drop(wr) 无作用，应该怎么解决？

[TwinklerG]

没看到什么问题

[Rime9527]

tokio::spawn(async move {
    wr.write_all(b"hello\r\n").await?;
    wr.write_all(b"world\r\n").await?;

    Ok::<_, io::Error>(())
});

修改为：

tokio::spawn(async move {
    wr.write_all(b"hello\r\n").await ? ;
    wr.write_all(b"world\r\n").await ? ;
    wr.shutdown().await ? ;
    Ok::<_, io::Error>(())
});

使用 wr.shutdown().await ? 关闭通道，确保值可以被干净地释放。shutdown()标记写入完成并关闭连接