tools/grpcz/grpcz_client.cc - platform/external/grpc-grpc - Gitiles

 /*
  *
  * 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' data-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('data-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;
 }
	/*
	*
	* 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' data-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('data-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;
	}