From 9710719b9d0f2e90a689f57494c4ca0fc6c930c0 Mon Sep 17 00:00:00 2001
From: zwang28 <84491488@qq.com>
Date: Tue, 9 Jul 2024 15:08:24 +0800
Subject: [PATCH] fix: include as_of in index

---
 src/frontend/src/optimizer/rule/index_selection_rule.rs | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/src/frontend/src/optimizer/rule/index_selection_rule.rs b/src/frontend/src/optimizer/rule/index_selection_rule.rs
index d47ad04de2763..b3f33efc0441d 100644
--- a/src/frontend/src/optimizer/rule/index_selection_rule.rs
+++ b/src/frontend/src/optimizer/rule/index_selection_rule.rs
@@ -95,9 +95,6 @@ impl Rule for IndexSelectionRule {
         if indexes.is_empty() {
             return None;
         }
-        if logical_scan.as_of().is_some() {
-            return None;
-        }
         let primary_table_row_size = TableScanIoEstimator::estimate_row_size(logical_scan);
         let primary_cost = min(
             self.estimate_table_scan_cost(logical_scan, primary_table_row_size),
@@ -230,7 +227,7 @@ impl IndexSelectionRule {
             index.index_table.clone(),
             vec![],
             logical_scan.ctx(),
-            None,
+            logical_scan.as_of().clone(),
             index.index_table.cardinality,
         );
 
@@ -239,7 +236,7 @@ impl IndexSelectionRule {
             index.primary_table.clone(),
             vec![],
             logical_scan.ctx(),
-            None,
+            logical_scan.as_of().clone(),
             index.primary_table.cardinality,
         );
 
@@ -338,7 +335,7 @@ impl IndexSelectionRule {
             logical_scan.table_catalog(),
             vec![],
             logical_scan.ctx(),
-            None,
+            logical_scan.as_of().clone(),
             logical_scan.table_cardinality(),
         );