Package io.trino.operator.aggregation.builder

Class InMemoryHashAggregationBuilder

java.lang.Object

io.trino.operator.aggregation.builder.InMemoryHashAggregationBuilder

All Implemented Interfaces:

HashAggregationBuilder, AutoCloseable

public class InMemoryHashAggregationBuilder
extends Object
implements HashAggregationBuilder

Constructor Summary

Constructors

Constructor	Description
InMemoryHashAggregationBuilder(List<AggregatorFactory> aggregatorFactories, AggregationNode.Step step, int expectedGroups, List<Type> groupByTypes, List<Integer> groupByChannels, Optional<Integer> hashChannel, OperatorContext operatorContext, Optional<io.airlift.units.DataSize> maxPartialMemory, FlatHashStrategyCompiler hashStrategyCompiler, UpdateMemory updateMemory)
InMemoryHashAggregationBuilder(List<AggregatorFactory> aggregatorFactories, AggregationNode.Step step, int expectedGroups, List<Type> groupByTypes, List<Integer> groupByChannels, Optional<Integer> hashChannel, OperatorContext operatorContext, Optional<io.airlift.units.DataSize> maxPartialMemory, Optional<Integer> unspillIntermediateChannelOffset, FlatHashStrategyCompiler hashStrategyCompiler, UpdateMemory updateMemory)

Method Summary

Modifier and Type	Method	Description
WorkProcessor<Page>	buildHashSortedResult()
WorkProcessor<Page>	buildResult()
List<Type>	buildSpillTypes()
List<Type>	buildTypes()
void	close()
void	finishMemoryRevoke()
int	getCapacity()
long	getGroupCount()
long	getGroupIdsSortingSize()	building hash sorted results requires memory for sorting group IDs.
int	getKeyChannels()
long	getSizeInMemory()
boolean	isFull()
Work<?>	processPage(Page page)
void	setSpillOutput()
com.google.common.util.concurrent.ListenableFuture<Void>	startMemoryRevoke()
static List<Type>	toTypes(List<? extends Type> groupByType, List<AggregatorFactory> factories, Optional<Integer> hashChannel)
void	updateMemory()

Methods inherited from class java.lang.Object

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

Constructor Details

InMemoryHashAggregationBuilder

public InMemoryHashAggregationBuilder(List<AggregatorFactory> aggregatorFactories,
 AggregationNode.Step step,
 int expectedGroups,
 List<Type> groupByTypes,
 List<Integer> groupByChannels,
 Optional<Integer> hashChannel,
 OperatorContext operatorContext,
 Optional<io.airlift.units.DataSize> maxPartialMemory,
 FlatHashStrategyCompiler hashStrategyCompiler,
 UpdateMemory updateMemory)

InMemoryHashAggregationBuilder

public InMemoryHashAggregationBuilder(List<AggregatorFactory> aggregatorFactories,
 AggregationNode.Step step,
 int expectedGroups,
 List<Type> groupByTypes,
 List<Integer> groupByChannels,
 Optional<Integer> hashChannel,
 OperatorContext operatorContext,
 Optional<io.airlift.units.DataSize> maxPartialMemory,
 Optional<Integer> unspillIntermediateChannelOffset,
 FlatHashStrategyCompiler hashStrategyCompiler,
 UpdateMemory updateMemory)

Method Details

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface HashAggregationBuilder

processPage

public Work<?> processPage(Page page)

Specified by:

processPage in interface HashAggregationBuilder

updateMemory

public void updateMemory()

Specified by:

updateMemory in interface HashAggregationBuilder

isFull

public boolean isFull()

Specified by:

isFull in interface HashAggregationBuilder

startMemoryRevoke

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

Specified by:

startMemoryRevoke in interface HashAggregationBuilder

finishMemoryRevoke

public void finishMemoryRevoke()

Specified by:

finishMemoryRevoke in interface HashAggregationBuilder

getSizeInMemory

public long getSizeInMemory()

getGroupIdsSortingSize

public long getGroupIdsSortingSize()

building hash sorted results requires memory for sorting group IDs. This method returns size of that memory requirement.

setSpillOutput

public void setSpillOutput()

getKeyChannels

public int getKeyChannels()

getGroupCount

public long getGroupCount()

buildResult

public WorkProcessor<Page> buildResult()

Specified by:

buildResult in interface HashAggregationBuilder

buildHashSortedResult

public WorkProcessor<Page> buildHashSortedResult()

buildSpillTypes

public List<Type> buildSpillTypes()

getCapacity

public int getCapacity()

buildTypes

public List<Type> buildTypes()

toTypes

public static List<Type> toTypes(List<? extends Type> groupByType,
 List<AggregatorFactory> factories,
 Optional<Integer> hashChannel)