Method FinishExecQueryOrDmlRequest

be/src/service/client-request-state.cc:678–784

}

void ClientRequestState::FinishExecQueryOrDmlRequest() {
  const TExecRequest& exec_req = exec_request();
  DCHECK(exec_req.__isset.query_exec_request);
  UniqueIdPB query_id_pb;
  TUniqueIdToUniqueIdPB(query_id(), &query_id_pb);

  if (otel_trace_query() && !IsCTAS()) {
    otel_trace_manager_->StartChildSpanAdmissionControl();
  }

  TQueryExecRequest req;
  req = exec_req.query_exec_request;
  req.query_ctx.client_request.query_options = exec_req.query_options;
  if (ExecEnv::GetInstance()->AdmissionServiceEnabled()) {
    if (req.__isset.query_plan) {
      // Use the swap() to ensure the string's memory is deallocated.
      // Using clear() sets the size to 0 but may not release the capacity.
      std::string().swap(req.query_plan);
      req.__isset.query_plan = false;
    }
    if (req.__isset.lineage_graph) {
      req.lineage_graph = TLineageGraph();
      req.__isset.lineage_graph = false;
    }
    if (req.__isset.result_set_metadata) {
      req.result_set_metadata = TResultSetMetadata();
      req.__isset.result_set_metadata = false;
    }
    if (req.__isset.finalize_params) {
      req.finalize_params = TFinalizeParams();
      req.__isset.finalize_params = false;
    }
    TClientRequest& client_req = req.query_ctx.client_request;
    if (client_req.__isset.redacted_stmt) {
      // Use the swap() to ensure the string's memory is deallocated.
      std::string().swap(client_req.redacted_stmt);
      client_req.__isset.redacted_stmt = false;
    }
  }

  DCHECK(!admission_exec_request_);
  admission_exec_request_ = std::make_unique<AdmissionExecRequestUncompressed>(&req);
  Status admit_status = admission_control_client_->SubmitForAdmission(
      {query_id_pb, ExecEnv::GetInstance()->backend_id(), *admission_exec_request_,
          summary_profile_, blacklisted_executor_addresses_},
      query_events_, &schedule_, &wait_start_time_ms_, &wait_end_time_ms_,
      otel_trace_manager_.get());

  if (otel_trace_query() && !IsCTAS()) {
    otel_trace_manager_->EndChildSpanAdmissionControl(admit_status);
  }

  {
    lock_guard<mutex> l(lock_);
    if (!UpdateQueryStatus(admit_status).ok()) return;
  }

  // Don't start the query execution span for the insert portion of a CTAS query since it