Method StoreExecutionStats

be/src/service/impala-server.cc:1709–1759

}

Status ImpalaServer::StoreExecutionStats(const QueryHandle& query_handle) {
  if (!query_handle->query_options().store_hbo_stats) {
    return Status::OK();
  }
  if (query_handle->exec_state() != ClientRequestState::ExecState::FINISHED) {
    VLOG_QUERY << "Skip execution stats for query " << PrintId(query_handle->query_id())
        << " since query state is "
        << ClientRequestState::ExecStateToString(query_handle->exec_state());
    return Status::OK();
  }

  const TExecRequest& exec_req = query_handle->exec_request();
  if (query_handle->GetCoordinator() == nullptr) {
    LOG(ERROR) << "ExecSummary not found for HBO!";
    return Status("ExecSummary not found for HBO");
  }

  TExecSummary summary;
  query_handle->GetCoordinator()->GetTExecSummary(&summary);
  map<TPlanNodeId, TPlanNodeExecSummary> exec_summaries;
  for (const TPlanNodeExecSummary& s : summary.nodes) {
    exec_summaries[s.node_id] = s;
  }

  const auto& effective_filter_ids_to_node_ids =
      query_handle->GetCoordinator()->GetEffectiveFilterTargets();

  THistoricalStatsUpdate history_stats;
  vector<const TPlanNode*> nodes_with_hbo_keys;
  GetPlanNodesWithHboKeys(exec_req, &nodes_with_hbo_keys);
  for (const TPlanNode* p : nodes_with_hbo_keys) {
    // Skip nodes if runtime filters have filtered some rows. The output cardinality
    // depends on when the runtime filters arrive, which is unreliable.
    if (HasEffectiveRuntimeFilter(*p, effective_filter_ids_to_node_ids)) {
      LOG(INFO) << "Skip execution stats of " << p->label
          << " since it has effective runtime filters";
      continue;
    }

    // TODO: store the cumulative TExecStats instead of just cardinality
    int64_t cardinality = 0;
    if (!GetCardinalityIfComplete(exec_summaries[p->node_id], *p, &cardinality)) {
      continue;
    }
    AppendHboPlanNodeRuns(query_handle->query_id(), *p, cardinality, &history_stats);
  }
  if (history_stats.plan_node_runs.empty()) return Status::OK();
  history_stats.__isset.plan_node_runs = true;
  LOG(INFO) << "Storing history stats for query " << PrintId(query_handle->query_id());
  return exec_env_->frontend()->StoreExecStats(history_stats);
}

Status ImpalaServer::UnregisterQuery(
    const TUniqueId& query_id, const Status* cause, bool interrupted) {