Package io.trino.execution

Class SqlQueryManager

java.lang.Object

io.trino.execution.SqlQueryManager

All Implemented Interfaces:

QueryManager

@ThreadSafe
public class SqlQueryManager
extends Object
implements QueryManager

Constructor Summary

Constructors

Constructor	Description
SqlQueryManager(ClusterMemoryManager memoryManager, io.opentelemetry.api.trace.Tracer tracer, QueryManagerConfig queryManagerConfig)

Method Summary

Modifier and Type	Method	Description
void	addFinalQueryInfoListener(QueryId queryId, StateMachine.StateChangeListener<QueryInfo> stateChangeListener)
void	addStateChangeListener(QueryId queryId, StateMachine.StateChangeListener<QueryState> listener)	Add a listener that fires each time the query state changes.
void	cancelQuery(QueryId queryId)	Attempts to fail the query due to a user cancellation.
void	cancelStage(StageId stageId)	Attempts to cancel the stage and continue the query.
void	createQuery(QueryExecution queryExecution)	Creates a new query using the specified query execution.
void	failQuery(QueryId queryId, Throwable cause)	Attempts to fail the query for the specified reason.
io.airlift.concurrent.ThreadPoolExecutorMBean	getExecutor()
QueryInfo	getFullQueryInfo(QueryId queryId)
io.airlift.concurrent.ThreadPoolExecutorMBean	getManagementExecutor()
List<BasicQueryInfo>	getQueries()
BasicQueryInfo	getQueryInfo(QueryId queryId)
Optional<Plan>	getQueryPlan(QueryId queryId)
Session	getQuerySession(QueryId queryId)
Slug	getQuerySlug(QueryId queryId)
QueryState	getQueryState(QueryId queryId)
QueryTracker<QueryExecution>	getQueryTracker()
ResultQueryInfo	getResultQueryInfo(QueryId queryId)
com.google.common.util.concurrent.ListenableFuture<QueryState>	getStateChange(QueryId queryId, QueryState currentState)	Gets a future that completes when the query changes from the specified current state or immediately if the query is already in a final state.
boolean	hasQuery(QueryId queryId)
void	outputTaskFailed(TaskId taskId, Throwable failure)	Notify that one of the output tasks failed for a given query
void	recordHeartbeat(QueryId queryId)	Updates the client heartbeat time, to prevent the query from be automatically purged.
void	resultsConsumed(QueryId queryId)	Notify that the query results for a query have been fully consumed by the client
void	setOutputInfoListener(QueryId queryId, Consumer<QueryExecution.QueryOutputInfo> listener)	Add a listener that fires once the query output locations are known.
void	start()
void	stop()

Methods inherited from class java.lang.Object

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

Constructor Details

SqlQueryManager

@Inject
public SqlQueryManager(ClusterMemoryManager memoryManager,
 io.opentelemetry.api.trace.Tracer tracer,
 QueryManagerConfig queryManagerConfig)

Method Details

start

@PostConstruct
public void start()

stop

@PreDestroy
public void stop()

getQueries

public List<BasicQueryInfo> getQueries()

Specified by:

getQueries in interface QueryManager

setOutputInfoListener

public void setOutputInfoListener(QueryId queryId,
 Consumer<QueryExecution.QueryOutputInfo> listener)

Description copied from interface: QueryManager

Add a listener that fires once the query output locations are known.

Specified by:

setOutputInfoListener in interface QueryManager

outputTaskFailed

public void outputTaskFailed(TaskId taskId,
 Throwable failure)

Description copied from interface: QueryManager

Notify that one of the output tasks failed for a given query

Specified by:

outputTaskFailed in interface QueryManager

resultsConsumed

public void resultsConsumed(QueryId queryId)

Description copied from interface: QueryManager

Notify that the query results for a query have been fully consumed by the client

Specified by:

resultsConsumed in interface QueryManager

addStateChangeListener

public void addStateChangeListener(QueryId queryId,
 StateMachine.StateChangeListener<QueryState> listener)

Description copied from interface: QueryManager

Add a listener that fires each time the query state changes. 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.

Specified by:

addStateChangeListener in interface QueryManager

getStateChange

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

Description copied from interface: QueryManager

Gets a future that completes when the query changes from the specified current state or immediately if the query is already in a final state. If the query does not exist, the future will contain a NoSuchElementException

Specified by:

getStateChange in interface QueryManager

getQueryInfo

public BasicQueryInfo getQueryInfo(QueryId queryId)

Specified by:

getQueryInfo in interface QueryManager

getFullQueryInfo

public QueryInfo getFullQueryInfo(QueryId queryId)
                           throws NoSuchElementException

Specified by:

getFullQueryInfo in interface QueryManager

Throws:

NoSuchElementException - if query does not exist

getResultQueryInfo

public ResultQueryInfo getResultQueryInfo(QueryId queryId)
                                   throws NoSuchElementException

Specified by:

getResultQueryInfo in interface QueryManager

Throws:

NoSuchElementException - if query does not exist

getQuerySession

public Session getQuerySession(QueryId queryId)
                        throws NoSuchElementException

Specified by:

getQuerySession in interface QueryManager

Throws:

NoSuchElementException - if query does not exist

getQuerySlug

public Slug getQuerySlug(QueryId queryId)

Specified by:

getQuerySlug in interface QueryManager

getQueryPlan

public Optional<Plan> getQueryPlan(QueryId queryId)

addFinalQueryInfoListener

public void addFinalQueryInfoListener(QueryId queryId,
 StateMachine.StateChangeListener<QueryInfo> stateChangeListener)

getQueryState

public QueryState getQueryState(QueryId queryId)

Specified by:

getQueryState in interface QueryManager

hasQuery

public boolean hasQuery(QueryId queryId)

Specified by:

hasQuery in interface QueryManager

recordHeartbeat

public void recordHeartbeat(QueryId queryId)

Description copied from interface: QueryManager

Updates the client heartbeat time, to prevent the query from be automatically purged. If the query does not exist, the call is ignored.

Specified by:

recordHeartbeat in interface QueryManager

createQuery

public void createQuery(QueryExecution queryExecution)

Description copied from interface: QueryManager

Creates a new query using the specified query execution.

Specified by:

createQuery in interface QueryManager

failQuery

public void failQuery(QueryId queryId,
 Throwable cause)

Description copied from interface: QueryManager

Attempts to fail the query for the specified reason. If the query is already in a final state, the call is ignored. If the query does not exist, the call is ignored.

Specified by:

failQuery in interface QueryManager

cancelQuery

public void cancelQuery(QueryId queryId)

Description copied from interface: QueryManager

Attempts to fail the query due to a user cancellation. If the query is already in a final state, the call is ignored. If the query does not exist, the call is ignored.

Specified by:

cancelQuery in interface QueryManager

cancelStage

public void cancelStage(StageId stageId)

Description copied from interface: QueryManager

Attempts to cancel the stage and continue the query. If the stage is already in a final state, the call is ignored. If the query does not exist, the call is ignored.

Specified by:

cancelStage in interface QueryManager

getExecutor

public io.airlift.concurrent.ThreadPoolExecutorMBean getExecutor()

getManagementExecutor

public io.airlift.concurrent.ThreadPoolExecutorMBean getManagementExecutor()

getQueryTracker

public QueryTracker<QueryExecution> getQueryTracker()