Module org.elasticsearch.compute

Package org.elasticsearch.compute.aggregation

Class CountAggregatorFunction

java.lang.Object

org.elasticsearch.compute.aggregation.CountAggregatorFunction

All Implemented Interfaces:

Closeable, AutoCloseable, AggregatorFunction, org.elasticsearch.core.Releasable

Direct Known Subclasses:

DenseVectorCountAggregatorFunction

public class CountAggregatorFunction
extends Object
implements AggregatorFunction

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
protected static class	CountAggregatorFunction.CountAggregatorFunctionSupplier

Method Summary

Modifier and Type	Method	Description
void	addIntermediateInput(Page page)	Add a pre-aggregated page worth of "intermediate" input.
void	addRawInput(Page page, BooleanVector mask)	Add a page worth of data to the aggregation.
void	close()
void	evaluateFinal(Block[] blocks, int offset, DriverContext driverContext)	Build the final results from running this agg.
void	evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext)	Build pre-aggregated "intermediate" data to pass to the AggregatorFunction.addIntermediateInput(org.elasticsearch.compute.data.Page).
protected int	getBlockTotalValueCount(Block block)	Returns the number of total values in a block
protected int	getBlockValueCountAtPosition(Block block, int position)	Returns the number of values at a given position in a block
int	intermediateBlockCount()	The number of blocks used by intermediate state.
static List<IntermediateStateDesc>	intermediateStateDesc()
static AggregatorFunctionSupplier	supplier()
String	toString()

Methods inherited from class java.lang.Object

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

Method Details

intermediateStateDesc

public static List<IntermediateStateDesc> intermediateStateDesc()

intermediateBlockCount

public int intermediateBlockCount()

Description copied from interface: AggregatorFunction

The number of blocks used by intermediate state.

Specified by:

intermediateBlockCount in interface AggregatorFunction

addRawInput

public void addRawInput(Page page,
 BooleanVector mask)

Description copied from interface: AggregatorFunction

Add a page worth of data to the aggregation.

Specified by:

addRawInput in interface AggregatorFunction

Parameters:

mask - a mask to apply to the positions. If the position is false then the aggregation should skip it.

getBlockTotalValueCount

protected int getBlockTotalValueCount(Block block)

Returns the number of total values in a block

Parameters:

block - block to count values for

Returns:

number of total values present in the block

getBlockValueCountAtPosition

protected int getBlockValueCountAtPosition(Block block,
 int position)

Returns the number of values at a given position in a block

Parameters:

block - block

position - position to get the number of values

Returns:

addIntermediateInput

public void addIntermediateInput(Page page)

Description copied from interface: AggregatorFunction

Add a pre-aggregated page worth of "intermediate" input. This intermediate input will have been created by calling AggregatorFunction.evaluateIntermediate(org.elasticsearch.compute.data.Block[], int, org.elasticsearch.compute.operator.DriverContext) on this agg, though likely in a different Driver and maybe on a different physical node.

Specified by:

addIntermediateInput in interface AggregatorFunction

evaluateIntermediate

public void evaluateIntermediate(Block[] blocks,
 int offset,
 DriverContext driverContext)

Description copied from interface: AggregatorFunction

Build pre-aggregated "intermediate" data to pass to the AggregatorFunction.addIntermediateInput(org.elasticsearch.compute.data.Page).

Specified by:

evaluateIntermediate in interface AggregatorFunction

Parameters:

blocks - write the output into this array

offset - write the first Block at this offset in blocks

evaluateFinal

public void evaluateFinal(Block[] blocks,
 int offset,
 DriverContext driverContext)

Description copied from interface: AggregatorFunction

Build the final results from running this agg.

Specified by:

evaluateFinal in interface AggregatorFunction

Parameters:

blocks - write the output into this array

offset - write the first Block at this offset in blocks

toString

public String toString()

Overrides:

toString in class Object

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable

Specified by:

close in interface org.elasticsearch.core.Releasable

supplier

public static AggregatorFunctionSupplier supplier()