Skip to content

Commit 0bb8cc2

Browse files
committed
WIP: Implement AsyncFileReader
1 parent fb4360f commit 0bb8cc2

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

rust/cubestore/cubestore/src/queryplanner/metadata_cache.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ use bytes::Bytes;
22
use datafusion::datasource::physical_plan::parquet::DefaultParquetFileReaderFactory;
33
use datafusion::datasource::physical_plan::{FileMeta, ParquetFileReaderFactory};
44
use datafusion::parquet::arrow::async_reader::AsyncFileReader;
5+
use datafusion::parquet::file::encryption::ParquetEncryptionConfig;
56
use datafusion::parquet::file::metadata::ParquetMetaData;
67
use datafusion::physical_plan::metrics::ExecutionPlanMetricsSet;
78
use 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
}

0 commit comments

Comments
 (0)