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
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)
}
}