Package io.trino.operator

Class HashAggregationOperator

java.lang.Object

io.trino.operator.HashAggregationOperator

All Implemented Interfaces:

Operator, AutoCloseable

public class HashAggregationOperator
extends Object
implements Operator

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	HashAggregationOperator.HashAggregationOperatorFactory

Field Summary

Fields inherited from interface io.trino.operator.Operator

NOT_BLOCKED

Constructor Summary

Constructors

Constructor

Description

HashAggregationOperator(OperatorContext operatorContext, List<Type> groupByTypes, List<Integer> groupByChannels, List<Integer> globalAggregationGroupIds, AggregationNode.Step step, boolean produceDefaultOutput, List<AccumulatorFactory> accumulatorFactories, Optional<Integer> hashChannel, Optional<Integer> groupIdChannel, int expectedGroups, Optional<io.airlift.units.DataSize> maxPartialMemory, boolean spillEnabled, io.airlift.units.DataSize memoryLimitForMerge, io.airlift.units.DataSize memoryLimitForMergeWithMemory, SpillerFactory spillerFactory, JoinCompiler joinCompiler, BlockTypeOperators blockTypeOperators, boolean useSystemMemory)

Method Summary

Modifier and Type	Method	Description
void	addInput(Page page)	Adds an input page to the operator.
void	close()	This method will always be called before releasing the Operator reference.
void	finish()	Notifies the operator that no more pages will be added and the operator should finish processing and flush results.
void	finishMemoryRevoke()	Clean up and release resources after completed memory revoking.
HashAggregationBuilder	getAggregationBuilder()
OperatorContext	getOperatorContext()
Page	getOutput()	Gets an output page from the operator.
boolean	isFinished()	Is this operator completely finished processing and no more output pages will be produced.
boolean	needsInput()	Returns true if and only if this operator can accept an input page.
com.google.common.util.concurrent.ListenableFuture<Void>	startMemoryRevoke()	After calling this method operator should revoke all reserved revocable memory.

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.trino.operator.Operator

isBlocked

Constructor Detail

HashAggregationOperator

public HashAggregationOperator(OperatorContext operatorContext,
                               List<Type> groupByTypes,
                               List<Integer> groupByChannels,
                               List<Integer> globalAggregationGroupIds,
                               AggregationNode.Step step,
                               boolean produceDefaultOutput,
                               List<AccumulatorFactory> accumulatorFactories,
                               Optional<Integer> hashChannel,
                               Optional<Integer> groupIdChannel,
                               int expectedGroups,
                               Optional<io.airlift.units.DataSize> maxPartialMemory,
                               boolean spillEnabled,
                               io.airlift.units.DataSize memoryLimitForMerge,
                               io.airlift.units.DataSize memoryLimitForMergeWithMemory,
                               SpillerFactory spillerFactory,
                               JoinCompiler joinCompiler,
                               BlockTypeOperators blockTypeOperators,
                               boolean useSystemMemory)

Method Detail

getOperatorContext

public OperatorContext getOperatorContext()

Specified by:

getOperatorContext in interface Operator

finish

public void finish()

Description copied from interface: Operator

Notifies the operator that no more pages will be added and the operator should finish processing and flush results. This method will not be called if the Task is already failed or canceled.

Specified by:

finish in interface Operator

isFinished

public boolean isFinished()

Description copied from interface: Operator

Is this operator completely finished processing and no more output pages will be produced.

Specified by:

isFinished in interface Operator

needsInput

public boolean needsInput()

Description copied from interface: Operator

Returns true if and only if this operator can accept an input page.

Specified by:

needsInput in interface Operator

addInput

public void addInput(Page page)

Description copied from interface: Operator

Adds an input page to the operator. This method will only be called if needsInput() returns true.

Specified by:

addInput in interface Operator

startMemoryRevoke

public com.google.common.util.concurrent.ListenableFuture<Void> startMemoryRevoke()

Description copied from interface: Operator

After calling this method operator should revoke all reserved revocable memory. As soon as memory is revoked returned future should be marked as done.

Spawned threads cannot modify OperatorContext because it's not thread safe. For this purpose implement Operator.finishMemoryRevoke()

Since memory revoking signal is delivered asynchronously to the Operator, implementation must gracefully handle the case when there no longer is any revocable memory allocated.

After this method is called on Operator the Driver is disallowed to call any processing methods on it (isBlocked/needsInput/addInput/getOutput) until Operator.finishMemoryRevoke() is called.

Specified by:

startMemoryRevoke in interface Operator

finishMemoryRevoke

public void finishMemoryRevoke()

Description copied from interface: Operator

Clean up and release resources after completed memory revoking. Called by driver once future returned by startMemoryRevoke is completed.

Specified by:

finishMemoryRevoke in interface Operator

getOutput

public Page getOutput()

Description copied from interface: Operator

Gets an output page from the operator. If no output data is currently available, return null.

Specified by:

getOutput in interface Operator

close

public void close()

Description copied from interface: Operator

This method will always be called before releasing the Operator reference.

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Operator

getAggregationBuilder

public HashAggregationBuilder getAggregationBuilder()