Package io.trino.operator

Class TaskContext

java.lang.Object
io.trino.operator.TaskContext
@ThreadSafe public class TaskContext extends Object

  • Method Details

    • 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()

    • getSystemMemoryReservation

      public io.airlift.units.DataSize getSystemMemoryReservation()

    • 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)

    • localSystemMemoryContext

      public LocalMemoryContext localSystemMemoryContext()

    • 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)