Package io.trino.execution.scheduler

Class StageTaskSourceFactory.HashDistributionTaskSource

java.lang.Object

io.trino.execution.scheduler.StageTaskSourceFactory.HashDistributionTaskSource

All Implemented Interfaces:

TaskSource, Closeable, AutoCloseable

Enclosing class:

StageTaskSourceFactory

public static class StageTaskSourceFactory.HashDistributionTaskSource
extends Object
implements TaskSource

Constructor Summary

Constructors

Constructor

Description

HashDistributionTaskSource(Map<PlanNodeId,SplitSource> splitSources, com.google.common.collect.Multimap<PlanNodeId,ExchangeSourceHandle> partitionedExchangeSourceHandles, com.google.common.collect.Multimap<PlanNodeId,ExchangeSourceHandle> replicatedExchangeSourceHandles, int splitBatchSize, LongConsumer getSplitTimeRecorder, int[] bucketToPartitionMap, Optional<BucketNodeMap> bucketNodeMap, Optional<CatalogName> catalogRequirement)

Method Summary

Modifier and Type	Method	Description
void	close()
static StageTaskSourceFactory.HashDistributionTaskSource	create(Session session, PlanFragment fragment, SplitSourceFactory splitSourceFactory, com.google.common.collect.Multimap<PlanFragmentId,ExchangeSourceHandle> exchangeSourceHandles, int splitBatchSize, LongConsumer getSplitTimeRecorder, int[] bucketToPartitionMap, Optional<BucketNodeMap> bucketNodeMap)
List<TaskDescriptor>	getMoreTasks()
boolean	isFinished()

Methods inherited from class java.lang.Object

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

Constructor Detail

HashDistributionTaskSource

public HashDistributionTaskSource(Map<PlanNodeId,SplitSource> splitSources,
                                  com.google.common.collect.Multimap<PlanNodeId,ExchangeSourceHandle> partitionedExchangeSourceHandles,
                                  com.google.common.collect.Multimap<PlanNodeId,ExchangeSourceHandle> replicatedExchangeSourceHandles,
                                  int splitBatchSize,
                                  LongConsumer getSplitTimeRecorder,
                                  int[] bucketToPartitionMap,
                                  Optional<BucketNodeMap> bucketNodeMap,
                                  Optional<CatalogName> catalogRequirement)

Method Detail

create

public static StageTaskSourceFactory.HashDistributionTaskSource create(Session session,
                                                                       PlanFragment fragment,
                                                                       SplitSourceFactory splitSourceFactory,
                                                                       com.google.common.collect.Multimap<PlanFragmentId,ExchangeSourceHandle> exchangeSourceHandles,
                                                                       int splitBatchSize,
                                                                       LongConsumer getSplitTimeRecorder,
                                                                       int[] bucketToPartitionMap,
                                                                       Optional<BucketNodeMap> bucketNodeMap)

getMoreTasks

public List<TaskDescriptor> getMoreTasks()

Specified by:

getMoreTasks in interface TaskSource

isFinished

public boolean isFinished()

Specified by:

isFinished in interface TaskSource

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable

Specified by:

close in interface TaskSource