Package io.trino.execution

Class SqlStage

java.lang.Object

io.trino.execution.SqlStage

@ThreadSafe
public final class SqlStage
extends Object

This class is merely a container used by coordinator to track tasks for a single stage.

It is designed to keep track of execution statistics for tasks from the same stage as well as aggregating them and providing a final stage info when the stage execution is completed.

This class doesn't imply anything about the nature of execution. It is not responsible for scheduling tasks in a certain order, gang scheduling or any other execution primitives.

Method Summary

Modifier and Type	Method	Description
void	abort()
void	addFinalStageInfoListener(StateMachine.StateChangeListener<StageInfo> stateChangeListener)	Add a listener for the final stage info.
static SqlStage	createSqlStage(StageId stageId, PlanFragment fragment, Map<PlanNodeId,TableInfo> tables, RemoteTaskFactory remoteTaskFactory, Session session, boolean summarizeTaskInfo, NodeTaskMap nodeTaskMap, Executor executor, SplitSchedulerStats schedulerStats)
Optional<RemoteTask>	createTask(InternalNode node, int partition, int attempt, Optional<int[]> bucketToPartition, OutputBuffers outputBuffers, com.google.common.collect.Multimap<PlanNodeId,Split> splits, com.google.common.collect.Multimap<PlanNodeId,Lifespan> noMoreSplitsForLifespan, Set<PlanNodeId> noMoreSplits)
void	fail(Throwable throwable)
void	finish()
BasicStageStats	getBasicStageStats()
PlanFragment	getFragment()
StageId	getStageId()
StageInfo	getStageInfo()
io.airlift.units.Duration	getTotalCpuTime()
long	getTotalMemoryReservation()
long	getUserMemoryReservation()
void	recordGetSplitTime(long start)
String	toString()

Methods inherited from class java.lang.Object

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

Method Detail

createSqlStage

public static SqlStage createSqlStage(StageId stageId,
                                      PlanFragment fragment,
                                      Map<PlanNodeId,TableInfo> tables,
                                      RemoteTaskFactory remoteTaskFactory,
                                      Session session,
                                      boolean summarizeTaskInfo,
                                      NodeTaskMap nodeTaskMap,
                                      Executor executor,
                                      SplitSchedulerStats schedulerStats)

getStageId

public StageId getStageId()

finish

public void finish()

abort

public void abort()

fail

public void fail(Throwable throwable)

addFinalStageInfoListener

public void addFinalStageInfoListener(StateMachine.StateChangeListener<StageInfo> stateChangeListener)

Add a listener for the final stage info. This notification is guaranteed to be fired only once. Listener is always notified asynchronously using a dedicated notification thread pool so, care should be taken to avoid leaking this when adding a listener in a constructor. Additionally, it is possible notifications are observed out of order due to the asynchronous execution.

getFragment

public PlanFragment getFragment()

getUserMemoryReservation

public long getUserMemoryReservation()

getTotalMemoryReservation

public long getTotalMemoryReservation()

getTotalCpuTime

public io.airlift.units.Duration getTotalCpuTime()

getBasicStageStats

public BasicStageStats getBasicStageStats()

getStageInfo

public StageInfo getStageInfo()

createTask

public Optional<RemoteTask> createTask(InternalNode node,
                                       int partition,
                                       int attempt,
                                       Optional<int[]> bucketToPartition,
                                       OutputBuffers outputBuffers,
                                       com.google.common.collect.Multimap<PlanNodeId,Split> splits,
                                       com.google.common.collect.Multimap<PlanNodeId,Lifespan> noMoreSplitsForLifespan,
                                       Set<PlanNodeId> noMoreSplits)

recordGetSplitTime

public void recordGetSplitTime(long start)

toString

public String toString()

Overrides:

toString in class Object