Question

当写入二进制文件格式时，能够检查已写入的字节数（例如用于对齐）或仅确保嵌套函数写入正确的数据量非常有用。

有没有办法检查std::io::Write知道已写了多少？如果没有，那么包装编写器的好方法是什么，以便它可以跟踪已写入的字节数？

Answer 1

Write有两种必需的方法：write和flush。由于write已经返回写入的字节数，因此您只需跟踪：

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

struct ByteCounter<W> {
    inner: W,
    count: usize,
}

impl<W> ByteCounter<W>
    where W: Write
{
    fn new(inner: W) -> Self {
        ByteCounter {
            inner: inner,
            count: 0,
        }
    }

    fn into_inner(self) -> W {
        self.inner
    }

    fn bytes_written(&self) -> usize {
        self.count
    }
}

impl<W> Write for ByteCounter<W>
    where W: Write
{
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
        let res = self.inner.write(buf);
        if let Ok(size) = res {
            self.count += size
        }
        res
    }

    fn flush(&mut self) -> io::Result<()> {
        self.inner.flush()
    }
}

fn main() {
    let out = std::io::stdout();
    let mut out = ByteCounter::new(out);
    writeln!(&mut out, "Hello, world! {}", 42).unwrap();
    println!("Wrote {} bytes", out.bytes_written());
}

不委托write_all或write_fmt非常重要，因为这些不会返回字节数。委托它们将允许写入字节而不进行跟踪。

Answer 2

如果您编写的类型为std::io::Seek，您可以使用seek获取当前位置：

pos = f.seek(SeekFrom::Current(0))?;

Seek由std::fs::File实现（如果包装类型也实现了Seek，则std::io::BufWriter）。

所以函数签名：

use ::std::io::{Write, Seek, SeekFrom, Error};

fn my_write<W: Write>(f: &mut W) -> Result<(), Error> { ... }

需要添加Seek特征：

fn my_write<W: Write + Seek>(f: &mut W) -> Result<(), Error> { ... }

如何跟踪使用＆＃39; std :: io :: Write＆＃39;？时写入的字节数

2 个答案: