org.apache.calcite.adapter.druid

Class DruidRules.DruidAggregateProjectRule

java.lang.Object

org.apache.calcite.plan.RelOptRule

org.apache.calcite.adapter.druid.DruidRules.DruidAggregateProjectRule

Enclosing class:

DruidRules

public static class DruidRules.DruidAggregateProjectRule
extends RelOptRule

Rule to push an Aggregate and Project into a DruidQuery.

Field Summary

Fields inherited from class org.apache.calcite.plan.RelOptRule

description, operands, relBuilderFactory

Constructor Summary

Constructors

Constructor and Description
DruidAggregateProjectRule(RelBuilderFactory relBuilderFactory) Creates a DruidAggregateProjectRule.

Method Summary

Modifier and Type	Method and Description
private static boolean	allAggregatesHaveFilters(List<AggregateCall> calls)
private static List<RelNode>	constructNewNodes(List<RelNode> oldNodes, boolean addFilter, int startIndex, RelNode filter, RelNode... trailingNodes) Returns a new List of RelNodes in the order of the given order of the oldNodes, the given Filter, and any extra nodes.
private static List<Integer>	getFilterRefs(List<AggregateCall> calls)
private Set<Integer>	getUniqueFilterRefs(List<AggregateCall> calls) Returns an array of unique filter references from the given list of AggregateCall
void	onMatch(RelOptRuleCall call)
private DruidQuery	optimizeFilteredAggregations(RelOptRuleCall call, DruidQuery query, Project project, Aggregate aggregate) Attempts to optimize any aggregations with filters in the DruidQuery.
private static RexNode	stripFilter(RexNode node)

Methods inherited from class org.apache.calcite.plan.RelOptRule

any, convert, convert, convertList, convertOperand, equals, equals, getOperand, getOperands, getOutConvention, getOutTrait, hashCode, matches, none, operand, operand, operand, operand, operand, some, toString, unordered

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

DruidAggregateProjectRule

public DruidAggregateProjectRule(RelBuilderFactory relBuilderFactory)

Creates a DruidAggregateProjectRule.

Parameters:

relBuilderFactory - Builder for relational expressions

Method Detail

onMatch

public void onMatch(RelOptRuleCall call)

Specified by:

onMatch in class RelOptRule

getUniqueFilterRefs

private Set<Integer> getUniqueFilterRefs(List<AggregateCall> calls)

Returns an array of unique filter references from the given list of AggregateCall

optimizeFilteredAggregations

private DruidQuery optimizeFilteredAggregations(RelOptRuleCall call,
                                                DruidQuery query,
                                                Project project,
                                                Aggregate aggregate)

Attempts to optimize any aggregations with filters in the DruidQuery. Uses the following steps:

1. Tries to abstract common filters out into the "filter" field;
2. Eliminates expressions that are always true or always false when possible;
3. ANDs aggregate filters together with the outer filter to allow for pruning of data.

Should be called before pushing both the aggregate and project into Druid. Assumes that at least one aggregate call has a filter attached to it.

allAggregatesHaveFilters

private static boolean allAggregatesHaveFilters(List<AggregateCall> calls)

constructNewNodes

private static List<RelNode> constructNewNodes(List<RelNode> oldNodes,
                                               boolean addFilter,
                                               int startIndex,
                                               RelNode filter,
                                               RelNode... trailingNodes)

Returns a new List of RelNodes in the order of the given order of the oldNodes, the given Filter, and any extra nodes.

stripFilter

private static RexNode stripFilter(RexNode node)

getFilterRefs

private static List<Integer> getFilterRefs(List<AggregateCall> calls)