Method QueryAiGenerateAnalysisHandler

be/src/service/impala-http-handler.cc:455–486

}

void ImpalaHttpHandler::QueryAiGenerateAnalysisHandler(const Webserver::WebRequest& req,
    Document* document) {
  auto set_error = [&](const string& error_msg) {
    Value error(error_msg.c_str(), document->GetAllocator());
    document->AddMember("error", error, document->GetAllocator());
  };

  TUniqueId unique_id;
  Status status = ParseIdFromRequest(req, &unique_id, "query_id");
  if (!status.ok()) return set_error(status.GetDetail());

  ImpalaServer::RuntimeProfileOutput runtime_profile;
  Document profile_doc;
  profile_doc.SetObject();
  runtime_profile.json_output = &profile_doc;
  status = server_->GetRuntimeProfileOutput(
      unique_id, "", TRuntimeProfileFormat::JSON, &runtime_profile);
  if (!status.ok()) return set_error(status.GetDetail());

  const Value* profile_json = &profile_doc;
  if (profile_doc.HasMember("contents")) profile_json = &profile_doc["contents"];

  string analysis;
  status = GenerateAiAnalysisFromProfile(*profile_json, &analysis);
  if (!status.ok()) return set_error(status.GetDetail());

  Value query_id(PrintId(unique_id), document->GetAllocator());
  document->AddMember("query_id", query_id, document->GetAllocator());
  Value analysis_value(analysis, document->GetAllocator());
  document->AddMember("analysis", analysis_value, document->GetAllocator());
  document->AddMember("status", rapidjson::StringRef("OK"), document->GetAllocator());
}

void ImpalaHttpHandler::AddQueryRecordTips(Document* document) {
  document->AddMember("tips_query_id", "Unique ID of the query, click ID to jump to the "