fix(query): fix lazy columns missed in constant table scan (#17258)

* fix(query): fix lazy columns missed in constant table scan

Signed-off-by: damon <[email protected]>

* clear lazy columns when rewriting false filter to be empty scan

Signed-off-by: damon <[email protected]>

---------

Signed-off-by: damon <[email protected]>

Author: Damon07

src/query/sql/src/planner/metadata.rs

@@ -170,6 +170,10 @@ impl Metadata {
         self.lazy_columns.extend(indices);
     }
 
+    pub fn clear_lazy_columns(&mut self) {
+        self.lazy_columns.clear();
+    }
+
     pub fn add_non_lazy_columns(&mut self, indices: HashSet<usize>) {
         debug_assert!(indices.iter().all(|i| *i < self.columns.len()));
         self.non_lazy_columns.extend(indices);

src/query/sql/src/planner/optimizer/rule/rewrite/rule_eliminate_filter.rs

@@ -77,6 +77,12 @@ impl Rule for RuleEliminateFilter {
                 .derive_relational_prop(&RelExpr::with_s_expr(s_expr))?
                 .output_columns
                 .clone();
+
+            {
+                let mut metadata = self.metadata.write();
+                metadata.clear_lazy_columns();
+            }
+
             let metadata = self.metadata.read();
             let mut fields = Vec::with_capacity(output_columns.len());
 

tests/sqllogictests/suites/query/issues/issue_17158.test

@@ -0,0 +1,8 @@
+statement ok
+create table tt2 (c0 bool, c1 int);
+
+statement ok
+insert into tt2 values(true, 1),(false, 2),(true, 3);
+
+statement ok
+select null, c0, 30, c1 from tt2 where false order by c0 LIMIT 3 OFFSET 0;