Factor SizeLimited constructor

remram44 · Dec 11, 2024 · c26983e · c26983e
1 parent 5fcc2e9
commit c26983e
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -255,11 +255,7 @@ impl<I: ChunkerImpl> Chunker<I> {
     pub fn max_size(self, max: usize) -> Chunker<SizeLimited<I>> {
         assert!(max > 0);
         Chunker {
-            inner: SizeLimited {
-                inner: self.inner,
-                pos: 0,
-                max_size: max,
-            },
+            inner: SizeLimited::new(self.inner, max),
         }
     }
 }
@@ -445,6 +441,17 @@ pub struct SizeLimited<I: ChunkerImpl> {
     max_size: usize,
 }
 
+impl<I: ChunkerImpl> SizeLimited<I> {
+    /// Wraps the given chunker implementation to limit the size of produced chunks.
+    pub fn new(inner: I, max_size: usize) -> Self {
+        SizeLimited {
+            inner,
+            pos: 0,
+            max_size,
+        }
+    }
+}
+
 impl<I: ChunkerImpl> ChunkerImpl for SizeLimited<I> {
     fn find_boundary(&mut self, data: &[u8]) -> Option<usize> {
         assert!(self.max_size > self.pos);