mirror of https://github.com/grpc/grpc.git
ncteisen
7 years ago
parent
b69f1f6aac
commit
c3c6e064b3
32 changed files with 579 additions and 33 deletions
@ -0,0 +1,164 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include <grpc/impl/codegen/port_platform.h> |
||||
|
||||
#include "src/core/lib/channel/channelz.h" |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <grpc/support/string_util.h> |
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
|
||||
#include "src/core/lib/channel/channelz_registry.h" |
||||
#include "src/core/lib/channel/status_util.h" |
||||
#include "src/core/lib/gpr/string.h" |
||||
#include "src/core/lib/gpr/useful.h" |
||||
#include "src/core/lib/gprpp/memory.h" |
||||
#include "src/core/lib/iomgr/error.h" |
||||
#include "src/core/lib/slice/slice_internal.h" |
||||
#include "src/core/lib/surface/channel.h" |
||||
#include "src/core/lib/transport/connectivity_state.h" |
||||
#include "src/core/lib/transport/error_utils.h" |
||||
|
||||
// TODO(ncteisen): actually implement this
|
||||
char* grpc_channelz_get_channel(intptr_t channel_id) { return nullptr; } |
||||
|
||||
namespace grpc_core { |
||||
namespace channelz { |
||||
|
||||
// TODO(ncteisen): more this functions to a loc where it can be used
|
||||
namespace { |
||||
|
||||
// returns an allocated string that represents tm according to RFC-3339, and,
|
||||
// more specifically, follows:
|
||||
// https://developers.google.com/protocol-buffers/docs/proto3#json
|
||||
//
|
||||
// "Uses RFC 3339, where generated output will always be Z-normalized and uses
|
||||
// 0, 3, 6 or 9 fractional digits."
|
||||
char* fmt_time(gpr_timespec tm) { |
||||
char time_buffer[35]; |
||||
char ns_buffer[11]; // '.' + 9 digits of precision
|
||||
struct tm* tm_info = localtime((const time_t*)&tm.tv_sec); |
||||
strftime(time_buffer, sizeof(time_buffer), "%Y-%m-%dT%H:%M:%S", tm_info); |
||||
snprintf(ns_buffer, 11, ".%09d", tm.tv_nsec); |
||||
// This loop trims off trailing zeros by inserting a null character that the
|
||||
// right point. We iterate in chunks of three because we want 0, 3, 6, or 9
|
||||
// fractional digits.
|
||||
for (int i = 7; i >= 1; i -= 3) { |
||||
if (ns_buffer[i] == '0' && ns_buffer[i + 1] == '0' && |
||||
ns_buffer[i + 2] == '0') { |
||||
ns_buffer[i] = '\0'; |
||||
// Edge case in which all fractional digits were 0.
|
||||
if (i == 1) { |
||||
ns_buffer[0] = '\0'; |
||||
} |
||||
} else { |
||||
break; |
||||
} |
||||
} |
||||
char* full_time_str; |
||||
gpr_asprintf(&full_time_str, "%s%sZ", time_buffer, ns_buffer); |
||||
return full_time_str; |
||||
} |
||||
|
||||
// TODO(ncteisen); move this to json library
|
||||
grpc_json* add_num_str(grpc_json* parent, grpc_json* it, const char* name, |
||||
uint64_t num) { |
||||
char* num_str; |
||||
gpr_asprintf(&num_str, "%" PRIu64, num); |
||||
return grpc_json_create_child(it, parent, name, num_str, GRPC_JSON_STRING, |
||||
true); |
||||
} |
||||
|
||||
} // namespace
|
||||
|
||||
Channel::Channel(grpc_channel* channel) : channel_(channel) { |
||||
target_ = grpc_channel_get_target(channel_); |
||||
channel_uuid_ = ChannelzRegistry::Register(this); |
||||
} |
||||
|
||||
Channel::~Channel() { |
||||
gpr_free(const_cast<char*>(target_)); |
||||
ChannelzRegistry::Unregister(channel_uuid_); |
||||
} |
||||
|
||||
void Channel::CallStarted() { |
||||
calls_started_++; |
||||
last_call_started_timestamp_ = grpc_millis_to_timespec( |
||||
grpc_core::ExecCtx::Get()->Now(), GPR_CLOCK_REALTIME); |
||||
} |
||||
|
||||
grpc_connectivity_state Channel::GetConnectivityState() { |
||||
if (channel_destroyed_) { |
||||
return GRPC_CHANNEL_SHUTDOWN; |
||||
} else { |
||||
return grpc_channel_check_connectivity_state(channel_, false); |
||||
} |
||||
} |
||||
|
||||
char* Channel::RenderJSON() { |
||||
// We need to track these three json objects to build our object
|
||||
grpc_json* top_level_json = grpc_json_create(GRPC_JSON_OBJECT); |
||||
grpc_json* json = top_level_json; |
||||
grpc_json* json_iterator = nullptr; |
||||
|
||||
// create and fill the ref child
|
||||
json_iterator = grpc_json_create_child(json_iterator, json, "ref", nullptr, |
||||
GRPC_JSON_OBJECT, true); |
||||
json = json_iterator; |
||||
json_iterator = nullptr; |
||||
json_iterator = add_num_str(json, json_iterator, "channelId", channel_uuid_); |
||||
|
||||
// reset json iterators to top level object
|
||||
json = top_level_json; |
||||
json_iterator = nullptr; |
||||
|
||||
// create and fill the data child
|
||||
json_iterator = grpc_json_create_child(json_iterator, json, "data", nullptr, |
||||
GRPC_JSON_OBJECT, true); |
||||
json = json_iterator; |
||||
json_iterator = nullptr; |
||||
json_iterator = |
||||
add_num_str(json, json_iterator, "callsStarted", calls_started_); |
||||
json_iterator = |
||||
add_num_str(json, json_iterator, "callsSucceeded", calls_succeeded_); |
||||
json_iterator = |
||||
add_num_str(json, json_iterator, "callsFailed", calls_failed_); |
||||
json_iterator = grpc_json_create_child( |
||||
json_iterator, json, "lastCallStartedTimestamp", |
||||
fmt_time(last_call_started_timestamp_), GRPC_JSON_STRING, true); |
||||
json_iterator = grpc_json_create_child(json_iterator, json, "target", target_, |
||||
GRPC_JSON_STRING, false); |
||||
grpc_connectivity_state connectivity_state = GetConnectivityState(); |
||||
json_iterator = |
||||
grpc_json_create_child(json_iterator, json, "state", |
||||
grpc_connectivity_state_name(connectivity_state), |
||||
GRPC_JSON_STRING, false); |
||||
|
||||
// render and return the over json object
|
||||
char* json_str = grpc_json_dump_to_string(top_level_json, 0); |
||||
grpc_json_destroy(top_level_json); |
||||
return json_str; |
||||
} |
||||
|
||||
} // namespace channelz
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,68 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2018 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_CORE_LIB_CHANNEL_CHANNELZ_H |
||||
#define GRPC_CORE_LIB_CHANNEL_CHANNELZ_H |
||||
|
||||
#include <grpc/impl/codegen/port_platform.h> |
||||
|
||||
#include <grpc/grpc.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/client_channel.h" |
||||
#include "src/core/lib/gprpp/ref_counted.h" |
||||
#include "src/core/lib/gprpp/ref_counted_ptr.h" |
||||
#include "src/core/lib/iomgr/error.h" |
||||
#include "src/core/lib/json/json.h" |
||||
|
||||
namespace grpc_core { |
||||
namespace channelz { |
||||
|
||||
// owned by the client_channel that it points to and tracks
|
||||
class Channel : public RefCounted<Channel> { |
||||
public: |
||||
Channel(grpc_channel* channel); |
||||
~Channel(); |
||||
|
||||
void CallStarted(); |
||||
void CallFailed() { calls_failed_++; } |
||||
void CallSucceeded() { calls_succeeded_++; } |
||||
|
||||
char* RenderJSON(); |
||||
|
||||
void set_channel_destroyed() { |
||||
GPR_ASSERT(!channel_destroyed_); |
||||
channel_destroyed_ = true; |
||||
} |
||||
|
||||
private: |
||||
bool channel_destroyed_ = false; |
||||
grpc_channel* channel_; |
||||
const char* target_; |
||||
uint64_t calls_started_ = 0; |
||||
uint64_t calls_succeeded_ = 0; |
||||
uint64_t calls_failed_ = 0; |
||||
gpr_timespec last_call_started_timestamp_; |
||||
intptr_t channel_uuid_; |
||||
|
||||
grpc_connectivity_state GetConnectivityState(); |
||||
}; |
||||
|
||||
} // namespace channelz
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif /* GRPC_CORE_LIB_CHANNEL_CHANNELZ_H */ |
||||
@ -0,0 +1,57 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
|
||||
#include <gtest/gtest.h> |
||||
|
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
|
||||
#include "src/core/lib/channel/channel_trace.h" |
||||
#include "src/core/lib/channel/channelz_registry.h" |
||||
#include "src/core/lib/gpr/useful.h" |
||||
#include "src/core/lib/iomgr/exec_ctx.h" |
||||
#include "src/core/lib/json/json.h" |
||||
|
||||
#include "test/core/util/test_config.h" |
||||
#include "test/cpp/util/channel_trace_proto_helper.h" |
||||
|
||||
// remove me
|
||||
#include <grpc/support/string_util.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
|
||||
namespace grpc_core { |
||||
namespace testing { |
||||
namespace {} // anonymous namespace
|
||||
|
||||
TEST(ChannelzTest, Channel) {} |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc_core
|
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc_test_init(argc, argv); |
||||
grpc_init(); |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
int ret = RUN_ALL_TESTS(); |
||||
grpc_shutdown(); |
||||
return ret; |
||||
} |
||||
Loading…
Reference in new issue