File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
rust/cubestore/cubestore/src/queryplanner Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -2,6 +2,7 @@ use bytes::Bytes;
22use datafusion:: datasource:: physical_plan:: parquet:: DefaultParquetFileReaderFactory ;
33use datafusion:: datasource:: physical_plan:: { FileMeta , ParquetFileReaderFactory } ;
44use datafusion:: parquet:: arrow:: async_reader:: AsyncFileReader ;
5+ use datafusion:: parquet:: file:: encryption:: ParquetEncryptionConfig ;
56use datafusion:: parquet:: file:: metadata:: ParquetMetaData ;
67use datafusion:: physical_plan:: metrics:: ExecutionPlanMetricsSet ;
78use futures_util:: future:: BoxFuture ;
@@ -148,14 +149,16 @@ impl AsyncFileReader for LruCachingFileReader {
148149
149150 fn get_metadata (
150151 & mut self ,
152+ encryption_config : & Option < ParquetEncryptionConfig >
151153 ) -> BoxFuture < ' _ , datafusion:: parquet:: errors:: Result < Arc < ParquetMetaData > > > {
152154 let cache = self . cache . clone ( ) ;
153155 let path = self . path . clone ( ) ;
156+ let encryption_config = encryption_config. clone ( ) ;
154157 async move {
155158 match cache. get ( & path) {
156159 Some ( metadata) => Ok ( metadata) ,
157160 None => {
158- let metadata = self . reader . get_metadata ( ) . await ?;
161+ let metadata = self . reader . get_metadata ( & encryption_config ) . await ?;
159162 cache. insert ( path, metadata. clone ( ) ) ;
160163 Ok ( metadata)
161164 }
You can’t perform that action at this time.
0 commit comments