refactor arrow reader to parquet reader in MetadataTest.cpp

facebookincubator · Nov 7, 2024 · 4af254d · 4af254d
1 parent 327fa89
commit 4af254d
Show file tree

Hide file tree

Showing 3 changed files with 881 additions and 0 deletions.
diff --git a/velox/dwio/parquet/reader/ParquetReader.cpp b/velox/dwio/parquet/reader/ParquetReader.cpp
@@ -74,6 +74,10 @@ class ReaderBase {
     return *fileMetaData_;
   }
 
+  std::shared_ptr<thrift::FileMetaData> thriftFileMetaDatas() const {
+      return std::make_shared<thrift::FileMetaData>(*fileMetaData_);
+  }
+
   FileMetaDataPtr fileMetaData() const {
     return FileMetaDataPtr(reinterpret_cast<const void*>(fileMetaData_.get()));
   }
@@ -1166,4 +1170,8 @@ FileMetaDataPtr ParquetReader::fileMetaData() const {
   return readerBase_->fileMetaData();
 }
 
+std::shared_ptr<thrift::FileMetaData> ParquetReader::thriftFileMetaData() const {
+  return readerBase_->thriftFileMetaDatas();
+}
+
 } // namespace facebook::velox::parquet
diff --git a/velox/dwio/parquet/reader/ParquetReader.h b/velox/dwio/parquet/reader/ParquetReader.h
@@ -106,6 +106,8 @@ class ParquetReader : public dwio::common::Reader {
 
   FileMetaDataPtr fileMetaData() const;
 
+  std::shared_ptr<thrift::FileMetaData> thriftFileMetaData() const;
+
  private:
   std::shared_ptr<ReaderBase> readerBase_;
 };