database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[Feature](Nereids) Support CaseWhen with subquery (#16385)

Browse Source 
Co-authored-by: jianghaochen <jianghaochen@meituan.com>
This commit is contained in:
zhengshiJ
2023-02-03 18:20:47 +08:00
committed by GitHub
parent 3891083474
commit 929b31bd3c
4 changed files with 59 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fe/fe-core/src/main/java/org/apache/doris/nereids/rules/RuleType.java
View File

@ -103,6 +103,7 @@ public enum RuleType {
// subquery analyze
ANALYZE_FILTER_SUBQUERY(RuleTypeClass.REWRITE),
ANALYZE_PROJECT_SUBQUERY(RuleTypeClass.REWRITE),
// subquery rewrite rule
ELIMINATE_LIMIT_UNDER_APPLY(RuleTypeClass.REWRITE),
ELIMINATE_SORT_UNDER_APPLY(RuleTypeClass.REWRITE),

32
fe/fe-core/src/main/java/org/apache/doris/nereids/rules/analysis/AnalyzeSubquery.java
View File

@ -20,6 +20,8 @@ package org.apache.doris.nereids.rules.analysis;
import org.apache.doris.nereids.CascadesContext;
import org.apache.doris.nereids.rules.Rule;
import org.apache.doris.nereids.rules.RuleType;
import org.apache.doris.nereids.trees.expressions.Alias;
import org.apache.doris.nereids.trees.expressions.CaseWhen;
import org.apache.doris.nereids.trees.expressions.Exists;
import org.apache.doris.nereids.trees.expressions.Expression;
import org.apache.doris.nereids.trees.expressions.InSubquery;
@ -38,6 +40,7 @@ import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Optional;
import java.util.Set;
@ -67,7 +70,30 @@ public class AnalyzeSubquery implements AnalysisRuleFactory {
subqueryExprs, filter.child(), ctx.cascadesContext
));
})
)
),
RuleType.ANALYZE_PROJECT_SUBQUERY.build(
logicalProject().thenApply(ctx -> {
LogicalProject<GroupPlan> project = ctx.root;
Set<SubqueryExpr> subqueryExprs = new HashSet<>();
project.getProjects().stream()
.filter(Alias.class::isInstance)
.map(Alias.class::cast)
.filter(alias -> alias.child() instanceof CaseWhen)
.forEach(alias -> alias.child().children().stream()
.forEach(e ->
subqueryExprs.addAll(e.collect(SubqueryExpr.class::isInstance))));
if (subqueryExprs.isEmpty()) {
return project;
}
return new LogicalProject(project.getProjects().stream()
.map(p -> p.withChildren(new ReplaceSubquery().replace(p)))
.collect(ImmutableList.toImmutableList()),
analyzedSubquery(
subqueryExprs, project.child(), ctx.cascadesContext
));
})
)
);
}
@ -117,6 +143,10 @@ public class AnalyzeSubquery implements AnalysisRuleFactory {
.collect(ImmutableSet.toImmutableSet());
}
public Expression replace(Expression expressions) {
return expressions.accept(this, null);
}
@Override
public Expression visitExistsSubquery(Exists exists, Void context) {
return BooleanLiteral.TRUE;