Fixed error in declaring size of compressed dict page.

src/page/page_dict/mod.rs

@@ -37,11 +37,16 @@ impl EncodedDictPage {
 pub struct CompressedDictPage {
     pub(crate) buffer: Vec<u8>,
     pub(crate) num_values: usize,
+    pub(crate) uncompressed_page_size: usize,
 }
 
 impl CompressedDictPage {
-    pub fn new(buffer: Vec<u8>, num_values: usize) -> Self {
-        Self { buffer, num_values }
+    pub fn new(buffer: Vec<u8>, uncompressed_page_size: usize, num_values: usize) -> Self {
+        Self {
+            buffer,
+            uncompressed_page_size,
+            num_values,
+        }
     }
 }
 

src/write/compression.rs

@@ -59,12 +59,17 @@ fn compress_dict(
         mut buffer,
         num_values,
     } = page;
+    let uncompressed_page_size = buffer.len();
     if compression != Compression::Uncompressed {
         compression::compress(compression, &buffer, &mut compressed_buffer)?;
     } else {
         std::mem::swap(&mut buffer, &mut compressed_buffer);
     }
-    Ok(CompressedDictPage::new(compressed_buffer, num_values))
+    Ok(CompressedDictPage::new(
+        compressed_buffer,
+        uncompressed_page_size,
+        num_values,
+    ))
 }
 
 pub fn compress(

src/write/page.rs

@@ -127,7 +127,7 @@ fn assemble_data_page_header(compressed_page: &CompressedDataPage) -> ParquetPag
 fn assemble_dict_page_header(page: &CompressedDictPage) -> ParquetPageHeader {
     ParquetPageHeader {
         type_: PageType::DICTIONARY_PAGE,
-        uncompressed_page_size: page.buffer.len() as i32,
+        uncompressed_page_size: page.uncompressed_page_size as i32,
         compressed_page_size: page.buffer.len() as i32,
         crc: None,
         data_page_header: None,