[feature](Nereids): eliminate sort under subquery (#26993)

This commit is contained in:
谢健
2023-11-16 10:30:28 +08:00
committed by GitHub
parent a54cfb7558
commit f3ee6dd55a
4 changed files with 61 additions and 3 deletions

View File

@ -62,6 +62,7 @@ import org.apache.doris.nereids.rules.rewrite.EliminateNotNull;
import org.apache.doris.nereids.rules.rewrite.EliminateNullAwareLeftAntiJoin;
import org.apache.doris.nereids.rules.rewrite.EliminateOrderByConstant;
import org.apache.doris.nereids.rules.rewrite.EliminateSort;
import org.apache.doris.nereids.rules.rewrite.EliminateSortUnderSubquery;
import org.apache.doris.nereids.rules.rewrite.EliminateUnnecessaryProject;
import org.apache.doris.nereids.rules.rewrite.EnsureProjectOnTopJoin;
import org.apache.doris.nereids.rules.rewrite.ExtractAndNormalizeWindowExpression;
@ -125,6 +126,7 @@ public class Rewriter extends AbstractBatchJobExecutor {
topic("Plan Normalization",
topDown(
new EliminateOrderByConstant(),
new EliminateSortUnderSubquery(),
new EliminateGroupByConstant(),
// MergeProjects depends on this rule
new LogicalSubQueryAliasToLogicalProject(),

View File

@ -0,0 +1,33 @@
// 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.rules.Rule;
import org.apache.doris.nereids.rules.RuleType;
/**
* SELECT * FROM lineorder ORDER BY 'f' -> SELECT * FROM lineorder
*/
public class EliminateSortUnderSubquery extends OneRewriteRuleFactory {
@Override
public Rule build() {
return logicalSubQueryAlias(logicalSort())
.then(subq -> subq.withChildren(subq.child().child(0)))
.toRule(RuleType.ELIMINATE_ORDER_BY_CONSTANT);
}
}

View File

@ -26,7 +26,7 @@ import org.junit.jupiter.api.Test;
/**
* column prune ut.
*/
public class EliminateSortTest extends TestWithFeService implements MemoPatternMatchSupported {
class EliminateSortTest extends TestWithFeService implements MemoPatternMatchSupported {
@Override
protected void runBeforeAll() throws Exception {
createDatabase("test");
@ -37,7 +37,7 @@ public class EliminateSortTest extends TestWithFeService implements MemoPatternM
}
@Test
public void test() {
void test() {
PlanChecker.from(connectContext)
.analyze("select * from student order by id")
.rewrite()
@ -47,4 +47,27 @@ public class EliminateSortTest extends TestWithFeService implements MemoPatternM
.rewrite()
.nonMatch(logicalSort());
}
@Test
void testSortLimit() {
PlanChecker.from(connectContext)
.analyze("select count(*) from (select * from student order by id) t limit 1")
.rewrite()
.nonMatch(logicalTopN());
PlanChecker.from(connectContext)
.analyze("select count(*) from (select * from student order by id limit 1) t")
.rewrite()
.matches(logicalTopN());
PlanChecker.from(connectContext)
.analyze("select count(*) from "
+ "(select * from student order by id limit 1) t1 left join student t2 on t1.id = t2.id")
.rewrite()
.matches(logicalTopN());
PlanChecker.from(connectContext)
.analyze("select count(*) from "
+ "(select * from student order by id) t1 left join student t2 on t1.id = t2.id limit 1")
.rewrite()
.nonMatch(logicalTopN());
}
}

View File

@ -115,7 +115,7 @@ suite("test_group_concat") {
"""
sql """create view if not exists test_view as SELECT b1, group_concat(cast(abs(b3) as varchar) order by abs(b2) desc, b3 desc) FROM table_group_concat group by b1 order by b1;"""
qt_select_group_concat_order_by_desc4 """
order_qt_select_group_concat_order_by_desc4 """
select * from test_view;
"""
sql """drop view if exists test_view"""