@Override public void addStateChangeListener(StateChangeListener<QueryState> stateChangeListener) { stateMachine.addStateChangeListener(stateChangeListener); }
@Override public void addStateChangeListener(StateChangeListener<QueryState> stateChangeListener) { try (SetThreadName ignored = new SetThreadName("Query-%s", stateMachine.getQueryId())) { stateMachine.addStateChangeListener(stateChangeListener); } }
metadata, warningCollector); queryStateMachine.addStateChangeListener(newState -> QUERY_STATE_LOG.debug("Query %s is %s", queryStateMachine.getQueryId(), newState));
stateMachine.addStateChangeListener(state -> { if (!state.isDone()) { return;
queryStateMachine.addStateChangeListener(newState -> { if (newState.isDone()) { queryStateMachine.updateQueryInfo(Optional.ofNullable(getStageInfo()));
stateMachine.addStateChangeListener(state -> { if (state.isDone()) { closeSplitSources(outputStageExecutionPlan);
@Override public void addStateChangeListener(StateChangeListener<QueryState> stateChangeListener) { stateMachine.addStateChangeListener(stateChangeListener); }
@Override public void addStateChangeListener(StateChangeListener<QueryState> stateChangeListener) { try (SetThreadName ignored = new SetThreadName("Query-%s", stateMachine.getQueryId())) { stateMachine.addStateChangeListener(stateChangeListener); } }
stateMachine.addStateChangeListener(state -> { if (!state.isDone()) { return;
/** * Created QueryStateMachines must be transitioned to terminal states to clean up resources. */ public static QueryStateMachine begin(QueryId queryId, String query, Session session, URI self, boolean transactionControl, TransactionManager transactionManager, Executor executor) { session.getTransactionId().ifPresent(transactionControl ? transactionManager::trySetActive : transactionManager::checkAndSetActive); Session querySession; boolean autoCommit = !session.getTransactionId().isPresent() && !transactionControl; if (autoCommit) { // TODO: make autocommit isolation level a session parameter TransactionId transactionId = transactionManager.beginTransaction(true); querySession = session.withTransactionId(transactionId); } else { querySession = session; } QueryStateMachine queryStateMachine = new QueryStateMachine(queryId, query, querySession, self, autoCommit, transactionManager, executor); queryStateMachine.addStateChangeListener(newState -> log.debug("Query %s is %s", queryId, newState)); queryStateMachine.addStateChangeListener(newState -> { if (newState.isDone()) { session.getTransactionId().ifPresent(transactionManager::trySetInactive); } }); return queryStateMachine; }