std :: io :: Cursor

Question

Vec支持std::io::Write，因此代码可以编写为File或Vec。从API参考来看，Vec和切片都不支持std::io::Read。

有没有方便的方法来实现这一目标？它是否需要编写包装器结构？

这是一个工作代码的示例，它读取和写入一个文件，其中一行注释应该读取一个向量。

use ::std::io;

// Generic IO
fn write_4_bytes<W>(mut file: W) -> Result<usize, io::Error>
    where W: io::Write,
{
    let len = file.write(b"1234")?;
    Ok(len)
}

fn read_4_bytes<R>(mut file: R) -> Result<[u8; 4], io::Error>
    where R: io::Read,
{
    let mut buf: [u8; 4] = [0; 4];
    file.read(&mut buf)?;
    Ok(buf)
}

// Type specific

fn write_read_vec() {
    let mut vec_as_file: Vec<u8> = Vec::new();

    {   // Write
        println!("Writing Vec... {}", write_4_bytes(&mut vec_as_file).unwrap());
    }

    {   // Read
//      println!("Reading File... {:?}", read_4_bytes(&vec_as_file).unwrap());
        //                               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        //                               Comment this line above to avoid an error!
    }
}

fn write_read_file() {
    let filepath = "temp.txt";
    {   // Write
        let mut file_as_file = ::std::fs::File::create(filepath).expect("open failed");
        println!("Writing File... {}", write_4_bytes(&mut file_as_file).unwrap());
    }

    {   // Read
        let mut file_as_file = ::std::fs::File::open(filepath).expect("open failed");
        println!("Reading File... {:?}", read_4_bytes(&mut file_as_file).unwrap());
    }
}

fn main() {
    write_read_vec();
    write_read_file();
}

失败并显示错误：

error[E0277]: the trait bound `std::vec::Vec<u8>: std::io::Read` is not satisfied
  --> src/main.rs:29:42
   |
29 |         println!("Reading File... {:?}", read_4_bytes(&vec_as_file).unwrap());
   |                                          ^^^^^^^^^^^^ the trait `std::io::Read` is not implemented for `std::vec::Vec<u8>`
   |
   = note: required by `read_4_bytes`

我想为文件格式编码器/解码器编写测试，而不必写入文件系统。

Answer 1

std :: io :: Cursor

std::io::Cursor是一个简单而有用的包装器，它为Read实现了Vec<u8>，因此可以将vector用作可读实体。

let mut file = Cursor::new(vector);

read_something(&mut file);

documentation展示了如何使用Cursor而不是File编写单元测试！

工作示例：

use std::io::Cursor;
use std::io::Read;

fn read_something(file: &mut impl Read) {
    let _ = file.read(&mut [0; 8]);
}

fn main() {
    let vector = vec![1, 2, 3, 4];

    let mut file = Cursor::new(vector);

    read_something(&mut file);
}

从documentation关于std::io::Cursor：

游标通常与内存缓冲区一起使用，以允许它们实现Read和/或Write ...

标准库在通常用作缓冲区的各种类型上实现了一些I / O特性，例如Cursor<Vec<u8>>和Cursor<&[u8]>。

切片

上面的示例也适用于切片。在这种情况下，它将如下所示：

read_something(&mut &vector[..]);

工作示例：

use std::io::Read;

fn read_something(file: &mut impl Read) {
    let _ = file.read(&mut [0; 8]);
}

fn main() {
    let vector = vec![1, 2, 3, 4];

    read_something(&mut &vector[..]);
}

&mut &vector[..]是“对切片的可变引用”（对矢量部分的引用），因此我只是发现Cursor的显式选项更加清晰和优雅。

光标<->切片

更多：如果您有一个拥有缓冲区的Cursor，并且您需要模拟例如“文件”的一部分，则可以从{{ 1}}，然后传递给函数。

slice

Answer 2

虽然向量不支持std::io::Read，但切片会支持。

这里有一些混乱，因为Rust在某些情况下能够将Vec强制转换为切片而不是其他情况。

在这种情况下，需要明确强制切片，因为在应用强制阶段时，编译器不知道Vec<u8> 不实现{{1} }。

当向量被强制转换为切片时，问题中的代码将起作用：Read或read_4_bytes(&*vec_as_file)。

注意：

最初提问时，我正在使用read_4_bytes(&vec_as_file[..])代替&Read。这使得传递对切片的引用失败，除非我传入了Read我不认为这样做。
感谢&&*vec_as_file上的@arete寻找解决方案！

如何从Vec或Slice读取（std :: io :: Read）？

2 个答案:

std :: io :: Cursor

切片

光标<->切片