From e10b73243a90dce5c988a7ecb688a0694b2e91cd Mon Sep 17 00:00:00 2001
From: evenyag <realevenyag@gmail.com>
Date: Thu, 29 Aug 2024 00:04:48 +0800
Subject: [PATCH] chore: avoid get column

---
 src/mito2/src/sst/parquet/row_group.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/mito2/src/sst/parquet/row_group.rs b/src/mito2/src/sst/parquet/row_group.rs
index 10ee706e1df5..8a480bd86d4c 100644
--- a/src/mito2/src/sst/parquet/row_group.rs
+++ b/src/mito2/src/sst/parquet/row_group.rs
@@ -202,8 +202,8 @@ impl<'a> InMemoryRowGroup<'a> {
                     continue;
                 };
 
+                let column = self.metadata.column(idx);
                 if let Some(cache) = &self.cache_manager {
-                    let column = self.metadata.column(idx);
                     // Put the page to the cache if we don't cache the whole row group.
                     if !cache_row_group_pages(column) {
                         let page_key = PageKey::Compressed {
@@ -218,7 +218,7 @@ impl<'a> InMemoryRowGroup<'a> {
                 }
 
                 *chunk = Some(Arc::new(ColumnChunkData::Dense {
-                    offset: self.metadata.column(idx).byte_range().0 as usize,
+                    offset: column.byte_range().0 as usize,
                     data,
                 }));
             }
@@ -260,7 +260,7 @@ impl<'a> InMemoryRowGroup<'a> {
 
                     *chunk = cache.get_pages(&page_key).map(|page_value| {
                         Arc::new(ColumnChunkData::Dense {
-                            offset: self.metadata.column(idx).byte_range().0 as usize,
+                            offset: column.byte_range().0 as usize,
                             data: page_value.compressed.clone(),
                         })
                     });