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

   1 //===--- Monitor.cpp - Request server monitoring information through CLI --===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "MonitoringService.grpc.pb.h"
  10 #include "MonitoringService.pb.h"
  11
  12 #include "support/Logger.h"
  13 #include "clang/Basic/Version.h"
  14 #include "llvm/Support/CommandLine.h"
  15 #include "llvm/Support/Signals.h"
  16
  17 #include <chrono>
  18 #include <google/protobuf/util/json_util.h>
  19 #include <grpc++/grpc++.h>
  20
  21 namespace clang {
  22 namespace clangd {
  23 namespace remote {
  24 namespace {
  25
  26 static constexpr char Overview[] = R"(
  27 This tool requests monitoring information (uptime, index freshness) from the
  28 server and prints it to stdout.
  29 )";
  30
  31 llvm::cl::opt<std::string>
  32     ServerAddress("server-address", llvm::cl::Positional,
  33                   llvm::cl::desc("Address of the invoked server."),
  34                   llvm::cl::Required);
  35
  36 } // namespace
  37 } // namespace remote
  38 } // namespace clangd
  39 } // namespace clang
  40
  41 int main(int argc, char *argv[]) {
  42   using namespace clang::clangd::remote;
  43   llvm::cl::ParseCommandLineOptions(argc, argv, Overview);
  44   llvm::sys::PrintStackTraceOnErrorSignal(argv[0]);
  45
  46   const auto Channel =
  47       grpc::CreateChannel(ServerAddress, grpc::InsecureChannelCredentials());
  48   const auto Stub = clang::clangd::remote::v1::Monitor::NewStub(Channel);
  49   grpc::ClientContext Context;
  50   Context.set_deadline(std::chrono::system_clock::now() +
  51                        std::chrono::seconds(10));
  52   Context.AddMetadata("version", clang::getClangToolFullVersion("clangd"));
  53   const clang::clangd::remote::v1::MonitoringInfoRequest Request;
  54   clang::clangd::remote::v1::MonitoringInfoReply Response;
  55   const auto Status = Stub->MonitoringInfo(&Context, Request, &Response);
  56   if (!Status.ok()) {
  57     clang::clangd::elog("Can not request monitoring information ({0}): {1}\n",
  58                         Status.error_code(), Status.error_message());
  59     return -1;
  60   }
  61   std::string Output;
  62   google::protobuf::util::JsonPrintOptions Options;
  63   Options.add_whitespace = true;
  64   Options.always_print_primitive_fields = true;
  65   Options.preserve_proto_field_names = true;
  66   const auto JsonStatus =
  67       google::protobuf::util::MessageToJsonString(Response, &Output, Options);
  68   if (!JsonStatus.ok()) {
  69     clang::clangd::elog("Can not convert response ({0}) to JSON ({1}): {2}\n",
  70                         Response.DebugString(),
  71                         static_cast<int>(JsonStatus.code()),
  72                         JsonStatus.message().as_string());
  73     return -1;
  74   }
  75   llvm::outs() << Output;
  76 }