about summary refs log tree commit diff
path: root/tvix/nix-compat/src/wire/bytes/reader/mod.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tvix/nix-compat/src/wire/bytes/reader/mod.rs')
-rw-r--r--tvix/nix-compat/src/wire/bytes/reader/mod.rs14
1 files changed, 14 insertions, 0 deletions
diff --git a/tvix/nix-compat/src/wire/bytes/reader/mod.rs b/tvix/nix-compat/src/wire/bytes/reader/mod.rs
index 78615faf0f4c..9f7013a4e9b1 100644
--- a/tvix/nix-compat/src/wire/bytes/reader/mod.rs
+++ b/tvix/nix-compat/src/wire/bytes/reader/mod.rs
@@ -89,6 +89,20 @@ where
             },
         }
     }
+
+    /// Remaining data length, ie not including data already read.
+    ///
+    /// If the size has not been read yet, this is [None].
+    #[allow(clippy::len_without_is_empty)] // if size is unknown, we can't answer that
+    pub fn len(&self) -> Option<u64> {
+        match self.state {
+            State::Size { .. } => None,
+            State::Body {
+                consumed, user_len, ..
+            } => Some(user_len - consumed),
+            State::Trailer(ref r) => Some(r.len() as u64),
+        }
+    }
 }
 
 impl<R: AsyncRead + Unpin> AsyncRead for BytesReader<R> {