fix: incorrectly remove mark index

Author: xudong963

src/query/sql/src/planner/optimizer/rule/rewrite/filter_join/mark_join_to_semi_join.rs

@@ -32,11 +32,13 @@ pub fn convert_mark_to_semi_join(s_expr: &SExpr) -> Result<SExpr> {
     }
 
     let mark_index = join.marker_index.unwrap();
+    let mut find_mark_index = false;
 
     // remove mark index filter
     for (idx, predicate) in filter.predicates.iter().enumerate() {
         if let ScalarExpr::BoundColumnRef(col) = predicate {
             if col.column.index == mark_index {
+                find_mark_index = true;
                 filter.predicates.remove(idx);
                 break;
             }
@@ -51,6 +53,11 @@ pub fn convert_mark_to_semi_join(s_expr: &SExpr) -> Result<SExpr> {
         }
     }
 
+    if !find_mark_index {
+        // To be conservative, we do not convert
+        return Ok(s_expr.clone());
+    }
+
     join.join_type = match join.join_type {
         JoinType::LeftMark => JoinType::RightSemi,
         JoinType::RightMark => JoinType::LeftSemi,

tests/sqllogictests/suites/query/subquery.test

@@ -39,6 +39,11 @@ SELECT * FROM c WHERE EXISTS(SELECT * FROM o WHERE o.c_id=c.c_id) ORDER BY c_id
 4  TX
 6  FL
 
+query IT
+SELECT * FROM c WHERE EXISTS(SELECT * FROM o WHERE o.c_id=c.c_id) OR EXISTS(SELECT * FROM o where o.ship = o.bill) ORDER BY c_id
+----
+
+
 query IT
 SELECT * FROM c WHERE NOT EXISTS(SELECT * FROM o WHERE o.c_id=c.c_id) ORDER BY c_id
 ----