Add delete node processing in subfield prune optimizer

prestodb · Dec 9, 2024 · cfb487c · cfb487c
1 parent 30686ee
commit cfb487c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/...o-main/src/main/java/com/facebook/presto/sql/planner/optimizations/PushdownSubfields.java b/...o-main/src/main/java/com/facebook/presto/sql/planner/optimizations/PushdownSubfields.java
@@ -35,6 +35,7 @@
 import com.facebook.presto.spi.function.StandardFunctionResolution;
 import com.facebook.presto.spi.plan.AggregationNode;
 import com.facebook.presto.spi.plan.CteProducerNode;
+import com.facebook.presto.spi.plan.DeleteNode;
 import com.facebook.presto.spi.plan.DistinctLimitNode;
 import com.facebook.presto.spi.plan.EquiJoinClause;
 import com.facebook.presto.spi.plan.FilterNode;
@@ -408,6 +409,16 @@ public PlanNode visitTableWriter(TableWriterNode node, RewriteContext<Context> c
             return context.defaultRewrite(node, context.get());
         }
 
+        @Override
+        public PlanNode visitDelete(DeleteNode node, RewriteContext<Context> context)
+        {
+            if (node.getInputDistribution().isPresent()) {
+                context.get().variables.addAll(node.getInputDistribution().get().getInputVariables());
+            }
+            context.get().variables.add(node.getRowId());
+            return context.defaultRewrite(node, context.get());
+        }
+
         @Override
         public PlanNode visitTopN(TopNNode node, RewriteContext<Context> context)
         {