From 9a28a1eb5e21ebadea7ac92c12953e70399ac4e2 Mon Sep 17 00:00:00 2001 From: Ruihang Xia Date: Mon, 29 Jan 2024 16:29:21 +0800 Subject: [PATCH] fix: decouple columns in projection and prune (#3253) Signed-off-by: Ruihang Xia --- src/mito2/src/sst/parquet/reader.rs | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/src/mito2/src/sst/parquet/reader.rs b/src/mito2/src/sst/parquet/reader.rs index 837d708a0e..06b6442917 100644 --- a/src/mito2/src/sst/parquet/reader.rs +++ b/src/mito2/src/sst/parquet/reader.rs @@ -311,14 +311,11 @@ impl ParquetReaderBuilder { // Prunes row groups by min-max index. if let Some(predicate) = &self.predicate { let region_meta = read_format.metadata(); - let column_ids = match &self.projection { - Some(ids) => ids.iter().cloned().collect(), - None => region_meta - .column_metadatas - .iter() - .map(|c| c.column_id) - .collect(), - }; + let column_ids = region_meta + .column_metadatas + .iter() + .map(|c| c.column_id) + .collect(); let row_groups = row_group_ids .iter()