/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.facebook.presto.sql.planner;
import com.facebook.presto.sql.planner.plan.AggregationNode;
import com.facebook.presto.sql.planner.plan.ApplyNode;
import com.facebook.presto.sql.planner.plan.FilterNode;
import com.facebook.presto.sql.planner.plan.JoinNode;
import com.facebook.presto.sql.planner.plan.PlanNode;
import com.facebook.presto.sql.planner.plan.ProjectNode;
import com.facebook.presto.sql.planner.plan.TableScanNode;
import com.facebook.presto.sql.planner.plan.ValuesNode;
import com.facebook.presto.sql.tree.Expression;
import com.google.common.collect.ImmutableList;
import java.util.List;
public class ExpressionExtractor
{
public static List<Expression> extractExpressions(PlanNode plan)
{
ImmutableList.Builder<Expression> expressionsBuilder = ImmutableList.builder();
plan.accept(new Visitor(true), expressionsBuilder);
return expressionsBuilder.build();
}
public static List<Expression> extractExpressionsNonRecursive(PlanNode plan)
{
ImmutableList.Builder<Expression> expressionsBuilder = ImmutableList.builder();
plan.accept(new Visitor(false), expressionsBuilder);
return expressionsBuilder.build();
}
private ExpressionExtractor()
{
}
private static class Visitor
extends SimplePlanVisitor<ImmutableList.Builder<Expression>>
{
private final boolean recursive;
public Visitor(boolean recursive)
{
this.recursive = recursive;
}
@Override
protected Void visitPlan(PlanNode node, ImmutableList.Builder<Expression> context)
{
if (recursive) {
return super.visitPlan(node, context);
}
return null;
}
@Override
public Void visitAggregation(AggregationNode node, ImmutableList.Builder<Expression> context)
{
node.getAssignments().values()
.forEach(aggregation -> context.add(aggregation.getCall()));
return super.visitAggregation(node, context);
}
@Override
public Void visitFilter(FilterNode node, ImmutableList.Builder<Expression> context)
{
context.add(node.getPredicate());
return super.visitFilter(node, context);
}
@Override
public Void visitProject(ProjectNode node, ImmutableList.Builder<Expression> context)
{
context.addAll(node.getAssignments().getExpressions());
return super.visitProject(node, context);
}
@Override
public Void visitTableScan(TableScanNode node, ImmutableList.Builder<Expression> context)
{
if (node.getOriginalConstraint() != null) {
context.add(node.getOriginalConstraint());
}
return super.visitTableScan(node, context);
}
@Override
public Void visitJoin(JoinNode node, ImmutableList.Builder<Expression> context)
{
node.getFilter().ifPresent(context::add);
return super.visitJoin(node, context);
}
@Override
public Void visitValues(ValuesNode node, ImmutableList.Builder<Expression> context)
{
node.getRows().forEach(context::addAll);
return super.visitValues(node, context);
}
@Override
public Void visitApply(ApplyNode node, ImmutableList.Builder<Expression> context)
{
context.addAll(node.getSubqueryAssignments().getExpressions());
return super.visitApply(node, context);
}
}
}