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.
Object	getOnlyValue(ColumnValueSelector selector) Retrieve a single object using the ColumnValueSelector.
void	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 columnIndex, 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	writeToKeyBuffer(int keyBufferPosition, Object obj, ByteBuffer keyBuffer) Write a given object 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 void 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

getOnlyValue

@Nullable
public Object getOnlyValue(ColumnValueSelector selector)

Description copied from interface: GroupByColumnSelectorStrategy

Retrieve a single object using the ColumnValueSelector. The reading column must have a single value.

Specified by:

getOnlyValue in interface GroupByColumnSelectorStrategy

Parameters:

selector - Value selector for a column

Returns:

an object retrieved from the column

writeToKeyBuffer

public void writeToKeyBuffer(int keyBufferPosition,
                             @Nullable
                             Object obj,
                             ByteBuffer keyBuffer)

Description copied from interface: GroupByColumnSelectorStrategy

Write a given object to the keyBuffer at keyBufferPosition.

Specified by:

writeToKeyBuffer in interface GroupByColumnSelectorStrategy

Parameters:

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

obj - row value object retrieved from GroupByColumnSelectorStrategy.getOnlyValue(ColumnValueSelector)

keyBuffer - grouping key

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

Returns:

BufferComparator for comparing values written

initGroupingKeyColumnValue

public void initGroupingKeyColumnValue(int keyBufferPosition,
                                       int columnIndex,
                                       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.

columnIndex - Index of the column within the row values 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