Method executeAndWait

be/src/service/impala-beeswax-server.cc:98–155

}

void ImpalaServer::executeAndWait(beeswax::QueryHandle& beeswax_handle,
    const Query& query, const LogContextId& client_ctx) {
  VLOG_QUERY << "executeAndWait(): query=" << query.query;
  RAISE_IF_ERROR(CheckNotShuttingDown(), SQLSTATE_GENERAL_ERROR);
  ScopedSessionState session_handle(this);
  shared_ptr<SessionState> session;
  RAISE_IF_ERROR(
      session_handle.WithBeeswaxSession(ThriftServer::GetThreadConnectionId(), &session),
      SQLSTATE_GENERAL_ERROR);
  TQueryCtx query_ctx;
  // raise general error for request conversion error;
  RAISE_IF_ERROR(QueryToTQueryContext(query, &query_ctx), SQLSTATE_GENERAL_ERROR);

  DCHECK(session != nullptr);  // The session should exist.
  {
    // The session is created when the client connects. Depending on the underlying
    // transport, the username may be known at that time. If the username hasn't been set
    // yet, set it now.
    lock_guard<mutex> l(session->lock);
    if (session->connected_user.empty()) {
      session->connected_user = query.hadoop_user.empty() ?
          FLAGS_anonymous_user_name : query.hadoop_user;
    }
  }

  // raise Syntax error or access violation; it's likely to be syntax/analysis error
  // TODO: that may not be true; fix this
  QueryHandle query_handle;
  RAISE_IF_ERROR(Execute(&query_ctx, session, &query_handle, nullptr),
      SQLSTATE_SYNTAX_ERROR_OR_ACCESS_VIOLATION);

  // Make query id available to the following RaiseBeeswaxException().
  ScopedThreadContext scoped_tdi(GetThreadDebugInfo(), query_handle->query_id());

  // Once the query is running do a final check for session closure and add it to the
  // set of in-flight queries.
  Status status = SetQueryInflight(session, query_handle);
  if (!status.ok()) {
    discard_result(UnregisterQuery(query_handle->query_id(), &status));
    RaiseBeeswaxException(status.GetDetail(), SQLSTATE_GENERAL_ERROR);
  }
  // block until results are ready
  query_handle->Wait();
  {
    lock_guard<mutex> l(*query_handle->lock());
    status = query_handle->query_status();
  }
  if (!status.ok()) {
    discard_result(UnregisterQuery(query_handle->query_id(), &status));
    RaiseBeeswaxException(status.GetDetail(), SQLSTATE_GENERAL_ERROR);
  }

  TUniqueIdToBeeswaxHandle(query_handle->query_id(), &beeswax_handle);

  // If the input log context id is an empty string, then create a new number and
  // set it to _return. Otherwise, set _return with the input log context
  beeswax_handle.log_context = client_ctx.empty() ? beeswax_handle.id : client_ctx;
}