clang-tools-extra/clangd/index/remote/monitor/Monitor.cpp

//===--- Monitor.cpp - Request server monitoring information through CLI --===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "MonitoringService.grpc.pb.h"
#include "MonitoringService.pb.h"

#include "support/Logger.h"
#include "clang/Basic/Version.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Signals.h"

#include <chrono>
#include <google/protobuf/util/json_util.h>
#include <grpc++/grpc++.h>

namespace clang {
namespace clangd {
namespace remote {
namespace {

static constexpr char Overview[] = R"(
This tool requests monitoring information (uptime, index freshness) from the
server and prints it to stdout.
)";

llvm::cl::opt<std::string>
    ServerAddress("server-address", llvm::cl::Positional,
                  llvm::cl::desc("Address of the invoked server."),
                  llvm::cl::Required);

} // namespace
} // namespace remote
} // namespace clangd
} // namespace clang

int main(int argc, char *argv[]) {
  using namespace clang::clangd::remote;
  llvm::cl::ParseCommandLineOptions(argc, argv, Overview);
  llvm::sys::PrintStackTraceOnErrorSignal(argv[0]);

  const auto Channel =
      grpc::CreateChannel(ServerAddress, grpc::InsecureChannelCredentials());
  const auto Stub = clang::clangd::remote::v1::Monitor::NewStub(Channel);
  grpc::ClientContext Context;
  Context.set_deadline(std::chrono::system_clock::now() +
                       std::chrono::seconds(10));
  Context.AddMetadata("version", clang::getClangToolFullVersion("clangd"));
  const clang::clangd::remote::v1::MonitoringInfoRequest Request;
  clang::clangd::remote::v1::MonitoringInfoReply Response;
  const auto Status = Stub->MonitoringInfo(&Context, Request, &Response);
  if (!Status.ok()) {
    clang::clangd::elog("Can not request monitoring information ({0}): {1}\n",
                        Status.error_code(), Status.error_message());
    return -1;
  }
  std::string Output;
  google::protobuf::util::JsonPrintOptions Options;
  Options.add_whitespace = true;
  Options.always_print_primitive_fields = true;
  Options.preserve_proto_field_names = true;
  const auto JsonStatus =
      google::protobuf::util::MessageToJsonString(Response, &Output, Options);
  if (!JsonStatus.ok()) {
    clang::clangd::elog("Can not convert response ({0}) to JSON ({1}): {2}\n",
                        Response.DebugString(), JsonStatus.error_code(),
                        JsonStatus.error_message().as_string());
    return -1;
  }
  llvm::outs() << Output;
}