blob: 6de0a084698759f492ed73a3d296a6620b566db6 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
//
//
// Copyright 2023 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_SRC_CPP_SERVER_BACKEND_METRIC_RECORDER_H
#define GRPC_SRC_CPP_SERVER_BACKEND_METRIC_RECORDER_H
#include <stdint.h>
#include <atomic>
#include <map>
#include "y_absl/base/thread_annotations.h"
#include "y_absl/strings/string_view.h"
#include <grpcpp/ext/call_metric_recorder.h>
#include <grpcpp/ext/server_metric_recorder.h>
#include <grpcpp/impl/sync.h>
#include <grpcpp/support/string_ref.h>
#include "src/core/ext/filters/backend_metrics/backend_metric_provider.h"
#include "src/core/ext/filters/client_channel/lb_policy/backend_metric_data.h"
namespace grpc {
namespace experimental {
// Backend metrics and an associated update sequence number.
struct ServerMetricRecorder::BackendMetricDataState {
grpc_core::BackendMetricData data;
uint64_t sequence_number = 0;
};
} // namespace experimental
class BackendMetricState : public grpc_core::BackendMetricProvider,
public experimental::CallMetricRecorder {
public:
// `server_metric_recorder` is optional. When set, GetBackendMetricData()
// merges metrics from `server_metric_recorder` with metrics recorded to this.
explicit BackendMetricState(
experimental::ServerMetricRecorder* server_metric_recorder)
: server_metric_recorder_(server_metric_recorder) {}
experimental::CallMetricRecorder& RecordCpuUtilizationMetric(
double value) override;
experimental::CallMetricRecorder& RecordMemoryUtilizationMetric(
double value) override;
experimental::CallMetricRecorder& RecordQpsMetric(double value) override;
experimental::CallMetricRecorder& RecordEpsMetric(double value) override;
experimental::CallMetricRecorder& RecordUtilizationMetric(
string_ref name, double value) override;
experimental::CallMetricRecorder& RecordRequestCostMetric(
string_ref name, double value) override;
experimental::CallMetricRecorder& RecordNamedMetric(string_ref name,
double value) override;
// This clears metrics currently recorded. Don't call twice.
grpc_core::BackendMetricData GetBackendMetricData() override;
private:
experimental::ServerMetricRecorder* server_metric_recorder_;
std::atomic<double> cpu_utilization_{-1.0};
std::atomic<double> mem_utilization_{-1.0};
std::atomic<double> qps_{-1.0};
std::atomic<double> eps_{-1.0};
internal::Mutex mu_;
std::map<y_absl::string_view, double> utilization_ Y_ABSL_GUARDED_BY(mu_);
std::map<y_absl::string_view, double> request_cost_ Y_ABSL_GUARDED_BY(mu_);
std::map<y_absl::string_view, double> named_metrics_ Y_ABSL_GUARDED_BY(mu_);
};
} // namespace grpc
#endif // GRPC_SRC_CPP_SERVER_BACKEND_METRIC_RECORDER_H
|
