Add semijoin join delete tablewriter and table finish node to connect…

…or optimizer
prestodb · Dec 2, 2024 · 99757d1 · 99757d1
1 parent 6cbc7c0
commit 99757d1
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 6 deletions.
diff --git a/...c/main/java/com/facebook/presto/sql/planner/optimizations/ApplyConnectorOptimization.java b/...c/main/java/com/facebook/presto/sql/planner/optimizations/ApplyConnectorOptimization.java
@@ -23,17 +23,22 @@
 import com.facebook.presto.spi.plan.CteConsumerNode;
 import com.facebook.presto.spi.plan.CteProducerNode;
 import com.facebook.presto.spi.plan.CteReferenceNode;
+import com.facebook.presto.spi.plan.DeleteNode;
 import com.facebook.presto.spi.plan.DistinctLimitNode;
 import com.facebook.presto.spi.plan.ExceptNode;
 import com.facebook.presto.spi.plan.FilterNode;
 import com.facebook.presto.spi.plan.IntersectNode;
+import com.facebook.presto.spi.plan.JoinNode;
 import com.facebook.presto.spi.plan.LimitNode;
 import com.facebook.presto.spi.plan.MarkDistinctNode;
 import com.facebook.presto.spi.plan.PlanNode;
 import com.facebook.presto.spi.plan.PlanNodeIdAllocator;
 import com.facebook.presto.spi.plan.ProjectNode;
+import com.facebook.presto.spi.plan.SemiJoinNode;
 import com.facebook.presto.spi.plan.SortNode;
+import com.facebook.presto.spi.plan.TableFinishNode;
 import com.facebook.presto.spi.plan.TableScanNode;
+import com.facebook.presto.spi.plan.TableWriterNode;
 import com.facebook.presto.spi.plan.TopNNode;
 import com.facebook.presto.spi.plan.UnionNode;
 import com.facebook.presto.spi.plan.ValuesNode;
@@ -78,7 +83,12 @@ public class ApplyConnectorOptimization
             MarkDistinctNode.class,
             UnionNode.class,
             IntersectNode.class,
-            ExceptNode.class);
+            ExceptNode.class,
+            SemiJoinNode.class,
+            JoinNode.class,
+            TableWriterNode.class,
+            TableFinishNode.class,
+            DeleteNode.class);
 
     // for a leaf node that does not belong to any connector (e.g., ValuesNode)
     private static final ConnectorId EMPTY_CONNECTOR_ID = new ConnectorId("$internal$" + ApplyConnectorOptimization.class + "_CONNECTOR");

diff --git a/presto-spi/src/main/java/com/facebook/presto/spi/plan/DeleteNode.java b/presto-spi/src/main/java/com/facebook/presto/spi/plan/DeleteNode.java
@@ -29,7 +29,7 @@
 import static java.util.Objects.requireNonNull;
 
 @Immutable
-public class DeleteNode
+public final class DeleteNode
         extends PlanNode
 {
     private final PlanNode source;

diff --git a/presto-spi/src/main/java/com/facebook/presto/spi/plan/JoinNode.java b/presto-spi/src/main/java/com/facebook/presto/spi/plan/JoinNode.java
@@ -46,7 +46,7 @@
 import static java.util.Objects.requireNonNull;
 
 @Immutable
-public class JoinNode
+public final class JoinNode
         extends AbstractJoinNode
 {
     private final JoinType type;

diff --git a/presto-spi/src/main/java/com/facebook/presto/spi/plan/SemiJoinNode.java b/presto-spi/src/main/java/com/facebook/presto/spi/plan/SemiJoinNode.java
@@ -30,7 +30,7 @@
 import static java.util.Objects.requireNonNull;
 
 @Immutable
-public class SemiJoinNode
+public final class SemiJoinNode
         extends AbstractJoinNode
 {
     private final PlanNode source;

diff --git a/presto-spi/src/main/java/com/facebook/presto/spi/plan/TableFinishNode.java b/presto-spi/src/main/java/com/facebook/presto/spi/plan/TableFinishNode.java
@@ -29,7 +29,7 @@
 import static java.util.Objects.requireNonNull;
 
 @Immutable
-public class TableFinishNode
+public final class TableFinishNode
         extends PlanNode
 {
     private final PlanNode source;

diff --git a/presto-spi/src/main/java/com/facebook/presto/spi/plan/TableWriterNode.java b/presto-spi/src/main/java/com/facebook/presto/spi/plan/TableWriterNode.java
@@ -39,7 +39,7 @@
 import static java.util.Objects.requireNonNull;
 
 @Immutable
-public class TableWriterNode
+public final class TableWriterNode
         extends PlanNode
 {
     private final PlanNode source;