fix inefficient multi readers

1 files changed, 8 insertions, 2 deletions

diff --git a/src/unityfs/multi_reader.rs b/src/unityfs/multi_reader.rs
index 3de6cd5..64e1670 100644
--- a/src/unityfs/multi_reader.rs
+++ b/src/unityfs/multi_reader.rs
@@ -1,10 +1,11 @@
+use anyhow::Result;
+use log::trace;
 use std::{
     io::{Read, Seek, SeekFrom},
     sync::{Arc, Mutex},
 };
 
-use anyhow::Result;
-
+/// Splits an reader into many by seeking in between reads.
 pub struct MultiReader<T> {
     position: u64,
     inner: Arc<Mutex<(u64, T)>>,
@@ -30,7 +31,12 @@ impl<T: Read + Seek> Read for MultiReader<T> {
     fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> {
         let mut g = self.inner.lock().unwrap();
         if g.0 != self.position {
+            trace!(
+                "seeking to match stream position ({} actual vs {} required)",
+                g.0, self.position
+            );
             g.1.seek(SeekFrom::Start(self.position))?;
+            g.0 = self.position;
         }
         let size = g.1.read(buf)?;
         g.0 += size as u64;