@Override
public ActualProperties visitAggregation(
AggregationNode node, List<ActualProperties> inputProperties) {
ActualProperties properties = Iterables.getOnlyElement(inputProperties);
ActualProperties translated =
properties.translate(
symbol ->
node.getGroupingKeys().contains(symbol)
? Optional.of(symbol)
: Optional.<Symbol>empty());
return ActualProperties.builderFrom(translated)
.local(LocalProperties.grouped(node.getGroupingKeys()))
.build();
}
@Override
public Map<Symbol, Symbol> visitAggregation(AggregationNode node, Set<Symbol> lookupSymbols) {
Set<Symbol> groupByLookupSymbols =
lookupSymbols
.stream()
.filter(node.getGroupingKeys()::contains)
.collect(toImmutableSet());
checkState(
!groupByLookupSymbols.isEmpty(),
"No lookup symbols were able to pass through the aggregation group by");
return node.getSource().accept(this, groupByLookupSymbols);
}
@Override
public PlanNode visitAggregation(AggregationNode node, RewriteContext<Context> context) {
// Lookup symbols can only be passed through if they are part of the group by columns
Set<Symbol> groupByLookupSymbols =
context
.get()
.getLookupSymbols()
.stream()
.filter(node.getGroupingKeys()::contains)
.collect(toImmutableSet());
if (groupByLookupSymbols.isEmpty()) {
return node;
}
return context.defaultRewrite(
node, new Context(groupByLookupSymbols, context.get().getSuccess()));
}