limit-reader 0.1.4

A simple reader that limits the number of bytes read from an underlying reader.
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
37
38
39
40
41
42
43
44
45
46
47
48

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

struct LimitReader<T: Read> {
    reader: T,
    limit: usize,
    reader_count: usize,
}

impl<T: Read> LimitReader<T> {
    fn new(reader: T, limit: usize) -> Self {
        LimitReader {
            reader,
            limit,
            reader_count: 0,
        }
    }
}

impl<T: Read> Read for LimitReader<T> {
    fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
        let max_read = self.limit.min(buf.len()); // min of limit and buf.len()
        let bytes_read = self.reader.read(&mut buf[..max_read])?;
        self.reader_count += 1;
        Ok(bytes_read)
    }
}

#[cfg(test)]
mod tests {
    use std::fs;
    use std::io::Read;

    use crate::LimitReader;

    #[test]
    fn test_read() {
        let hello_txt = fs::File::open("testdata/hello.txt").unwrap();
        let mut reader = LimitReader::new(hello_txt, 5);
        let mut r1 = vec![0u8];
        reader.read_to_end(&mut r1).unwrap();

        let mut hello_txt = fs::File::open("testdata/hello.txt").unwrap();
        let mut r2 = vec![0u8];
        hello_txt.read_to_end(&mut r2).unwrap();
        assert_eq!(r1, r2);
    }
}