Push down dereference expression

Cherry-pick of trinodb/trino@69ef682 Co-authored-by: qqibrow <[email protected]> Co-authored-by: Pratham Desai <[email protected]>
Intel-bigdata · Aug 1, 2020 · d45962c · d45962c
1 parent a4f14e4
commit d45962c
Show file tree

Hide file tree

Showing 5 changed files with 859 additions and 1 deletion.
diff --git a/presto-main/src/main/java/com/facebook/presto/sql/planner/PlanOptimizers.java b/presto-main/src/main/java/com/facebook/presto/sql/planner/PlanOptimizers.java
@@ -71,6 +71,7 @@
 import com.facebook.presto.sql.planner.iterative.rule.PruneValuesColumns;
 import com.facebook.presto.sql.planner.iterative.rule.PruneWindowColumns;
 import com.facebook.presto.sql.planner.iterative.rule.PushAggregationThroughOuterJoin;
+import com.facebook.presto.sql.planner.iterative.rule.PushDownDereferences;
 import com.facebook.presto.sql.planner.iterative.rule.PushLimitThroughMarkDistinct;
 import com.facebook.presto.sql.planner.iterative.rule.PushLimitThroughOuterJoin;
 import com.facebook.presto.sql.planner.iterative.rule.PushLimitThroughProject;
@@ -365,7 +366,14 @@ public PlanOptimizers(
                                 new InlineProjections(metadata.getFunctionManager()),
                                 new RemoveRedundantIdentityProjections(),
                                 new TransformCorrelatedSingleRowSubqueryToProject())),
-                new CheckSubqueryNodesAreRewritten());
+                new CheckSubqueryNodesAreRewritten(),
+                new IterativeOptimizer(
+                        ruleStats,
+                        statsCalculator,
+                        estimatedExchangesCostCalculator,
+                        ImmutableSet.<Rule<?>>builder()
+                                .addAll(new PushDownDereferences(metadata, sqlParser).rules())
+                                .build()));
 
         // TODO: move this before optimization if possible!!
         // Replace all expressions with row expressions