From a8012147ab52f43513580f17ae210a2dbb439318 Mon Sep 17 00:00:00 2001
From: Niwaka <61189782+NiwakaDev@users.noreply.github.com>
Date: Wed, 11 Dec 2024 22:46:23 +0900
Subject: [PATCH] feat: support push down IN filter (#5129)

* feat: support push down IN filter

* chore: move tests to prune.sql
---
 src/query/src/dist_plan/commutativity.rs      |  2 +-
 .../standalone/common/select/prune.result     | 26 +++++++++++++++++++
 .../cases/standalone/common/select/prune.sql  | 10 +++++++
 3 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/src/query/src/dist_plan/commutativity.rs b/src/query/src/dist_plan/commutativity.rs
index 8166400b8fbd..45378e532c0c 100644
--- a/src/query/src/dist_plan/commutativity.rs
+++ b/src/query/src/dist_plan/commutativity.rs
@@ -146,6 +146,7 @@ impl Categorizer {
             | Expr::Between(_)
             | Expr::Sort(_)
             | Expr::Exists(_)
+            | Expr::InList(_)
             | Expr::ScalarFunction(_) => Commutativity::Commutative,
 
             Expr::Like(_)
@@ -157,7 +158,6 @@ impl Categorizer {
             | Expr::TryCast(_)
             | Expr::AggregateFunction(_)
             | Expr::WindowFunction(_)
-            | Expr::InList(_)
             | Expr::InSubquery(_)
             | Expr::ScalarSubquery(_)
             | Expr::Wildcard { .. } => Commutativity::Unimplemented,
diff --git a/tests/cases/standalone/common/select/prune.result b/tests/cases/standalone/common/select/prune.result
index 13ddee5510d2..04282b6035a0 100644
--- a/tests/cases/standalone/common/select/prune.result
+++ b/tests/cases/standalone/common/select/prune.result
@@ -94,6 +94,32 @@ explain analyze select * from demo where idc='idc1';
 |_|_| Total rows: 2_|
 +-+-+-+
 
+SELECT * FROM demo where host in ('test1');
+
++-------------------------+-------+-------+------+-----------+
+| ts                      | value | host  | idc  | collector |
++-------------------------+-------+-------+------+-----------+
+| 1970-01-01T00:00:00.001 | 2.0   | test1 | idc1 | disk      |
++-------------------------+-------+-------+------+-----------+
+
+-- SQLNESS REPLACE (metrics.*) REDACTED
+-- SQLNESS REPLACE (RoundRobinBatch.*) REDACTED
+-- SQLNESS REPLACE (-+) -
+-- SQLNESS REPLACE (\s\s+) _
+-- SQLNESS REPLACE (peers.*) REDACTED
+-- SQLNESS REPLACE region=\d+\(\d+,\s+\d+\) region=REDACTED
+explain analyze SELECT * FROM demo where host in ('test1');
+
++-+-+-+
+| stage | node | plan_|
++-+-+-+
+| 0_| 0_|_MergeScanExec: REDACTED
+|_|_|_|
+| 1_| 0_|_SeqScan: region=REDACTED, partition_count=1 (1 memtable ranges, 0 file 0 ranges) REDACTED
+|_|_|_|
+|_|_| Total rows: 1_|
++-+-+-+
+
 drop table demo;
 
 Affected Rows: 0
diff --git a/tests/cases/standalone/common/select/prune.sql b/tests/cases/standalone/common/select/prune.sql
index e7fd643537a1..4b976cdb1c7d 100644
--- a/tests/cases/standalone/common/select/prune.sql
+++ b/tests/cases/standalone/common/select/prune.sql
@@ -27,4 +27,14 @@ select * from demo where collector='disk' order by ts;
 -- SQLNESS REPLACE region=\d+\(\d+,\s+\d+\) region=REDACTED
 explain analyze select * from demo where idc='idc1';
 
+SELECT * FROM demo where host in ('test1');
+
+-- SQLNESS REPLACE (metrics.*) REDACTED
+-- SQLNESS REPLACE (RoundRobinBatch.*) REDACTED
+-- SQLNESS REPLACE (-+) -
+-- SQLNESS REPLACE (\s\s+) _
+-- SQLNESS REPLACE (peers.*) REDACTED
+-- SQLNESS REPLACE region=\d+\(\d+,\s+\d+\) region=REDACTED
+explain analyze SELECT * FROM demo where host in ('test1');
+
 drop table demo;