Merge pull request #278 from tobz/tobz/expose-encoder-stats-methods

Expose `total_in`/`total_out` from underlying `flate2` encoder types.

Author: robjtede

src/codec/flate/encoder.rs

@@ -17,6 +17,10 @@ impl FlateEncoder {
         }
     }
 
+    pub(crate) fn get_ref(&self) -> &Compress {
+        &self.compress
+    }
+
     fn encode(
         &mut self,
         input: &mut PartialBuffer<impl AsRef<[u8]>>,

src/codec/zlib/encoder.rs

@@ -14,6 +14,10 @@ impl ZlibEncoder {
             inner: crate::codec::FlateEncoder::new(level, true),
         }
     }
+
+    pub(crate) fn get_ref(&self) -> &crate::codec::FlateEncoder {
+        &self.inner
+    }
 }
 
 impl Encode for ZlibEncoder {

src/futures/bufread/generic/encoder.rs

@@ -47,6 +47,10 @@ impl<R: AsyncBufRead, E: Encode> Encoder<R, E> {
         self.project().reader
     }
 
+    pub(crate) fn get_encoder_ref(&self) -> &E {
+        &self.encoder
+    }
+
     pub fn into_inner(self) -> R {
         self.reader
     }

src/futures/write/generic/encoder.rs

@@ -51,6 +51,10 @@ impl<W: AsyncWrite, E: Encode> Encoder<W, E> {
         self.project().writer.get_pin_mut()
     }
 
+    pub(crate) fn get_encoder_ref(&self) -> &E {
+        &self.encoder
+    }
+
     pub fn into_inner(self) -> W {
         self.writer.into_inner()
     }

src/macros.rs

@@ -131,6 +131,16 @@ macro_rules! algos {
                     ),
                 }
             }
+
+            /// Returns the total number of input bytes which have been processed by this compression object.
+            pub fn total_in(&self) -> u64 {
+                self.inner.get_encoder_ref().get_ref().get_ref().total_in()
+            }
+
+            /// Returns the total number of output bytes which have been produced by this compression object.
+            pub fn total_out(&self) -> u64 {
+                self.inner.get_encoder_ref().get_ref().get_ref().total_out()
+            }
         }
         { @dec }
         );

src/tokio/bufread/generic/encoder.rs

@@ -47,6 +47,10 @@ impl<R: AsyncBufRead, E: Encode> Encoder<R, E> {
         self.project().reader
     }
 
+    pub(crate) fn get_encoder_ref(&self) -> &E {
+        &self.encoder
+    }
+
     pub fn into_inner(self) -> R {
         self.reader
     }

src/tokio/write/generic/encoder.rs

@@ -51,6 +51,10 @@ impl<W: AsyncWrite, E: Encode> Encoder<W, E> {
         self.project().writer.get_pin_mut()
     }
 
+    pub(crate) fn get_encoder_ref(&self) -> &E {
+        &self.encoder
+    }
+
     pub fn into_inner(self) -> W {
         self.writer.into_inner()
     }