com.facebook.presto.memory

Class QueryContext

java.lang.Object

com.facebook.presto.memory.QueryContext

@ThreadSafe
public class QueryContext
extends Object

Constructor Summary

Constructors

Constructor and Description

QueryContext(QueryId queryId, io.airlift.units.DataSize maxUserMemory, io.airlift.units.DataSize maxTotalMemory, io.airlift.units.DataSize maxBroadcastUsedMemory, io.airlift.units.DataSize maxRevocableMemory, MemoryPool memoryPool, com.facebook.airlift.stats.GcMonitor gcMonitor, Executor notificationExecutor, ScheduledExecutorService yieldExecutor, io.airlift.units.DataSize maxSpill, SpillSpaceTracker spillSpaceTracker, com.facebook.airlift.json.JsonCodec<List<TaskMemoryReservationSummary>> memoryReservationSummaryJsonCodec)

Method Summary

Modifier and Type	Method and Description
<C,R> R	accept(QueryContextVisitor<C,R> visitor, C context)
<C,R> List<R>	acceptChildren(QueryContextVisitor<C,R> visitor, C context)
TaskContext	addTaskContext(TaskStateMachine taskStateMachine, Session session, Optional<PlanNode> taskPlan, boolean perOperatorCpuTimerEnabled, boolean cpuTimerEnabled, boolean perOperatorAllocationTrackingEnabled, boolean allocationTrackingEnabled, boolean legacyLifespanCompletionCondition)
void	freeSpill(long bytes)
String	getAdditionalFailureInfo(long allocated, long delta, String allocationTag)
Collection<TaskContext>	getAllTaskContexts()
long	getMaxTotalMemory()
long	getMaxUserMemory()
MemoryPool	getMemoryPool()
long	getPeakNodeTotalMemory()
QueryId	getQueryId()
TaskContext	getTaskContextByTaskId(TaskId taskId)
boolean	isMemoryLimitsInitialized()
com.google.common.util.concurrent.ListenableFuture<?>	reserveSpill(long bytes)
void	setHeapDumpFilePath(String heapDumpFilePath)
void	setHeapDumpOnExceededMemoryLimitEnabled(boolean heapDumpOnExceededMemoryLimitEnabled)
void	setMemoryLimits(io.airlift.units.DataSize queryMaxTaskMemory, io.airlift.units.DataSize queryMaxTotalTaskMemory, io.airlift.units.DataSize queryMaxBroadcastMemory, io.airlift.units.DataSize queryMaxRevocableMemory)
void	setMemoryPool(MemoryPool newMemoryPool)
void	setPeakNodeTotalMemory(long peakNodeTotalMemoryInBytes)
void	setResourceOvercommit()
void	setVerboseExceededMemoryLimitErrorsEnabled(boolean verboseExceededMemoryLimitErrorsEnabled)
void	updateBroadcastMemory(long delta, String allocationTag)

Methods inherited from class java.lang.Object

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

Constructor Detail

QueryContext

public QueryContext(QueryId queryId,
                    io.airlift.units.DataSize maxUserMemory,
                    io.airlift.units.DataSize maxTotalMemory,
                    io.airlift.units.DataSize maxBroadcastUsedMemory,
                    io.airlift.units.DataSize maxRevocableMemory,
                    MemoryPool memoryPool,
                    com.facebook.airlift.stats.GcMonitor gcMonitor,
                    Executor notificationExecutor,
                    ScheduledExecutorService yieldExecutor,
                    io.airlift.units.DataSize maxSpill,
                    SpillSpaceTracker spillSpaceTracker,
                    com.facebook.airlift.json.JsonCodec<List<TaskMemoryReservationSummary>> memoryReservationSummaryJsonCodec)

Method Detail

isMemoryLimitsInitialized

public boolean isMemoryLimitsInitialized()

setResourceOvercommit

public void setResourceOvercommit()

updateBroadcastMemory

public void updateBroadcastMemory(long delta,
                                  String allocationTag)

reserveSpill

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

freeSpill

public void freeSpill(long bytes)

setMemoryPool

public void setMemoryPool(MemoryPool newMemoryPool)

getMemoryPool

public MemoryPool getMemoryPool()

getMaxUserMemory

public long getMaxUserMemory()

getMaxTotalMemory

public long getMaxTotalMemory()

setHeapDumpOnExceededMemoryLimitEnabled

public void setHeapDumpOnExceededMemoryLimitEnabled(boolean heapDumpOnExceededMemoryLimitEnabled)

setHeapDumpFilePath

public void setHeapDumpFilePath(String heapDumpFilePath)

addTaskContext

public TaskContext addTaskContext(TaskStateMachine taskStateMachine,
                                  Session session,
                                  Optional<PlanNode> taskPlan,
                                  boolean perOperatorCpuTimerEnabled,
                                  boolean cpuTimerEnabled,
                                  boolean perOperatorAllocationTrackingEnabled,
                                  boolean allocationTrackingEnabled,
                                  boolean legacyLifespanCompletionCondition)

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)

getTaskContextByTaskId

public TaskContext getTaskContextByTaskId(TaskId taskId)

getAllTaskContexts

public Collection<TaskContext> getAllTaskContexts()

getQueryId

public QueryId getQueryId()

setMemoryLimits

public void setMemoryLimits(io.airlift.units.DataSize queryMaxTaskMemory,
                            io.airlift.units.DataSize queryMaxTotalTaskMemory,
                            io.airlift.units.DataSize queryMaxBroadcastMemory,
                            io.airlift.units.DataSize queryMaxRevocableMemory)

setVerboseExceededMemoryLimitErrorsEnabled

public void setVerboseExceededMemoryLimitErrorsEnabled(boolean verboseExceededMemoryLimitErrorsEnabled)

getPeakNodeTotalMemory

public long getPeakNodeTotalMemory()

setPeakNodeTotalMemory

public void setPeakNodeTotalMemory(long peakNodeTotalMemoryInBytes)

getAdditionalFailureInfo

public String getAdditionalFailureInfo(long allocated,
                                       long delta,
                                       String allocationTag)