org.apache.druid.query.groupby.epinephelinae.column

Class NullableNumericGroupByColumnSelectorStrategy

java.lang.Object

org.apache.druid.query.groupby.epinephelinae.column.NullableNumericGroupByColumnSelectorStrategy

All Implemented Interfaces:

ColumnSelectorStrategy, GroupByColumnSelectorStrategy

public class NullableNumericGroupByColumnSelectorStrategy
extends Object
implements GroupByColumnSelectorStrategy

A wrapper around a numeric GroupByColumnSelectorStrategy that makes it null-aware. Should only be used for numeric strategies, not for string strategies.

See Also:

for why this only works in the numeric case

Field Summary

Fields inherited from interface org.apache.druid.query.groupby.epinephelinae.column.GroupByColumnSelectorStrategy

GROUP_BY_MISSING_VALUE

Constructor Summary

Constructors

Constructor and Description
NullableNumericGroupByColumnSelectorStrategy(GroupByColumnSelectorStrategy delegate)

Method Summary

Modifier and Type	Method and Description
Grouper.BufferComparator	bufferComparator(int keyBufferPosition, StringComparator stringComparator) Return BufferComparator for values written using this strategy when limit is pushed down to segment scan.
boolean	checkRowIndexAndAddValueToGroupingKey(int keyBufferPosition, Object rowObj, int rowValIdx, ByteBuffer keyBuffer) If rowValIdx is less than the size of rowObj (haven't handled all of the row values): First, read the value at rowValIdx from a rowObj and write that value to the keyBuffer at keyBufferPosition.
int	getGroupingKeySize() Return the size, in bytes, of this dimension's values in the grouping key.
int	initColumnValues(ColumnValueSelector selector, int columnIndex, Object[] values) Retrieve a row object from the ColumnValueSelector and put it in valuess at columnIndex.
void	initGroupingKeyColumnValue(int keyBufferPosition, int dimensionIndex, Object rowObj, ByteBuffer keyBuffer, int[] stack) Read the first value within a row values object (e.
void	processValueFromGroupingKey(GroupByColumnSelectorPlus selectorPlus, ByteBuffer key, ResultRow resultRow, int keyBufferPosition) Read a value from a grouping key and add it to the group by query result row, using the output name specified in a DimensionSpec.
void	reset() Reset any internal state held by this selector.
int	writeToKeyBuffer(int keyBufferPosition, ColumnValueSelector selector, ByteBuffer keyBuffer) Write a single object from the given selector to the keyBuffer at keyBufferPosition.

Methods inherited from class java.lang.Object

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

Constructor Detail

NullableNumericGroupByColumnSelectorStrategy

public NullableNumericGroupByColumnSelectorStrategy(GroupByColumnSelectorStrategy delegate)

Method Detail

getGroupingKeySize

public int getGroupingKeySize()

Description copied from interface: GroupByColumnSelectorStrategy

Return the size, in bytes, of this dimension's values in the grouping key. For example, a String implementation would return 4, the size of an int.

Specified by:

getGroupingKeySize in interface GroupByColumnSelectorStrategy

Returns:

size, in bytes, of this dimension's values in the grouping key.

processValueFromGroupingKey

public void processValueFromGroupingKey(GroupByColumnSelectorPlus selectorPlus,
                                        ByteBuffer key,
                                        ResultRow resultRow,
                                        int keyBufferPosition)

Description copied from interface: GroupByColumnSelectorStrategy

Read a value from a grouping key and add it to the group by query result row, using the output name specified in a DimensionSpec. An implementation may choose to not add anything to the result row (e.g., as the String implementation does for empty rows) selectorPlus provides access to: - the keyBufferPosition offset from which to read the value - the dimension value selector - the DimensionSpec for this dimension from the query

Specified by:

processValueFromGroupingKey in interface GroupByColumnSelectorStrategy

Parameters:

selectorPlus - dimension info containing the key offset, value selector, and dimension spec

key - grouping key

resultRow - result row for the group by query being served

keyBufferPosition - buffer position for the grouping key, added to support chaining multiple ColumnSelectorStrategy

initColumnValues

public int initColumnValues(ColumnValueSelector selector,
                            int columnIndex,
                            Object[] values)

Description copied from interface: GroupByColumnSelectorStrategy

Retrieve a row object from the ColumnValueSelector and put it in valuess at columnIndex.

Specified by:

initColumnValues in interface GroupByColumnSelectorStrategy

Parameters:

selector - Value selector for a column.

columnIndex - Index of the column within the row values array

values - Row values array, one index per column

Returns:

estimated increase in internal state footprint, in bytes, as a result of this operation. May be zero if memory did not increase as a result of this operation. Will not be negative.

writeToKeyBuffer

public int writeToKeyBuffer(int keyBufferPosition,
                            ColumnValueSelector selector,
                            ByteBuffer keyBuffer)

Description copied from interface: GroupByColumnSelectorStrategy

Write a single object from the given selector to the keyBuffer at keyBufferPosition. The reading column must have a single value. The position of the keyBuffer may be modified.

Specified by:

writeToKeyBuffer in interface GroupByColumnSelectorStrategy

Parameters:

keyBufferPosition - starting offset for this column's value within the grouping key

selector - selector to retrieve row value object from

keyBuffer - grouping key

Returns:

estimated increase in internal state footprint, in bytes, as a result of this operation. May be zero if memory did not increase as a result of this operation. Will not be negative.

bufferComparator

public Grouper.BufferComparator bufferComparator(int keyBufferPosition,
                                                 @Nullable
                                                 StringComparator stringComparator)

Description copied from interface: GroupByColumnSelectorStrategy

Return BufferComparator for values written using this strategy when limit is pushed down to segment scan.

Specified by:

bufferComparator in interface GroupByColumnSelectorStrategy

Parameters:

keyBufferPosition - starting offset for this column's value within the grouping key

stringComparator - stringComparator from LimitSpec for this column. If this is null, implementations will use the StringComparators.LEXICOGRAPHIC comparator.

Returns:

BufferComparator for comparing values written

initGroupingKeyColumnValue

public void initGroupingKeyColumnValue(int keyBufferPosition,
                                       int dimensionIndex,
                                       Object rowObj,
                                       ByteBuffer keyBuffer,
                                       int[] stack)

Description copied from interface: GroupByColumnSelectorStrategy

Read the first value within a row values object (e. g. IndexedInts, as the value in a dictionary-encoded string column) and write that value to the keyBuffer at keyBufferPosition. If the row size is 0 (e. g. IndexedInts.size()), write GroupByColumnSelectorStrategy.GROUP_BY_MISSING_VALUE instead. If the size of the row is > 0, write 1 to stack[] at columnIndex, otherwise write 0.

Specified by:

initGroupingKeyColumnValue in interface GroupByColumnSelectorStrategy

Parameters:

keyBufferPosition - Starting offset for this column's value within the grouping key.

dimensionIndex - Index of this dimension within the stack array

rowObj - Row value object for this column

keyBuffer - grouping key

stack - array containing the current within-row value index for each column

checkRowIndexAndAddValueToGroupingKey

public boolean checkRowIndexAndAddValueToGroupingKey(int keyBufferPosition,
                                                     Object rowObj,
                                                     int rowValIdx,
                                                     ByteBuffer keyBuffer)

Description copied from interface: GroupByColumnSelectorStrategy

If rowValIdx is less than the size of rowObj (haven't handled all of the row values): First, read the value at rowValIdx from a rowObj and write that value to the keyBuffer at keyBufferPosition. Then return true Otherwise, return false.

Specified by:

checkRowIndexAndAddValueToGroupingKey in interface GroupByColumnSelectorStrategy

Parameters:

keyBufferPosition - Starting offset for this column's value within the grouping key.

rowObj - Row value object for this column (e.g., IndexedInts)

rowValIdx - Index of the current value being grouped on within the row

keyBuffer - grouping key

Returns:

true if rowValIdx < size of rowObj, false otherwise

reset

public void reset()

Description copied from interface: GroupByColumnSelectorStrategy

Reset any internal state held by this selector. After this method is called, any row objects or key objects generated by any methods of this class must be considered unreadable. Calling GroupByColumnSelectorStrategy.processValueFromGroupingKey(org.apache.druid.query.groupby.epinephelinae.column.GroupByColumnSelectorPlus, java.nio.ByteBuffer, org.apache.druid.query.groupby.ResultRow, int) on that memory has undefined behavior.

Specified by:

reset in interface GroupByColumnSelectorStrategy