use std::io::{self, Read};
use views::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: reader,
counter: 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 = try!(self.reader.read(buf));
self.counter.tick(result);
Ok(result)
}
}