Method UnregisterQuery

be/src/service/impala-server.cc:1761–1795

}

Status ImpalaServer::UnregisterQuery(
    const TUniqueId& query_id, const Status* cause, bool interrupted) {
  VLOG_QUERY << "UnregisterQuery(): query_id=" << PrintId(query_id);

  QueryHandle query_handle;
  // Skips updating RPCs since we'll finalize them right after, and this avoids
  // acquiring a ClientRequestState lock.
  RETURN_IF_ERROR(GetActiveQueryHandle(query_id, &query_handle, /* skip_rpcs */ true));

  DebugActionNoFail(query_handle->query_options(), "FINALIZE_INFLIGHT_QUERY");

  if (interrupted) {
    // Register interrupted query status with the session.
    shared_ptr<SessionState> session = query_handle->session();
    lock_guard<mutex> l(session->lock);
    if (!session->closed) {
      lock_guard<mutex> l(interrupted_query_statuses_lock_);
      interrupted_query_statuses_.emplace(query_id, *cause);
      session->interrupted_queries.emplace_back(query_id);
    }
  }

  // Do the work of unregistration that needs to be done synchronously. Once
  // Finalize() returns, the query is considered unregistered from the client's point of
  // view. If Finalize() returns OK, this thread is responsible for doing the
  // unregistration work. Finalize() succeeds for the first thread to call it to avoid
  // multiple threads unregistering.
  RETURN_IF_ERROR(
      query_handle.query_driver()->Finalize(&query_handle, cause));

  // Do the rest of the unregistration work in the background so that the client does
  // not need to wait for profile serialization, etc.
  unreg_thread_pool_->Offer(move(query_handle));
  return Status::OK();
}

void ImpalaServer::FinishUnregisterQuery(const QueryHandle& query_handle) {
  DCHECK_EQ(this, ExecEnv::GetInstance()->impala_server());