This is an automated email from the ASF dual-hosted git repository.
jakevin pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/master by this push:
new 66bd2a4862 [test](Nereids) add test push down filter (#24250)
66bd2a4862 is described below
commit 66bd2a4862cbb288b36c347c152cc6db5c8ec044
Author: 谢健 <[email protected]>
AuthorDate: Thu Sep 14 22:13:41 2023 +0800
[test](Nereids) add test push down filter (#24250)
Add test for pushDownFilterThroughProject
---
.../rewrite/PushdowFilterThroughProjectTest.java | 112 +++++++++++++++++++++
1 file changed, 112 insertions(+)
diff --git
a/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/PushdowFilterThroughProjectTest.java
b/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/PushdowFilterThroughProjectTest.java
new file mode 100644
index 0000000000..6118a98ebd
--- /dev/null
+++
b/fe/fe-core/src/test/java/org/apache/doris/nereids/rules/rewrite/PushdowFilterThroughProjectTest.java
@@ -0,0 +1,112 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements. See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership. The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License. You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied. See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package org.apache.doris.nereids.rules.rewrite;
+
+import org.apache.doris.nereids.properties.OrderKey;
+import org.apache.doris.nereids.trees.expressions.Alias;
+import org.apache.doris.nereids.trees.expressions.ExprId;
+import org.apache.doris.nereids.trees.expressions.Expression;
+import org.apache.doris.nereids.trees.expressions.IsNull;
+import org.apache.doris.nereids.trees.expressions.OrderExpression;
+import org.apache.doris.nereids.trees.expressions.StatementScopeIdGenerator;
+import org.apache.doris.nereids.trees.expressions.WindowExpression;
+import org.apache.doris.nereids.trees.expressions.functions.window.Rank;
+import org.apache.doris.nereids.trees.plans.logical.LogicalOlapScan;
+import org.apache.doris.nereids.trees.plans.logical.LogicalPlan;
+import org.apache.doris.nereids.util.LogicalPlanBuilder;
+import org.apache.doris.nereids.util.MemoPatternMatchSupported;
+import org.apache.doris.nereids.util.MemoTestUtils;
+import org.apache.doris.nereids.util.PlanChecker;
+import org.apache.doris.nereids.util.PlanConstructor;
+import org.apache.doris.qe.ConnectContext;
+
+import com.google.common.collect.ImmutableList;
+import org.junit.jupiter.api.Test;
+
+class PushdowFilterThroughProjectTest implements MemoPatternMatchSupported {
+ private final LogicalOlapScan scan = new
LogicalOlapScan(StatementScopeIdGenerator.newRelationId(),
+ PlanConstructor.student,
+ ImmutableList.of(""));
+
+ @Test
+ void pushDownFilterThroughProject() {
+ ConnectContext context = MemoTestUtils.createConnectContext();
+ LogicalPlan plan = new LogicalPlanBuilder(scan)
+ .projectExprs(ImmutableList.of(new Alias(new ExprId(1),
scan.getOutput().get(0), "a")))
+ .projectExprs(ImmutableList.of(new Alias(
+ new ExprId(1), new Alias(new ExprId(1),
scan.getOutput().get(0), "a").toSlot(),
+ "b")))
+ .filter(new IsNull((new Alias(
+ new ExprId(1), new Alias(new ExprId(1),
scan.getOutput().get(0), "a").toSlot(),
+ "b")).toSlot()))
+ .build();
+
+ PlanChecker.from(context, plan)
+ .applyTopDown(new PushdownFilterThroughProject())
+ .matches(logicalFilter().when(f ->
+ f.getPredicate().toSql().equals("id IS NULL")
+ ));
+ }
+
+ @Test
+ void notPushDownFilterThroughWindow() {
+ Expression window = new WindowExpression(new Rank(),
ImmutableList.of(scan.getOutput().get(0)),
+ ImmutableList.of(new OrderExpression(new
OrderKey(scan.getOutput().get(0), true, true))));
+ Alias alias = new Alias(new ExprId(1), window, "window");
+ ConnectContext context = MemoTestUtils.createConnectContext();
+
+ // filter -> limit -> project(windows)
+ LogicalPlan plan = new LogicalPlanBuilder(scan)
+ .projectExprs(ImmutableList.of(alias))
+ .filter(new IsNull(alias.toSlot()))
+ .build();
+
+ PlanChecker.from(context, plan)
+ .applyTopDown(new PushdownFilterThroughProject())
+ .matches(logicalFilter(logicalProject()));
+
+ // filter -> limit -> project(windows)
+ plan = new LogicalPlanBuilder(scan)
+ .projectExprs(ImmutableList.of(alias))
+ .limit(1)
+ .filter(new IsNull(alias.toSlot()))
+ .build();
+
+ PlanChecker.from(context, plan)
+ .applyTopDown(new PushdownFilterThroughProject())
+ .matches(logicalFilter(logicalLimit(logicalProject())));
+ }
+
+ @Test
+ void pushDownFilterThroughLimit() {
+ ConnectContext context = MemoTestUtils.createConnectContext();
+ // filter -> limit -> project
+ Alias alias = new Alias(new ExprId(1), scan.getOutput().get(0), "a");
+ LogicalPlan plan = new LogicalPlanBuilder(scan)
+ .projectExprs(ImmutableList.of(alias))
+ .limit(1)
+ .filter(new IsNull(alias.toSlot()))
+ .build();
+
+ PlanChecker.from(context, plan)
+ .applyTopDown(new PushdownFilterThroughProject())
+ .matches(logicalProject(logicalFilter(logicalLimit()).when(f ->
+ f.getPredicate().toSql().equals("id IS NULL"))));
+ }
+}
+
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
