provekit-common 1.0.0

Common types and utilities for the ProveKit proving system
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

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

/// Helper to count bytes written to a writer.
pub struct CountingWriter<T: Write> {
    writer: T,
    count:  usize,
}

impl<T: Write> CountingWriter<T> {
    #[must_use]
    pub const fn new(writer: T) -> Self {
        Self { writer, count: 0 }
    }

    #[must_use]
    pub const fn count(&self) -> usize {
        self.count
    }
}

impl<T: Write> Write for CountingWriter<T> {
    fn write(&mut self, buf: &[u8]) -> Result<usize> {
        let written = self.writer.write(buf)?;
        self.count += written;
        Ok(written)
    }

    fn flush(&mut self) -> Result<()> {
        self.writer.flush()
    }
}