Craig Tiller
8 years ago
commit
357f3f6162
21 changed files with 692 additions and 49 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