Question

Read::read返回实际读取的字节数，可以小于请求的缓冲区。在许多情况下，可以对read进行多次调用以完全填充缓冲区。

我有这段代码，但看起来很笨拙：

use std::io::{self, Read};

fn read_complete<R>(mut rdr: R, buf: &mut [u8]) -> io::Result<()>
    where R: Read
{
    let mut total_read = 0;

    loop {
        let window = &mut buf[total_read..];
        let bytes_read = try!(rdr.read(window));

        // Completely filled the buffer
        if window.len() == bytes_read {
            return Ok(());
        }

        // Unable to read anything
        if bytes_read == 0 {
            return Err(io::Error::new(io::ErrorKind::Other, "Unable to read complete buffer"));
        }

        // Partial read, continue
        total_read += bytes_read;
    }
}

fn main() {}

标准库中是否有一个函数可以为我抽象出这项工作？

Answer 1

此答案适用于1.6.0之前的Rust版本

据我所知。

查看byteorder crate's source，其中定义了read_all方法：

fn read_full<R: io::Read + ?Sized>(rdr: &mut R, buf: &mut [u8]) -> Result<()> {
    let mut nread = 0usize;
    while nread < buf.len() {
        match rdr.read(&mut buf[nread..]) {
            Ok(0) => return Err(Error::UnexpectedEOF),
            Ok(n) => nread += n,
            Err(ref e) if e.kind() == io::ErrorKind::Interrupted => {},
            Err(e) => return Err(From::from(e))
        }
    }
    Ok(())
}

请注意，这会处理中断的IO操作。

还有一个proposed RFC，几个月前提交，进入最后评论期，然后进行了更改，以便最终评论期间等待另一次复飞。

事实证明，这出乎意料地复杂了。：P

在读取字节时填充用户提供的缓冲区的惯用方法是什么？

1 个答案: