use std::io::{self, Read};
use utils::Counter;
pub struct ProgressReader<R: Read> {
reader: R,
counter: Counter,
}
impl<R: Read> ProgressReader<R> {
pub fn new(counter: Counter, reader: R) -> Self {
ProgressReader { reader, counter }
}
pub fn deconstruct(self) -> (R, Counter) {
(self.reader, self.counter)
}
}
impl<R: Read> Read for ProgressReader<R> {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
let result = self.reader.read(buf)?;
self.counter.tick(result);
Ok(result)
}
}