org.apache.calcite.rel.rules

Class DateRangeRules.ExtractShuttle

java.lang.Object

org.apache.calcite.rex.RexShuttle

org.apache.calcite.rel.rules.DateRangeRules.ExtractShuttle

All Implemented Interfaces:

RexVisitor<RexNode>

Enclosing class:

DateRangeRules

static class DateRangeRules.ExtractShuttle
extends RexShuttle

Walks over an expression, replacing calls to EXTRACT, FLOOR and CEIL with date ranges.

Field Summary

Fields

Modifier and Type	Field and Description
private Deque<RexCall>	calls
private Map<String,com.google.common.collect.RangeSet<Calendar>>	operandRanges
private RexBuilder	rexBuilder
private org.apache.calcite.avatica.util.TimeUnitRange	timeUnit
private com.google.common.collect.ImmutableSortedSet<org.apache.calcite.avatica.util.TimeUnitRange>	timeUnitRanges
private String	timeZone

Constructor Summary

Constructors

Constructor and Description
ExtractShuttle(RexBuilder rexBuilder, org.apache.calcite.avatica.util.TimeUnitRange timeUnit, Map<String,com.google.common.collect.RangeSet<Calendar>> operandRanges, com.google.common.collect.ImmutableSortedSet<org.apache.calcite.avatica.util.TimeUnitRange> timeUnitRanges, String timeZone)

Method Summary

Modifier and Type	Method and Description
private boolean	canRewriteExtract(RexNode operand)
private Calendar	ceil(Calendar c, org.apache.calcite.avatica.util.TimeUnitRange timeUnit)
private com.google.common.collect.Range<Calendar>	ceilRange(org.apache.calcite.avatica.util.TimeUnitRange timeUnit, SqlKind comparison, Calendar c)
(package private) RexNode	compareExtract(SqlKind comparison, RexNode operand, RexLiteral literal)
private RexNode	compareFloorCeil(SqlKind comparison, RexNode operand, RexLiteral timeLiteral, org.apache.calcite.avatica.util.TimeUnitRange timeUnit, boolean floor)
private RexLiteral	dateTimeLiteral(RexBuilder rexBuilder, Calendar calendar, RexNode operand)
private Calendar	decrement(Calendar c, org.apache.calcite.avatica.util.TimeUnitRange timeUnit)
private com.google.common.collect.Range<Calendar>	extractRange(org.apache.calcite.avatica.util.TimeUnitRange timeUnit, SqlKind comparison, Calendar c)
private Calendar	floor(Calendar c, org.apache.calcite.avatica.util.TimeUnitRange timeUnit) Computes floor of a calendar to a given time unit.
private com.google.common.collect.Range<Calendar>	floorRange(org.apache.calcite.avatica.util.TimeUnitRange timeUnit, SqlKind comparison, Calendar c)
private Calendar	increment(Calendar c, org.apache.calcite.avatica.util.TimeUnitRange timeUnit)
(package private) boolean	isExtractCall(RexNode e)
(package private) boolean	isFloorCeilCall(RexNode e)
private static boolean	isValid(int v, org.apache.calcite.avatica.util.TimeUnitRange timeUnit)
private boolean	next(Calendar c, org.apache.calcite.avatica.util.TimeUnitRange timeUnit, int v, com.google.common.collect.Range<Calendar> r, boolean strict)
private Calendar	round(Calendar c, org.apache.calcite.avatica.util.TimeUnitRange timeUnit, boolean down) Returns a copy of a calendar, optionally rounded up to the next time unit.
private Calendar	timestampValue(RexLiteral timeLiteral)
private RexNode	toRex(RexNode operand, com.google.common.collect.Range<Calendar> r)
RexNode	visitCall(RexCall call)
protected List<RexNode>	visitList(List<? extends RexNode> exprs, boolean[] update) Visits each of a list of expressions and returns a list of the results.

Methods inherited from class org.apache.calcite.rex.RexShuttle

apply, apply, apply, mutate, visitArray, visitCorrelVariable, visitDynamicParam, visitFieldAccess, visitFieldCollations, visitInputRef, visitList, visitLiteral, visitLocalRef, visitOver, visitPatternFieldRef, visitRangeRef, visitSubQuery, visitTableInputRef, visitWindow

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

