如果文件更短，如何读取文件的前 N ​​个字节或更短？ [复制]

Question

有没有一种简单的方法可以在 Rust 中读取文件的前 N ​​个字节？两个最相关的函数似乎是read 和read_exact，但read 可以返回比可用字节少的字节，所以我不得不在一个烦人的循环中调用它，如果read_exact 放弃了文件比 N 个字节短（而我希望它只读取整个文件）。

这不是这个问题的重复，可以用read_exact解决：How to read a specific number of bytes from a stream?

Answer 1

我只是复制read_exact 的实现并稍微修改一下。它已经非常接近按预期工作了。

/// This is the same as read_exact, except if it reaches EOF it doesn't return
/// an error, and it returns the number of bytes read.
fn read_up_to(file: &mut impl std::io::Read, mut buf: &mut [u8]) -> Result<usize, std::io::Error> {
    let buf_len = buf.len();

    while !buf.is_empty() {
        match file.read(buf) {
            Ok(0) => break,
            Ok(n) => {
                let tmp = buf;
                buf = &mut tmp[n..];
            }
            Err(ref e) if e.kind() == std::io::ErrorKind::Interrupted => {}
            Err(e) => return Err(e),
        }
    }
    Ok(buf_len - buf.len())
}

（完全未经测试！）