Package io.trino.operator

Class TaskContext

java.lang.Object

io.trino.operator.TaskContext

@ThreadSafe
public class TaskContext
extends Object

Method Summary

Modifier and Type	Method	Description
<C,R> R	accept(QueryContextVisitor<C,R> visitor, C context)
<C,R> List<R>	acceptChildren(QueryContextVisitor<C,R> visitor, C context)
DynamicFiltersCollector.VersionedDynamicFilterDomains	acknowledgeAndGetNewDynamicFilterDomains(long callersCurrentVersion)
void	addCompletedDriverGroup(Lifespan driverGroup)
void	addDynamicFilter(Map<DynamicFilterId,Domain> dynamicFilterDomains)
PipelineContext	addPipelineContext(int pipelineId, boolean inputPipeline, boolean outputPipeline, boolean partitioned)
static TaskContext	createTaskContext(QueryContext queryContext, TaskStateMachine taskStateMachine, io.airlift.stats.GcMonitor gcMonitor, Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session, MemoryTrackingContext taskMemoryContext, Runnable notifyStatusChanged, boolean perOperatorCpuTimerEnabled, boolean cpuTimerEnabled, Optional<io.airlift.units.DataSize> maxMemory)
void	failed(Throwable cause)
void	freeSpill(long bytes)
Set<Lifespan>	getCompletedDriverGroups()	Returns the completed driver groups (excluding taskWide).
long	getDynamicFiltersVersion()
int	getFullGcCount()
io.airlift.units.Duration	getFullGcTime()
io.airlift.stats.CounterStat	getInputPositions()
LocalDynamicFiltersCollector	getLocalDynamicFiltersCollector()
io.airlift.units.DataSize	getMemoryReservation()
io.airlift.stats.CounterStat	getOutputDataSize()
io.airlift.stats.CounterStat	getOutputPositions()
List<PipelineContext>	getPipelineContexts()
io.airlift.stats.CounterStat	getProcessedInputDataSize()
QueryContext	getQueryContext()
io.airlift.units.DataSize	getRevocableMemoryReservation()
Session	getSession()
TaskState	getState()
TaskId	getTaskId()
MemoryTrackingContext	getTaskMemoryContext()
TaskStats	getTaskStats()
boolean	isCpuTimerEnabled()
boolean	isDone()
boolean	isPerOperatorCpuTimerEnabled()
LocalMemoryContext	localMemoryContext()
com.google.common.util.concurrent.ListenableFuture<Void>	reserveSpill(long bytes)
void	sourceTaskFailed(TaskId taskId, Throwable failure)
void	start()
void	updateDomains(Map<DynamicFilterId,Domain> dynamicFilterDomains)

Methods inherited from class java.lang.Object

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

Method Detail

createTaskContext

public static TaskContext createTaskContext(QueryContext queryContext,
                                            TaskStateMachine taskStateMachine,
                                            io.airlift.stats.GcMonitor gcMonitor,
                                            Executor notificationExecutor,
                                            ScheduledExecutorService yieldExecutor,
                                            Session session,
                                            MemoryTrackingContext taskMemoryContext,
                                            Runnable notifyStatusChanged,
                                            boolean perOperatorCpuTimerEnabled,
                                            boolean cpuTimerEnabled,
                                            Optional<io.airlift.units.DataSize> maxMemory)

getTaskId

public TaskId getTaskId()

addPipelineContext

public PipelineContext addPipelineContext(int pipelineId,
                                          boolean inputPipeline,
                                          boolean outputPipeline,
                                          boolean partitioned)

getSession

public Session getSession()

start

public void start()

failed

public void failed(Throwable cause)

isDone

public boolean isDone()

getState

public TaskState getState()

getMemoryReservation

public io.airlift.units.DataSize getMemoryReservation()

getRevocableMemoryReservation

public io.airlift.units.DataSize getRevocableMemoryReservation()

getCompletedDriverGroups

public Set<Lifespan> getCompletedDriverGroups()

Returns the completed driver groups (excluding taskWide). A driver group is considered complete if all drivers associated with it has completed, and no new drivers associated with it will be created.

addCompletedDriverGroup

public void addCompletedDriverGroup(Lifespan driverGroup)

getPipelineContexts

public List<PipelineContext> getPipelineContexts()

reserveSpill

public com.google.common.util.concurrent.ListenableFuture<Void> reserveSpill(long bytes)

freeSpill

public void freeSpill(long bytes)

localMemoryContext

public LocalMemoryContext localMemoryContext()

isPerOperatorCpuTimerEnabled

public boolean isPerOperatorCpuTimerEnabled()

isCpuTimerEnabled

public boolean isCpuTimerEnabled()

getProcessedInputDataSize

public io.airlift.stats.CounterStat getProcessedInputDataSize()

getInputPositions

public io.airlift.stats.CounterStat getInputPositions()

getOutputDataSize

public io.airlift.stats.CounterStat getOutputDataSize()

getOutputPositions

public io.airlift.stats.CounterStat getOutputPositions()

getFullGcTime

public io.airlift.units.Duration getFullGcTime()

getFullGcCount

public int getFullGcCount()

updateDomains

public void updateDomains(Map<DynamicFilterId,Domain> dynamicFilterDomains)

getDynamicFiltersVersion

public long getDynamicFiltersVersion()

acknowledgeAndGetNewDynamicFilterDomains

public DynamicFiltersCollector.VersionedDynamicFilterDomains acknowledgeAndGetNewDynamicFilterDomains(long callersCurrentVersion)

getTaskStats

public TaskStats getTaskStats()

accept

public <C,R> R accept(QueryContextVisitor<C,R> visitor,
                            C context)

acceptChildren

public <C,R> List<R> acceptChildren(QueryContextVisitor<C,R> visitor,
                                          C context)

getTaskMemoryContext

public MemoryTrackingContext getTaskMemoryContext()

getQueryContext

public QueryContext getQueryContext()

getLocalDynamicFiltersCollector

public LocalDynamicFiltersCollector getLocalDynamicFiltersCollector()

addDynamicFilter

public void addDynamicFilter(Map<DynamicFilterId,Domain> dynamicFilterDomains)

sourceTaskFailed

public void sourceTaskFailed(TaskId taskId,
                             Throwable failure)