|
|
|
|
/*
|
|
|
|
|
*
|
|
|
|
|
* Copyright 2017, Google Inc.
|
|
|
|
|
* All rights reserved.
|
|
|
|
|
*
|
|
|
|
|
* Redistribution and use in source and binary forms, with or without
|
|
|
|
|
* modification, are permitted provided that the following conditions are
|
|
|
|
|
* met:
|
|
|
|
|
*
|
|
|
|
|
* * Redistributions of source code must retain the above copyright
|
|
|
|
|
* notice, this list of conditions and the following disclaimer.
|
|
|
|
|
* * Redistributions in binary form must reproduce the above
|
|
|
|
|
* copyright notice, this list of conditions and the following disclaimer
|
|
|
|
|
* in the documentation and/or other materials provided with the
|
|
|
|
|
* distribution.
|
|
|
|
|
* * Neither the name of Google Inc. nor the names of its
|
|
|
|
|
* contributors may be used to endorse or promote products derived from
|
|
|
|
|
* this software without specific prior written permission.
|
|
|
|
|
*
|
|
|
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
|
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
|
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
|
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
|
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
|
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
|
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
|
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
|
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
|
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
|
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
|
*
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#include <string>
|
|
|
|
|
|
|
|
|
|
#include <google/protobuf/util/json_util.h>
|
|
|
|
|
#include <grpc++/grpc++.h>
|
|
|
|
|
#include <grpc/support/log.h>
|
|
|
|
|
|
|
|
|
|
#include "gflags/gflags.h"
|
|
|
|
|
#include "mongoose.h"
|
|
|
|
|
|
|
|
|
|
// TODO (makdharma): remove local copies of these protos
|
|
|
|
|
#include "tools/grpcz/census.grpc.pb.h"
|
|
|
|
|
#include "tools/grpcz/monitoring.grpc.pb.h"
|
|
|
|
|
|
|
|
|
|
DEFINE_string(
|
|
|
|
|
grpcz_server, "127.0.0.1:8080",
|
|
|
|
|
"Unix domain socket path (e.g. unix://tmp/grpcz.sock) or IP address"
|
|
|
|
|
"(host:port) where grpcz server is running.");
|
|
|
|
|
DEFINE_string(http_port, "8000",
|
|
|
|
|
"Port id for accessing the HTTP server that renders /grpcz page");
|
|
|
|
|
DEFINE_bool(print_to_console, false,
|
|
|
|
|
"print the JSON retreived from grpcz server and quit");
|
|
|
|
|
|
|
|
|
|
using grpc::Channel;
|
|
|
|
|
using grpc::ClientContext;
|
|
|
|
|
using grpc::Status;
|
|
|
|
|
|
|
|
|
|
using ::grpc::instrumentation::v1alpha::CanonicalRpcStats;
|
|
|
|
|
using ::grpc::instrumentation::v1alpha::Monitoring;
|
|
|
|
|
|
|
|
|
|
static const std::string static_html_header =
|
|
|
|
|
"<!DOCTYPE html> <html> <head> <style> \
|
|
|
|
|
table { border-collapse: collapse; width: 100%; } \
|
|
|
|
|
table, td, th { border: 1px solid black; } \
|
|
|
|
|
</style> </head> <body>\
|
|
|
|
|
<div id='stats' stats='";
|
|
|
|
|
|
|
|
|
|
static const std::string static_html_footer =
|
|
|
|
|
"' class='hidden'></div>\
|
|
|
|
|
<h1>GRPCZ Statistics</h1> <div id='table'> </div> \
|
|
|
|
|
<script> \
|
|
|
|
|
var canonical_stats = JSON.parse(\
|
|
|
|
|
document.getElementById('stats').getAttribute('stats')); \
|
|
|
|
|
var table = document.createElement('table'); \
|
|
|
|
|
if (canonical_stats['Error Message'] != undefined) { \
|
|
|
|
|
document.getElementById('table').innerHTML = canonical_stats['Error Message']; } \
|
|
|
|
|
else {\
|
|
|
|
|
for (var key in canonical_stats) { \
|
|
|
|
|
name = canonical_stats[key]['view']['viewName']; \
|
|
|
|
|
distribution = canonical_stats[key]['view']['distributionView']; \
|
|
|
|
|
interval = canonical_stats[key]['view']['intervalView']; \
|
|
|
|
|
value = (interval == undefined) ? \
|
|
|
|
|
JSON.stringify(distribution, null, ' ') : \
|
|
|
|
|
JSON.stringify(interval, null, ' '); \
|
|
|
|
|
var row = table.insertRow(-1); \
|
|
|
|
|
var col1 = row.insertCell(0); \
|
|
|
|
|
var col2 = row.insertCell(1); \
|
|
|
|
|
col1.innerHTML = name; \
|
|
|
|
|
col2.innerHTML = '<pre>' + value + '</pre>'; \
|
|
|
|
|
} \
|
|
|
|
|
document.getElementById('table').appendChild(table); \
|
|
|
|
|
}\
|
|
|
|
|
</script> </body> </html>";
|
|
|
|
|
|
|
|
|
|
class GrpczClient {
|
|
|
|
|
public:
|
|
|
|
|
GrpczClient(std::shared_ptr<Channel> channel)
|
|
|
|
|
: stub_(Monitoring::NewStub(channel)) {}
|
|
|
|
|
|
|
|
|
|
std::string GetStatsAsJson() {
|
|
|
|
|
const ::google::protobuf::Empty request;
|
|
|
|
|
CanonicalRpcStats reply;
|
|
|
|
|
ClientContext context;
|
|
|
|
|
Status status = stub_->GetCanonicalRpcStats(&context, request, &reply);
|
|
|
|
|
|
|
|
|
|
if (status.ok()) {
|
|
|
|
|
std::string json_str;
|
|
|
|
|
::google::protobuf::util::MessageToJsonString(reply, &json_str);
|
|
|
|
|
return json_str;
|
|
|
|
|
} else {
|
|
|
|
|
static const std::string error_message_json =
|
|
|
|
|
"{\"Error Message\":\"" + status.error_message() + "\"}";
|
|
|
|
|
gpr_log(GPR_DEBUG, "%d: %s", status.error_code(),
|
|
|
|
|
status.error_message().c_str());
|
|
|
|
|
return error_message_json;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
std::unique_ptr<Monitoring::Stub> stub_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
static struct mg_serve_http_opts s_http_server_opts;
|
|
|
|
|
std::unique_ptr<GrpczClient> g_grpcz_client;
|
|
|
|
|
|
|
|
|
|
static void ev_handler(struct mg_connection *nc, int ev, void *p) {
|
|
|
|
|
if (ev == MG_EV_HTTP_REQUEST) {
|
|
|
|
|
mg_serve_http(nc, (struct http_message *)p, s_http_server_opts);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void grpcz_handler(struct mg_connection *nc, int ev, void *ev_data) {
|
|
|
|
|
(void)ev;
|
|
|
|
|
(void)ev_data;
|
|
|
|
|
gpr_log(GPR_INFO, "fetching grpcz stats from %s", FLAGS_grpcz_server.c_str());
|
|
|
|
|
std::string json_str = g_grpcz_client->GetStatsAsJson();
|
|
|
|
|
std::string rendered_html =
|
|
|
|
|
static_html_header + json_str + static_html_footer;
|
|
|
|
|
mg_printf(nc, "HTTP/1.0 200 OK\r\n\r\n%s", rendered_html.c_str());
|
|
|
|
|
nc->flags |= MG_F_SEND_AND_CLOSE;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int main(int argc, char **argv) {
|
|
|
|
|
gflags::ParseCommandLineFlags(&argc, &argv, true);
|
|
|
|
|
|
|
|
|
|
// Create a client
|
|
|
|
|
g_grpcz_client.reset(new GrpczClient(grpc::CreateChannel(
|
|
|
|
|
FLAGS_grpcz_server, grpc::InsecureChannelCredentials())));
|
|
|
|
|
if (FLAGS_print_to_console) {
|
|
|
|
|
// using GPR_ERROR since this is the default verbosity. _DEBUG or _INFO
|
|
|
|
|
// won't print unless GRPC_VERBOSITY env var is set appropriately, which
|
|
|
|
|
// might confuse users of this utility.
|
|
|
|
|
gpr_log(GPR_ERROR, "%s\n", g_grpcz_client->GetStatsAsJson().c_str());
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Set up a mongoose webserver handler
|
|
|
|
|
struct mg_mgr mgr;
|
|
|
|
|
mg_mgr_init(&mgr, NULL);
|
|
|
|
|
gpr_log(GPR_INFO, "Starting grpcz web server on port %s\n",
|
|
|
|
|
FLAGS_http_port.c_str());
|
|
|
|
|
|
|
|
|
|
struct mg_connection *nc = mg_bind(&mgr, FLAGS_http_port.c_str(), ev_handler);
|
|
|
|
|
if (nc == NULL) {
|
|
|
|
|
gpr_log(GPR_ERROR, "Failed to create listener on port %s\n",
|
|
|
|
|
FLAGS_http_port.c_str());
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
mg_register_http_endpoint(nc, "/grpcz", grpcz_handler);
|
|
|
|
|
mg_set_protocol_http_websocket(nc);
|
|
|
|
|
|
|
|
|
|
// Poll in a loop and serve /grpcz pages
|
|
|
|
|
for (;;) {
|
|
|
|
|
static const int k_sleep_millis = 100;
|
|
|
|
|
mg_mgr_poll(&mgr, k_sleep_millis);
|
|
|
|
|
}
|
|
|
|
|
mg_mgr_free(&mgr);
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
