gitoid 0.9.0

Git Object Identifiers in Rust
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
32
33
34
35
36

use crate::Result;

#[cfg(feature = "std")]
use std::io::BufRead;

#[cfg(feature = "std")]
/// Helper extension trait to give a convenient way to iterate over
/// chunks sized to the size of the internal buffer of the reader.
pub(crate) trait ForEachBufFill: BufRead {
    /// Takes a function to apply to each buffer fill, and returns if any
    /// errors arose along with the number of bytes read in total.
    fn for_each_buf_fill(&mut self, f: impl FnMut(&[u8])) -> Result<usize>;
}

#[cfg(feature = "std")]
impl<R: BufRead> ForEachBufFill for R {
    fn for_each_buf_fill(&mut self, mut f: impl FnMut(&[u8])) -> Result<usize> {
        let mut total_read = 0;

        loop {
            let buffer = self.fill_buf()?;
            let amount_read = buffer.len();

            if amount_read == 0 {
                break;
            }

            f(buffer);

            self.consume(amount_read);
            total_read += amount_read;
        }

        Ok(total_read)
    }
}