diff options
-rw-r--r-- | include/grpcpp/impl/codegen/config_protobuf.h | 16 | ||||
-rw-r--r-- | src/cpp/server/channelz/channelz_service.cc | 67 |
2 files changed, 33 insertions, 50 deletions
diff --git a/include/grpcpp/impl/codegen/config_protobuf.h b/include/grpcpp/impl/codegen/config_protobuf.h index 598848ee9f..94e593d1ef 100644 --- a/include/grpcpp/impl/codegen/config_protobuf.h +++ b/include/grpcpp/impl/codegen/config_protobuf.h @@ -66,11 +66,6 @@ #define GRPC_CUSTOM_CODEDINPUTSTREAM ::google::protobuf::io::CodedInputStream #endif -#ifndef GRPC_CUSTOM_UTIL_STATUS -#include <google/protobuf/util/json_util.h> -#define GRPC_CUSTOM_UTIL_STATUS ::google::protobuf::util::Status -#endif - namespace grpc { namespace protobuf { @@ -88,17 +83,6 @@ typedef GRPC_CUSTOM_SERVICEDESCRIPTOR ServiceDescriptor; typedef GRPC_CUSTOM_SIMPLEDESCRIPTORDATABASE SimpleDescriptorDatabase; typedef GRPC_CUSTOM_SOURCELOCATION SourceLocation; -namespace util { -typedef GRPC_CUSTOM_UTIL_STATUS Status; - -inline util::Status JsonStringToMessage(const std::string& input, - Message* message) { - return ::google::protobuf::util::JsonStringToMessage( - input, message, ::google::protobuf::util::JsonParseOptions()); -} - -} // namespace util - namespace io { typedef GRPC_CUSTOM_ZEROCOPYOUTPUTSTREAM ZeroCopyOutputStream; typedef GRPC_CUSTOM_ZEROCOPYINPUTSTREAM ZeroCopyInputStream; diff --git a/src/cpp/server/channelz/channelz_service.cc b/src/cpp/server/channelz/channelz_service.cc index 07383c5996..79ed9102e5 100644 --- a/src/cpp/server/channelz/channelz_service.cc +++ b/src/cpp/server/channelz/channelz_service.cc @@ -20,6 +20,9 @@ #include "src/cpp/server/channelz/channelz_service.h" +#include <google/protobuf/text_format.h> +#include <google/protobuf/util/json_util.h> + #include <grpc/grpc.h> #include <grpc/support/alloc.h> @@ -30,14 +33,13 @@ Status ChannelzService::GetTopChannels( channelz::v1::GetTopChannelsResponse* response) { char* json_str = grpc_channelz_get_top_channels(request->start_channel_id()); if (json_str == nullptr) { - return Status(StatusCode::INTERNAL, - "grpc_channelz_get_top_channels returned null"); + return Status(INTERNAL, "grpc_channelz_get_top_channels returned null"); } - grpc::protobuf::util::Status s = - grpc::protobuf::util::JsonStringToMessage(json_str, response); + google::protobuf::util::Status s = + google::protobuf::util::JsonStringToMessage(json_str, response); gpr_free(json_str); - if (!s.ok()) { - return Status(StatusCode::INTERNAL, s.ToString()); + if (s != google::protobuf::util::Status::OK) { + return Status(INTERNAL, s.ToString()); } return Status::OK; } @@ -47,14 +49,13 @@ Status ChannelzService::GetServers( channelz::v1::GetServersResponse* response) { char* json_str = grpc_channelz_get_servers(request->start_server_id()); if (json_str == nullptr) { - return Status(StatusCode::INTERNAL, - "grpc_channelz_get_servers returned null"); + return Status(INTERNAL, "grpc_channelz_get_servers returned null"); } - grpc::protobuf::util::Status s = - grpc::protobuf::util::JsonStringToMessage(json_str, response); + google::protobuf::util::Status s = + google::protobuf::util::JsonStringToMessage(json_str, response); gpr_free(json_str); - if (!s.ok()) { - return Status(StatusCode::INTERNAL, s.ToString()); + if (s != google::protobuf::util::Status::OK) { + return Status(INTERNAL, s.ToString()); } return Status::OK; } @@ -65,14 +66,13 @@ Status ChannelzService::GetServerSockets( char* json_str = grpc_channelz_get_server_sockets(request->server_id(), request->start_socket_id()); if (json_str == nullptr) { - return Status(StatusCode::INTERNAL, - "grpc_channelz_get_server_sockets returned null"); + return Status(INTERNAL, "grpc_channelz_get_server_sockets returned null"); } - grpc::protobuf::util::Status s = - grpc::protobuf::util::JsonStringToMessage(json_str, response); + google::protobuf::util::Status s = + google::protobuf::util::JsonStringToMessage(json_str, response); gpr_free(json_str); - if (!s.ok()) { - return Status(StatusCode::INTERNAL, s.ToString()); + if (s != google::protobuf::util::Status::OK) { + return Status(INTERNAL, s.ToString()); } return Status::OK; } @@ -82,13 +82,13 @@ Status ChannelzService::GetChannel( channelz::v1::GetChannelResponse* response) { char* json_str = grpc_channelz_get_channel(request->channel_id()); if (json_str == nullptr) { - return Status(StatusCode::NOT_FOUND, "No object found for that ChannelId"); + return Status(NOT_FOUND, "No object found for that ChannelId"); } - grpc::protobuf::util::Status s = - grpc::protobuf::util::JsonStringToMessage(json_str, response); + google::protobuf::util::Status s = + google::protobuf::util::JsonStringToMessage(json_str, response); gpr_free(json_str); - if (!s.ok()) { - return Status(StatusCode::INTERNAL, s.ToString()); + if (s != google::protobuf::util::Status::OK) { + return Status(INTERNAL, s.ToString()); } return Status::OK; } @@ -98,14 +98,13 @@ Status ChannelzService::GetSubchannel( channelz::v1::GetSubchannelResponse* response) { char* json_str = grpc_channelz_get_subchannel(request->subchannel_id()); if (json_str == nullptr) { - return Status(StatusCode::NOT_FOUND, - "No object found for that SubchannelId"); + return Status(NOT_FOUND, "No object found for that SubchannelId"); } - grpc::protobuf::util::Status s = - grpc::protobuf::util::JsonStringToMessage(json_str, response); + google::protobuf::util::Status s = + google::protobuf::util::JsonStringToMessage(json_str, response); gpr_free(json_str); - if (!s.ok()) { - return Status(StatusCode::INTERNAL, s.ToString()); + if (s != google::protobuf::util::Status::OK) { + return Status(INTERNAL, s.ToString()); } return Status::OK; } @@ -115,13 +114,13 @@ Status ChannelzService::GetSocket(ServerContext* unused, channelz::v1::GetSocketResponse* response) { char* json_str = grpc_channelz_get_socket(request->socket_id()); if (json_str == nullptr) { - return Status(StatusCode::NOT_FOUND, "No object found for that SocketId"); + return Status(NOT_FOUND, "No object found for that SocketId"); } - grpc::protobuf::util::Status s = - grpc::protobuf::util::JsonStringToMessage(json_str, response); + google::protobuf::util::Status s = + google::protobuf::util::JsonStringToMessage(json_str, response); gpr_free(json_str); - if (!s.ok()) { - return Status(StatusCode::INTERNAL, s.ToString()); + if (s != google::protobuf::util::Status::OK) { + return Status(INTERNAL, s.ToString()); } return Status::OK; } |