Module org.elasticsearch.compute

Package org.elasticsearch.compute.aggregation

Class ToPartialGroupingAggregatorFunction

java.lang.Object

org.elasticsearch.compute.aggregation.ToPartialGroupingAggregatorFunction

All Implemented Interfaces:

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

public class ToPartialGroupingAggregatorFunction
extends Object
implements GroupingAggregatorFunction

An internal aggregate function that always emits intermediate (or partial) output regardless of the aggregate mode. The intermediate output should be consumed by FromPartialGroupingAggregatorFunction, which always receives the intermediate input. Since an intermediate aggregate output can consist of multiple blocks, we wrap these output blocks in a single composite block. The FromPartialGroupingAggregatorFunction then unwraps this input block into multiple primitive blocks and passes them to the delegating GroupingAggregatorFunction. Both of these commands yield the same result, except the second plan executes aggregates twice:

 ```
 | ... before
 | af(x) BY g
 | ... after
 ```
 ```
 | ... before
 | $x = to_partial(af(x)) BY g
 | from_partial($x, af(_)) BY g
 | ...  after
 

```

Nested Class Summary

Nested classes/interfaces inherited from interface org.elasticsearch.compute.aggregation.GroupingAggregatorFunction

GroupingAggregatorFunction.AddInput, GroupingAggregatorFunction.PreparedForEvaluation

Constructor Summary

Constructors

Constructor	Description
ToPartialGroupingAggregatorFunction(GroupingAggregatorFunction delegate, List<Integer> channels)

Method Summary

Modifier and Type	Method	Description
void	addIntermediateInput(int positionOffset, IntArrayBlock groupIdVector, Page page)	Add data produced by GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext).
void	addIntermediateInput(int positionOffset, IntBigArrayBlock groupIdVector, Page page)	Add data produced by GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext).
void	addIntermediateInput(int positionOffset, IntVector groupIdVector, Page page)	Add data produced by GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext).
void	close()
int	intermediateBlockCount()	The number of blocks used by intermediate state.
static List<IntermediateStateDesc>	intermediateStateDesc()
GroupingAggregatorFunction.PreparedForEvaluation	prepareEvaluateFinal(IntVector selected, GroupingAggregatorEvaluationContext ctx)	Prepare to build the final results for this aggregation.
GroupingAggregatorFunction.PreparedForEvaluation	prepareEvaluateIntermediate(IntVector selected, GroupingAggregatorEvaluationContext ctx)	Prepare to build the intermediate results for this aggregation.
GroupingAggregatorFunction.AddInput	prepareProcessRawInputPage(SeenGroupIds seenGroupIds, Page page)	Returns null to opt out of the callback loop for this page entirely, e.g.
void	selectedMayContainUnseenGroups(SeenGroupIds seenGroupIds)	Call this to signal to the aggregation that the selected parameter that's passed to GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext) or GroupingAggregatorFunction.prepareEvaluateFinal(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext) may reference groups that haven't been seen.
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Details

ToPartialGroupingAggregatorFunction

public ToPartialGroupingAggregatorFunction(GroupingAggregatorFunction delegate,
 List<Integer> channels)

Method Details

intermediateStateDesc

public static List<IntermediateStateDesc> intermediateStateDesc()

prepareProcessRawInputPage

public GroupingAggregatorFunction.AddInput prepareProcessRawInputPage(SeenGroupIds seenGroupIds,
 Page page)

Description copied from interface: GroupingAggregatorFunction

Returns null to opt out of the callback loop for this page entirely, e.g. when the values block is all-null and contributes nothing to the aggregation.

Specified by:

prepareProcessRawInputPage in interface GroupingAggregatorFunction

selectedMayContainUnseenGroups

public void selectedMayContainUnseenGroups(SeenGroupIds seenGroupIds)

Description copied from interface: GroupingAggregatorFunction

Call this to signal to the aggregation that the selected parameter that's passed to GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext) or GroupingAggregatorFunction.prepareEvaluateFinal(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext) may reference groups that haven't been seen. This puts the underlying storage into a mode where it'll track which group ids have been seen, even if that increases the overhead.

Specified by:

selectedMayContainUnseenGroups in interface GroupingAggregatorFunction

addIntermediateInput

public void addIntermediateInput(int positionOffset,
 IntArrayBlock groupIdVector,
 Page page)

Description copied from interface: GroupingAggregatorFunction

Add data produced by GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext).

Specified by:

addIntermediateInput in interface GroupingAggregatorFunction

addIntermediateInput

public void addIntermediateInput(int positionOffset,
 IntBigArrayBlock groupIdVector,
 Page page)

Description copied from interface: GroupingAggregatorFunction

Add data produced by GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext).

Specified by:

addIntermediateInput in interface GroupingAggregatorFunction

addIntermediateInput

public void addIntermediateInput(int positionOffset,
 IntVector groupIdVector,
 Page page)

Description copied from interface: GroupingAggregatorFunction

Add data produced by GroupingAggregatorFunction.prepareEvaluateIntermediate(org.elasticsearch.compute.data.IntVector, org.elasticsearch.compute.aggregation.GroupingAggregatorEvaluationContext).

Specified by:

addIntermediateInput in interface GroupingAggregatorFunction

prepareEvaluateIntermediate

public GroupingAggregatorFunction.PreparedForEvaluation prepareEvaluateIntermediate(IntVector selected,
 GroupingAggregatorEvaluationContext ctx)

Description copied from interface: GroupingAggregatorFunction

Prepare to build the intermediate results for this aggregation.

Specified by:

prepareEvaluateIntermediate in interface GroupingAggregatorFunction

Parameters:

selected - the groupIds that have been selected to be included in the results. Always ascending.

prepareEvaluateFinal

public GroupingAggregatorFunction.PreparedForEvaluation prepareEvaluateFinal(IntVector selected,
 GroupingAggregatorEvaluationContext ctx)

Description copied from interface: GroupingAggregatorFunction

Prepare to build the final results for this aggregation.

Specified by:

prepareEvaluateFinal in interface GroupingAggregatorFunction

Parameters:

selected - the groupIds that have been selected to be included in the results. Always ascending.

This function is called in the coordinator node after all intermediate results have been gathered from the worker nodes, and aggregated into intermediate blocks.

intermediateBlockCount

public int intermediateBlockCount()

Description copied from interface: GroupingAggregatorFunction

The number of blocks used by intermediate state.

Specified by:

intermediateBlockCount in interface GroupingAggregatorFunction

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

toString

public String toString()

Overrides:

toString in class Object