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()
PlanNodeId	getPlanNodeId()
static SourceScheduler	newSourcePartitionedSchedulerAsSourceScheduler(StageExecution stageExecution, PlanNodeId partitionedNode, SplitSource splitSource, SplitPlacementPolicy splitPlacementPolicy, int splitBatchSize, DynamicFilterService dynamicFilterService, TableExecuteContextManager tableExecuteContextManager, BooleanSupplier anySourceTaskBlocked, PartitionIdAllocator partitionIdAllocator, Map<InternalNode,RemoteTask> scheduledTasks)	Obtains a SourceScheduler suitable for use in FixedSourcePartitionedScheduler.
static StageScheduler	newSourcePartitionedSchedulerAsStageScheduler(StageExecution 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.
ScheduleResult	schedule()
void	start()

Methods inherited from class java.lang.Object

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

Method Details

getPlanNodeId

public PlanNodeId getPlanNodeId()

Specified by:

getPlanNodeId in interface SourceScheduler

newSourcePartitionedSchedulerAsStageScheduler

public static StageScheduler newSourcePartitionedSchedulerAsStageScheduler(StageExecution 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(StageExecution stageExecution,
 PlanNodeId partitionedNode,
 SplitSource splitSource,
 SplitPlacementPolicy splitPlacementPolicy,
 int splitBatchSize,
 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.

start

public void start()

Specified by:

start in interface SourceScheduler

schedule

public ScheduleResult schedule()

Specified by:

schedule in interface SourceScheduler

close

public void close()

Specified by:

close in interface SourceScheduler