Allow access to underlying reader.

src/lib.rs

@@ -142,6 +142,11 @@ impl<R: Read> MessageStream<R> {
     pub fn bytes_read(&self) -> usize {
         self.bytes_read
     }
+
+    /// Returns a reference to the underlying reader.
+    pub fn get_ref(&self) -> &R {
+        &self.reader
+    }
 }
 
 impl<R: Read> Iterator for MessageStream<R> {
@@ -210,36 +215,6 @@ impl<R: Read> Iterator for MessageStream<R> {
     }
 }
 
-/// Trait for readers that can report their size.
-pub trait KnownSizeReader {
-    /// Returns the size of the reader in bytes.
-    ///
-    /// Note: This function is not guaranteed to be fast. Cache the result if you need to call it multiple times.
-    fn size(&self) -> Result<u64>;
-}
-
-impl KnownSizeReader for File {
-    fn size(&self) -> Result<u64> {
-        Ok(self.metadata()?.len())
-    }
-}
-
-impl KnownSizeReader for GzDecoder<File> {
-    fn size(&self) -> Result<u64> {
-        Ok(self.get_ref().metadata()?.len())
-    }
-}
-
-impl<R> MessageStream<R>
-where
-    R: KnownSizeReader,
-{
-    /// Returns the size of the underlying reader in bytes.
-    pub fn reader_size(&self) -> Result<u64> {
-        self.reader.size()
-    }
-}
-
 /// Opaque type representing a price to four decimal places
 #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 #[derive(Debug, Copy, Clone, PartialEq, Eq)]
@@ -1101,8 +1076,7 @@ mod tests {
     #[ignore]
     fn test_full_parse() {
         // Download sample data from ftp://emi.nasdaq.com/ITCH/
-        let mut stream = MessageStream::from_file("sample-data/20190830.PSX_ITCH_50").unwrap();
-        let stream_size = stream.reader_size().unwrap();
+        let stream_size = stream.get_ref().metadata().unwrap().len();
 
         let mut ct = 0;
         while let Some(msg) = stream.next() {