1 files changed, 82 insertions, 0 deletions

diff --git a/paasio/src/lib.rs b/paasio/src/lib.rs
new file mode 100644
index 0000000..71f4c9d
--- /dev/null
+++ b/paasio/src/lib.rs
@@ -0,0 +1,82 @@
+use std::io::{Read, Result, Write};
+
+pub struct ReadStats<R> {
+    source: R,
+    reads: usize,
+    bytes: usize,
+}
+
+impl<R: Read> ReadStats<R> {
+    pub fn new(wrapped: R) -> ReadStats<R> {
+        ReadStats {
+            source: wrapped,
+            reads: 0,
+            bytes: 0,
+        }
+    }
+
+    pub fn get_ref(&self) -> &R {
+        &self.source
+    }
+
+    pub fn bytes_through(&self) -> usize {
+        self.bytes
+    }
+
+    pub fn reads(&self) -> usize {
+        self.reads
+    }
+}
+
+impl<R: Read> Read for ReadStats<R> {
+    fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
+        let bytes = self.source.read(buf)?;
+        self.reads += 1;
+        self.bytes += bytes;
+        Ok(bytes)
+    }
+}
+
+pub struct WriteStats<W> {
+    sink: W,
+    writes: usize,
+    bytes: usize,
+}
+
+impl<W: Write> WriteStats<W> {
+    // _wrapped is ignored because W is not bounded on Debug or Display and therefore
+    // can't be passed through format!(). For actual implementation you will likely
+    // wish to remove the leading underscore so the variable is not ignored.
+    pub fn new(wrapped: W) -> WriteStats<W> {
+        WriteStats {
+            sink: wrapped,
+            writes: 0,
+            bytes: 0,
+        }
+    }
+
+    pub fn get_ref(&self) -> &W {
+        &self.sink
+    }
+
+    pub fn bytes_through(&self) -> usize {
+        self.bytes
+    }
+
+    pub fn writes(&self) -> usize {
+        self.writes
+    }
+}
+
+impl<W: Write> Write for WriteStats<W> {
+    fn write(&mut self, buf: &[u8]) -> Result<usize> {
+        let bytes = self.sink.write(buf)?;
+        self.writes += 1;
+        self.bytes += bytes;
+        Ok(bytes)
+    }
+
+    fn flush(&mut self) -> Result<()> {
+        self.sink.flush()
+    }
+}