Package io.trino.execution.scheduler

Class SourcePartitionedScheduler

java.lang.Object

io.trino.execution.scheduler.SourcePartitionedScheduler

All Implemented Interfaces:

SourceScheduler

public class SourcePartitionedScheduler
extends Object
implements SourceScheduler

Method Summary

Modifier and Type	Method	Description
void	close()
List<Lifespan>	drainCompletedLifespans()
PlanNodeId	getPlanNodeId()
static SourceScheduler	newSourcePartitionedSchedulerAsSourceScheduler(PipelinedStageExecution stageExecution, PlanNodeId partitionedNode, SplitSource splitSource, SplitPlacementPolicy splitPlacementPolicy, int splitBatchSize, boolean groupedExecution, DynamicFilterService dynamicFilterService, TableExecuteContextManager tableExecuteContextManager, BooleanSupplier anySourceTaskBlocked, PartitionIdAllocator partitionIdAllocator, Map<InternalNode,RemoteTask> scheduledTasks)	Obtains a SourceScheduler suitable for use in FixedSourcePartitionedScheduler.
static StageScheduler	newSourcePartitionedSchedulerAsStageScheduler(PipelinedStageExecution stageExecution, PlanNodeId partitionedNode, SplitSource splitSource, SplitPlacementPolicy splitPlacementPolicy, int splitBatchSize, DynamicFilterService dynamicFilterService, TableExecuteContextManager tableExecuteContextManager, BooleanSupplier anySourceTaskBlocked)	Obtains an instance of SourcePartitionedScheduler suitable for use as a stage scheduler.
void	noMoreLifespans()
ScheduleResult	schedule()
void	startLifespan(Lifespan lifespan, ConnectorPartitionHandle partitionHandle)

Methods inherited from class java.lang.Object

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

Method Detail

getPlanNodeId

public PlanNodeId getPlanNodeId()

Specified by:

getPlanNodeId in interface SourceScheduler

newSourcePartitionedSchedulerAsStageScheduler

public static StageScheduler newSourcePartitionedSchedulerAsStageScheduler(PipelinedStageExecution stageExecution,
                                                                           PlanNodeId partitionedNode,
                                                                           SplitSource splitSource,
                                                                           SplitPlacementPolicy splitPlacementPolicy,
                                                                           int splitBatchSize,
                                                                           DynamicFilterService dynamicFilterService,
                                                                           TableExecuteContextManager tableExecuteContextManager,
                                                                           BooleanSupplier anySourceTaskBlocked)

Obtains an instance of SourcePartitionedScheduler suitable for use as a stage scheduler.

This returns an ungrouped SourcePartitionedScheduler that requires minimal management from the caller, which is ideal for use as a stage scheduler.

newSourcePartitionedSchedulerAsSourceScheduler

public static SourceScheduler newSourcePartitionedSchedulerAsSourceScheduler(PipelinedStageExecution stageExecution,
                                                                             PlanNodeId partitionedNode,
                                                                             SplitSource splitSource,
                                                                             SplitPlacementPolicy splitPlacementPolicy,
                                                                             int splitBatchSize,
                                                                             boolean groupedExecution,
                                                                             DynamicFilterService dynamicFilterService,
                                                                             TableExecuteContextManager tableExecuteContextManager,
                                                                             BooleanSupplier anySourceTaskBlocked,
                                                                             PartitionIdAllocator partitionIdAllocator,
                                                                             Map<InternalNode,RemoteTask> scheduledTasks)

Obtains a SourceScheduler suitable for use in FixedSourcePartitionedScheduler.

This returns a SourceScheduler that can be used for a pipeline that is either ungrouped or grouped. However, the caller is responsible initializing the driver groups in this scheduler accordingly.

Besides, the caller is required to poll drainCompletedLifespans() in addition to schedule() on the returned object. Otherwise, lifecycle transitioning of the object will not work properly.

startLifespan

public void startLifespan(Lifespan lifespan,
                          ConnectorPartitionHandle partitionHandle)

Specified by:

startLifespan in interface SourceScheduler

noMoreLifespans

public void noMoreLifespans()

Specified by:

noMoreLifespans in interface SourceScheduler

schedule

public ScheduleResult schedule()

Specified by:

schedule in interface SourceScheduler

close

public void close()

Specified by:

close in interface SourceScheduler

drainCompletedLifespans

public List<Lifespan> drainCompletedLifespans()

Specified by:

drainCompletedLifespans in interface SourceScheduler