Improve ParquetFileArrowReader UX (apache#1773)

tustvold · Jun 1, 2022 · c1a91dc · c1a91dc
1 parent 2a12e50
commit c1a91dc
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/parquet/src/arrow/arrow_reader.rs b/parquet/src/arrow/arrow_reader.rs
@@ -32,6 +32,7 @@ use crate::arrow::ProjectionMask;
 use crate::errors::Result;
 use crate::file::metadata::{KeyValue, ParquetMetaData};
 use crate::file::reader::FileReader;
+use crate::schema::types::SchemaDescriptor;
 
 /// Arrow reader api.
 /// With this api, user can get arrow schema from parquet file, and read parquet data
@@ -164,10 +165,21 @@ impl ParquetFileArrowReader {
     }
 
     /// Expose the reader metadata
+    #[deprecated = "use metadata() instead"]
     pub fn get_metadata(&mut self) -> ParquetMetaData {
         self.file_reader.metadata().clone()
     }
 
+    /// Returns the parquet metadata
+    pub fn metadata(&self) -> &ParquetMetaData {
+        self.file_reader.metadata()
+    }
+
+    /// Returns the parquet schema
+    pub fn parquet_schema(&self) -> &SchemaDescriptor {
+        self.file_reader.metadata().file_metadata().schema_descr()
+    }
+
     /// Returns the key value metadata, returns `None` if [`ArrowReaderOptions::skip_arrow_metadata`]
     fn get_kv_metadata(&self) -> Option<&Vec<KeyValue>> {
         if self.options.skip_arrow_metadata {