From 82b11159b0d9618ae1df7f7b0d7460c258ee4c07 Mon Sep 17 00:00:00 2001
From: Jorge Leitao <jorgecarleitao@gmail.com>
Date: Tue, 12 Jul 2022 07:15:12 -0700
Subject: [PATCH] Added Compressor::into_inner (#158)

---
 src/write/compression.rs | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/src/write/compression.rs b/src/write/compression.rs
index a58187fa6..7ea00cc8f 100644
--- a/src/write/compression.rs
+++ b/src/write/compression.rs
@@ -107,11 +107,6 @@ pub struct Compressor<I: Iterator<Item = Result<EncodedPage>>> {
 }
 
 impl<I: Iterator<Item = Result<EncodedPage>>> Compressor<I> {
-    /// Creates a new [`Compressor`]
-    pub fn new_from_vec(iter: I, compression: CompressionOptions, buffer: Vec<u8>) -> Self {
-        Self::new(iter, compression, buffer)
-    }
-
     /// Creates a new [`Compressor`]
     pub fn new(iter: I, compression: CompressionOptions, buffer: Vec<u8>) -> Self {
         Self {
@@ -121,6 +116,22 @@ impl<I: Iterator<Item = Result<EncodedPage>>> Compressor<I> {
             current: None,
         }
     }
+
+    /// Creates a new [`Compressor`] (same as `new`)
+    pub fn new_from_vec(iter: I, compression: CompressionOptions, buffer: Vec<u8>) -> Self {
+        Self::new(iter, compression, buffer)
+    }
+
+    /// Deconstructs itself into its iterator and scratch buffer.
+    pub fn into_inner(mut self) -> (I, Vec<u8>) {
+        let mut buffer = if let Some(page) = self.current.as_mut() {
+            std::mem::take(page.buffer())
+        } else {
+            std::mem::take(&mut self.buffer)
+        };
+        buffer.clear();
+        (self.iter, buffer)
+    }
 }
 
 impl<I: Iterator<Item = Result<EncodedPage>>> FallibleStreamingIterator for Compressor<I> {