Package io.prestosql.execution

Class QueryStateMachine

java.lang.Object

io.prestosql.execution.QueryStateMachine

@ThreadSafe
public class QueryStateMachine
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	QueryStateMachine.QueryOutputManager

Method Summary

Modifier and Type	Method	Description
void	addOutputInfoListener(Consumer<QueryExecution.QueryOutputInfo> listener)
void	addPreparedStatement(String key, String value)
void	addQueryInfoStateChangeListener(StateMachine.StateChangeListener<QueryInfo> stateChangeListener)	Add a listener for the final query info.
void	addResetSessionProperties(String name)
void	addSetRole(String catalog, SelectedRole role)
void	addSetSessionProperties(String key, String value)
void	addStateChangeListener(StateMachine.StateChangeListener<QueryState> stateChangeListener)	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.
static QueryStateMachine	begin(String query, Optional<String> preparedQuery, Session session, URI self, ResourceGroupId resourceGroup, boolean transactionControl, TransactionManager transactionManager, AccessControl accessControl, Executor executor, Metadata metadata, WarningCollector warningCollector, Optional<QueryType> queryType)	Created QueryStateMachines must be transitioned to terminal states to clean up resources.
void	beginAnalysis()
void	clearTransactionId()
void	endAnalysis()
Map<String,String>	getAddedPreparedStatements()
BasicQueryInfo	getBasicQueryInfo(Optional<BasicStageStats> rootStage)
org.joda.time.DateTime	getCreateTime()
Set<String>	getDeallocatedPreparedStatements()
Optional<org.joda.time.DateTime>	getEndTime()
Optional<org.joda.time.DateTime>	getExecutionStartTime()
Optional<ExecutionFailureInfo>	getFailureInfo()
Optional<QueryInfo>	getFinalQueryInfo()
org.joda.time.DateTime	getLastHeartbeat()
VersionedMemoryPoolId	getMemoryPool()
long	getPeakNonRevocableMemoryInBytes()
long	getPeakRevocableMemoryInBytes()
long	getPeakTaskRevocableMemory()
long	getPeakTaskTotalMemory()
long	getPeakTaskUserMemory()
long	getPeakTotalMemoryInBytes()
long	getPeakUserMemoryInBytes()
QueryId	getQueryId()
QueryState	getQueryState()
Set<String>	getResetSessionProperties()
Session	getSession()
String	getSetPath()
Map<String,String>	getSetSessionProperties()
com.google.common.util.concurrent.ListenableFuture<QueryState>	getStateChange(QueryState currentState)
WarningCollector	getWarningCollector()
boolean	isDone()
void	pruneQueryInfo()
void	recordHeartbeat()
void	removePreparedStatement(String key)
void	setColumns(List<String> columnNames, List<Type> columnTypes)
void	setDynamicFiltersStatsSupplier(Supplier<DynamicFilterService.DynamicFiltersStats> dynamicFiltersStatsSupplier)
void	setInputs(List<Input> inputs)
void	setMemoryPool(VersionedMemoryPoolId memoryPool)
void	setOutput(Optional<Output> output)
void	setReferencedTables(List<TableInfo> tables)
void	setRoutines(List<RoutineInfo> routines)
void	setSetCatalog(String catalog)
void	setSetPath(String path)
void	setSetSchema(String schema)
void	setStartedTransactionId(TransactionId startedTransactionId)
void	setUpdateType(String updateType)
boolean	transitionToCanceled()
boolean	transitionToDispatching()
boolean	transitionToFailed(Throwable throwable)
boolean	transitionToFinishing()
boolean	transitionToPlanning()
boolean	transitionToRunning()
boolean	transitionToStarting()
boolean	transitionToWaitingForResources()
void	updateMemoryUsage(long deltaUserMemoryInBytes, long deltaRevocableMemoryInBytes, long deltaTotalMemoryInBytes, long taskUserMemoryInBytes, long taskRevocableMemoryInBytes, long taskTotalMemoryInBytes)
void	updateOutputLocations(Set<URI> newExchangeLocations, boolean noMoreExchangeLocations)
QueryInfo	updateQueryInfo(Optional<StageInfo> stageInfo)

Methods inherited from class java.lang.Object

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

Method Detail

begin

public static QueryStateMachine begin(String query,
                                      Optional<String> preparedQuery,
                                      Session session,
                                      URI self,
                                      ResourceGroupId resourceGroup,
                                      boolean transactionControl,
                                      TransactionManager transactionManager,
                                      AccessControl accessControl,
                                      Executor executor,
                                      Metadata metadata,
                                      WarningCollector warningCollector,
                                      Optional<QueryType> queryType)

Created QueryStateMachines must be transitioned to terminal states to clean up resources.

getQueryId

public QueryId getQueryId()

getSession

public Session getSession()

getPeakUserMemoryInBytes

public long getPeakUserMemoryInBytes()

