rebase code

apache · Jul 6, 2023 · 4f5fc89 · 4f5fc89
1 parent 6543d09
commit 4f5fc89
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/...re/src/main/java/org/apache/doris/nereids/rules/rewrite/BuildCTEAnchorAndCTEProducer.java b/...re/src/main/java/org/apache/doris/nereids/rules/rewrite/BuildCTEAnchorAndCTEProducer.java
@@ -48,13 +48,18 @@ private LogicalPlan rewrite(LogicalPlan p, CascadesContext cascadesContext) {
  }
  LogicalCTE logicalCTE = (LogicalCTE) p;
  LogicalPlan child = (LogicalPlan) logicalCTE.child();
- for (int i = logicalCTE.getAliasQueries().size() - 1; i >= 0; i--) {
- LogicalSubQueryAlias s = (LogicalSubQueryAlias) logicalCTE.getAliasQueries().get(i);
- CTEId id = logicalCTE.findCTEId(s.getAlias());
- if (cascadesContext.cteReferencedCount(id)
- <= ConnectContext.get().getSessionVariable().inlineCTEReferencedThreshold
- || !ConnectContext.get().getSessionVariable().getEnablePipelineEngine()) {
- continue;
+ if (!(child instanceof LogicalEmptyRelation)) {
+ for (int i = logicalCTE.getAliasQueries().size() - 1; i >= 0; i--) {
+ LogicalSubQueryAlias s = (LogicalSubQueryAlias) logicalCTE.getAliasQueries().get(i);
+ CTEId id = logicalCTE.findCTEId(s.getAlias());
+ if (cascadesContext.cteReferencedCount(id)
+ <= ConnectContext.get().getSessionVariable().inlineCTEReferencedThreshold
+ || !ConnectContext.get().getSessionVariable().getEnablePipelineEngine()) {
+ continue;
+ }
+ LogicalCTEProducer logicalCTEProducer = new LogicalCTEProducer(
+ rewrite((LogicalPlan) s.child(), cascadesContext), id);
+ child = new LogicalCTEAnchor(logicalCTEProducer, child, id);
  }
  }
  return child;