rexBuilder

private final RexBuilder rexBuilder

timeUnit

private final org.apache.calcite.avatica.util.TimeUnitRange timeUnit

operandRanges

private final Map<String,com.google.common.collect.RangeSet<Calendar>> operandRanges

calls

private final Deque<RexCall> calls

timeUnitRanges

private final com.google.common.collect.ImmutableSortedSet<org.apache.calcite.avatica.util.TimeUnitRange> timeUnitRanges

timeZone

private final String timeZone

Constructor Detail

ExtractShuttle

ExtractShuttle(RexBuilder rexBuilder,
               org.apache.calcite.avatica.util.TimeUnitRange timeUnit,
               Map<String,com.google.common.collect.RangeSet<Calendar>> operandRanges,
               com.google.common.collect.ImmutableSortedSet<org.apache.calcite.avatica.util.TimeUnitRange> timeUnitRanges,
               String timeZone)

Method Detail

visitCall

public RexNode visitCall(RexCall call)

Specified by:

visitCall in interface RexVisitor<RexNode>

Overrides:

visitCall in class RexShuttle

canRewriteExtract

private boolean canRewriteExtract(RexNode operand)

visitList

protected List<RexNode> visitList(List<? extends RexNode> exprs,
                                  boolean[] update)

Description copied from class: RexShuttle

Visits each of a list of expressions and returns a list of the results.

Overrides:

visitList in class RexShuttle

Parameters:

exprs - List of expressions

update - If not null, sets this to true if any of the expressions was modified

Returns:

Array of visited expressions

isExtractCall

boolean isExtractCall(RexNode e)

compareExtract

RexNode compareExtract(SqlKind comparison,
                       RexNode operand,
                       RexLiteral literal)

next

private boolean next(Calendar c,
                     org.apache.calcite.avatica.util.TimeUnitRange timeUnit,
                     int v,
                     com.google.common.collect.Range<Calendar> r,
                     boolean strict)

isValid

private static boolean isValid(int v,
                               org.apache.calcite.avatica.util.TimeUnitRange timeUnit)

toRex

private RexNode toRex(RexNode operand,
                      com.google.common.collect.Range<Calendar> r)

dateTimeLiteral

private RexLiteral dateTimeLiteral(RexBuilder rexBuilder,
                                   Calendar calendar,
                                   RexNode operand)

extractRange

private com.google.common.collect.Range<Calendar> extractRange(org.apache.calcite.avatica.util.TimeUnitRange timeUnit,
                                                               SqlKind comparison,
                                                               Calendar c)

round

private Calendar round(Calendar c,
                       org.apache.calcite.avatica.util.TimeUnitRange timeUnit,
                       boolean down)

Returns a copy of a calendar, optionally rounded up to the next time unit.

compareFloorCeil

private RexNode compareFloorCeil(SqlKind comparison,
                                 RexNode operand,
                                 RexLiteral timeLiteral,
                                 org.apache.calcite.avatica.util.TimeUnitRange timeUnit,
                                 boolean floor)

timestampValue

private Calendar timestampValue(RexLiteral timeLiteral)

floorRange

private com.google.common.collect.Range<Calendar> floorRange(org.apache.calcite.avatica.util.TimeUnitRange timeUnit,
                                                             SqlKind comparison,
                                                             Calendar c)

ceilRange

private com.google.common.collect.Range<Calendar> ceilRange(org.apache.calcite.avatica.util.TimeUnitRange timeUnit,
                                                            SqlKind comparison,
                                                            Calendar c)

isFloorCeilCall

boolean isFloorCeilCall(RexNode e)

increment

private Calendar increment(Calendar c,
                           org.apache.calcite.avatica.util.TimeUnitRange timeUnit)

decrement

private Calendar decrement(Calendar c,
                           org.apache.calcite.avatica.util.TimeUnitRange timeUnit)

ceil

private Calendar ceil(Calendar c,
                      org.apache.calcite.avatica.util.TimeUnitRange timeUnit)

floor

private Calendar floor(Calendar c,
                       org.apache.calcite.avatica.util.TimeUnitRange timeUnit)

Computes floor of a calendar to a given time unit.

Returns:

returns a copy of calendar, floored to the given time unit