getPeakRevocableMemoryInBytes

public long getPeakRevocableMemoryInBytes()

getPeakNonRevocableMemoryInBytes

public long getPeakNonRevocableMemoryInBytes()

getPeakTotalMemoryInBytes

public long getPeakTotalMemoryInBytes()

getPeakTaskUserMemory

public long getPeakTaskUserMemory()

getPeakTaskRevocableMemory

public long getPeakTaskRevocableMemory()

getPeakTaskTotalMemory

public long getPeakTaskTotalMemory()

getWarningCollector

public WarningCollector getWarningCollector()

updateMemoryUsage

public void updateMemoryUsage(long deltaUserMemoryInBytes,
                              long deltaRevocableMemoryInBytes,
                              long deltaTotalMemoryInBytes,
                              long taskUserMemoryInBytes,
                              long taskRevocableMemoryInBytes,
                              long taskTotalMemoryInBytes)

getBasicQueryInfo

public BasicQueryInfo getBasicQueryInfo(Optional<BasicStageStats> rootStage)

getMemoryPool

public VersionedMemoryPoolId getMemoryPool()

setMemoryPool

public void setMemoryPool(VersionedMemoryPoolId memoryPool)

addOutputInfoListener

public void addOutputInfoListener(Consumer<QueryExecution.QueryOutputInfo> listener)

setColumns

public void setColumns(List<String> columnNames,
                       List<Type> columnTypes)

updateOutputLocations

public void updateOutputLocations(Set<URI> newExchangeLocations,
                                  boolean noMoreExchangeLocations)

setInputs

public void setInputs(List<Input> inputs)

setOutput

public void setOutput(Optional<Output> output)

setReferencedTables

public void setReferencedTables(List<TableInfo> tables)

setRoutines

public void setRoutines(List<RoutineInfo> routines)

setDynamicFiltersStatsSupplier

public void setDynamicFiltersStatsSupplier(Supplier<DynamicFilterService.DynamicFiltersStats> dynamicFiltersStatsSupplier)

getSetSessionProperties

public Map<String,String> getSetSessionProperties()

setSetCatalog

public void setSetCatalog(String catalog)

setSetSchema

public void setSetSchema(String schema)

setSetPath

public void setSetPath(String path)

getSetPath

public String getSetPath()

addSetSessionProperties

public void addSetSessionProperties(String key,
                                    String value)

addSetRole

public void addSetRole(String catalog,
                       SelectedRole role)

getResetSessionProperties

public Set<String> getResetSessionProperties()

addResetSessionProperties

public void addResetSessionProperties(String name)

getAddedPreparedStatements

public Map<String,String> getAddedPreparedStatements()

getDeallocatedPreparedStatements

public Set<String> getDeallocatedPreparedStatements()

addPreparedStatement

public void addPreparedStatement(String key,
                                 String value)

removePreparedStatement

public void removePreparedStatement(String key)

setStartedTransactionId

public void setStartedTransactionId(TransactionId startedTransactionId)

clearTransactionId

public void clearTransactionId()

setUpdateType

public void setUpdateType(String updateType)

getQueryState

public QueryState getQueryState()

isDone

public boolean isDone()

transitionToWaitingForResources

public boolean transitionToWaitingForResources()

transitionToDispatching

public boolean transitionToDispatching()

transitionToPlanning

public boolean transitionToPlanning()

transitionToStarting

public boolean transitionToStarting()

transitionToRunning

public boolean transitionToRunning()

transitionToFinishing

public boolean transitionToFinishing()

transitionToFailed

public boolean transitionToFailed(Throwable throwable)

transitionToCanceled

public boolean transitionToCanceled()

addStateChangeListener

public void addStateChangeListener(StateMachine.StateChangeListener<QueryState> stateChangeListener)

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.

addQueryInfoStateChangeListener

public void addQueryInfoStateChangeListener(StateMachine.StateChangeListener<QueryInfo> stateChangeListener)

Add a listener for the final query 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.

getStateChange

public com.google.common.util.concurrent.ListenableFuture<QueryState> getStateChange(QueryState currentState)

recordHeartbeat

public void recordHeartbeat()

beginAnalysis

public void beginAnalysis()

endAnalysis

public void endAnalysis()

getCreateTime

public org.joda.time.DateTime getCreateTime()

getExecutionStartTime

public Optional<org.joda.time.DateTime> getExecutionStartTime()

getLastHeartbeat

public org.joda.time.DateTime getLastHeartbeat()

getEndTime

public Optional<org.joda.time.DateTime> getEndTime()

getFailureInfo

public Optional<ExecutionFailureInfo> getFailureInfo()

getFinalQueryInfo

public Optional<QueryInfo> getFinalQueryInfo()

updateQueryInfo

public QueryInfo updateQueryInfo(Optional<StageInfo> stageInfo)

pruneQueryInfo

public void pruneQueryInfo()