Class NotImplementedQueryManager

java.lang.Object

io.trino.testing.NotImplementedQueryManager

All Implemented Interfaces:

QueryManager

public class NotImplementedQueryManager
extends Object
implements QueryManager

Constructor Summary

Constructors

Constructor	Description
NotImplementedQueryManager()

Method Summary

Modifier and Type	Method	Description
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 execution)	Creates a new query using the specified query execution.
void	failQuery(QueryId queryId, Throwable cause)	Attempts to fail the query for the specified reason.
QueryInfo	getFullQueryInfo(QueryId queryId)
List<BasicQueryInfo>	getQueries()
BasicQueryInfo	getQueryInfo(QueryId queryId)
Session	getQuerySession(QueryId queryId)
Slug	getQuerySlug(QueryId queryId)
QueryState	getQueryState(QueryId queryId)
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.

Methods inherited from class java.lang.Object

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

Constructor Details

NotImplementedQueryManager

public NotImplementedQueryManager()

Method Details

getQueries

public List<BasicQueryInfo> getQueries()

Specified by:

getQueries in interface QueryManager

setOutputInfoListener

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

Description copied from interface: QueryManager

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

Specified by:

setOutputInfoListener in interface QueryManager

Throws:

NoSuchElementException - if query does not exist

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)
                            throws NoSuchElementException

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

Throws:

NoSuchElementException - if query does not exist

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)
                            throws NoSuchElementException

Specified by:

getQueryInfo in interface QueryManager

Throws:

NoSuchElementException - if query does not exist

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)

Specified by:

getQuerySession in interface QueryManager

getQuerySlug

public Slug getQuerySlug(QueryId queryId)

Specified by:

getQuerySlug in interface QueryManager

getQueryState

public QueryState getQueryState(QueryId queryId)
                         throws NoSuchElementException

Specified by:

getQueryState in interface QueryManager

Throws:

NoSuchElementException - if query does not exist

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

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