mirror of https://github.com/grpc/grpc.git
Craig Tiller
8 years ago
commit
56a84848c7
24 changed files with 759 additions and 212 deletions
@ -0,0 +1,90 @@ |
||||
/*
|
||||
* |
||||
* 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 "src/cpp/util/core_stats.h" |
||||
|
||||
#include <grpc/support/log.h> |
||||
|
||||
using grpc::core::Bucket; |
||||
using grpc::core::Histogram; |
||||
using grpc::core::Metric; |
||||
using grpc::core::Stats; |
||||
|
||||
namespace grpc { |
||||
|
||||
void CoreStatsToProto(const grpc_stats_data& core, Stats* proto) { |
||||
for (int i = 0; i < GRPC_STATS_COUNTER_COUNT; i++) { |
||||
Metric* m = proto->add_metrics(); |
||||
m->set_name(grpc_stats_counter_name[i]); |
||||
m->set_count(core.counters[i]); |
||||
} |
||||
for (int i = 0; i < GRPC_STATS_HISTOGRAM_COUNT; i++) { |
||||
Metric* m = proto->add_metrics(); |
||||
m->set_name(grpc_stats_histogram_name[i]); |
||||
Histogram* h = m->mutable_histogram(); |
||||
for (int j = 0; j < grpc_stats_histo_buckets[i]; j++) { |
||||
Bucket* b = h->add_buckets(); |
||||
b->set_start(grpc_stats_histo_bucket_boundaries[i][j]); |
||||
b->set_count(core.histograms[grpc_stats_histo_start[i] + j]); |
||||
} |
||||
} |
||||
} |
||||
|
||||
void ProtoToCoreStats(const grpc::core::Stats& proto, grpc_stats_data* core) { |
||||
memset(core, 0, sizeof(*core)); |
||||
for (const auto& m : proto.metrics()) { |
||||
switch (m.value_case()) { |
||||
case Metric::VALUE_NOT_SET: |
||||
break; |
||||
case Metric::kCount: |
||||
for (int i = 0; i < GRPC_STATS_COUNTER_COUNT; i++) { |
||||
if (m.name() == grpc_stats_counter_name[i]) { |
||||
core->counters[i] = m.count(); |
||||
break; |
||||
} |
||||
} |
||||
break; |
||||
case Metric::kHistogram: |
||||
for (int i = 0; i < GRPC_STATS_HISTOGRAM_COUNT; i++) { |
||||
if (m.name() == grpc_stats_histogram_name[i]) { |
||||
const auto& h = m.histogram(); |
||||
bool valid = true; |
||||
if (grpc_stats_histo_buckets[i] != h.buckets_size()) valid = false; |
||||
for (int j = 0; valid && j < h.buckets_size(); j++) { |
||||
if (grpc_stats_histo_bucket_boundaries[i][j] != |
||||
h.buckets(j).start()) { |
||||
valid = false; |
||||
} |
||||
} |
||||
if (!valid) { |
||||
gpr_log(GPR_ERROR, |
||||
"Found histogram %s but shape is different from proto", |
||||
m.name().c_str()); |
||||
} |
||||
for (int j = 0; valid && j < h.buckets_size(); j++) { |
||||
core->histograms[grpc_stats_histo_start[i] + j] = |
||||
h.buckets(j).count(); |
||||
} |
||||
} |
||||
} |
||||
break; |
||||
} |
||||
} |
||||
} |
||||
|
||||
} // namespace grpc
|
||||
@ -0,0 +1,35 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2016 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_INTERNAL_CPP_UTIL_CORE_STATS_H |
||||
#define GRPC_INTERNAL_CPP_UTIL_CORE_STATS_H |
||||
|
||||
#include "src/proto/grpc/core/stats.pb.h" |
||||
|
||||
extern "C" { |
||||
#include "src/core/lib/debug/stats.h" |
||||
} |
||||
|
||||
namespace grpc { |
||||
|
||||
void CoreStatsToProto(const grpc_stats_data& core, grpc::core::Stats* proto); |
||||
void ProtoToCoreStats(const grpc::core::Stats& proto, grpc_stats_data* core); |
||||
|
||||
} // namespace grpc
|
||||
|
||||
#endif // GRPC_INTERNAL_CPP_UTIL_CORE_STATS_H
|
||||
@ -0,0 +1,38 @@ |
||||
// 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. |
||||
|
||||
syntax = "proto3"; |
||||
|
||||
package grpc.core; |
||||
|
||||
message Bucket { |
||||
double start = 1; |
||||
uint64 count = 2; |
||||
} |
||||
|
||||
message Histogram { |
||||
repeated Bucket buckets = 1; |
||||
} |
||||
|
||||
message Metric { |
||||
string name = 1; |
||||
oneof value { |
||||
uint64 count = 10; |
||||
Histogram histogram = 11; |
||||
} |
||||
} |
||||
|
||||
message Stats { |
||||
repeated Metric metrics = 1; |
||||
} |
||||
Loading…
Reference in new issue