xref: /aosp_15_r20/external/grpc-grpc/src/ruby/stress/metrics_server.rb (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
1*cc02d7e2SAndroid Build Coastguard Worker# Copyright 2016 gRPC authors.
2*cc02d7e2SAndroid Build Coastguard Worker#
3*cc02d7e2SAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License");
4*cc02d7e2SAndroid Build Coastguard Worker# you may not use this file except in compliance with the License.
5*cc02d7e2SAndroid Build Coastguard Worker# You may obtain a copy of the License at
6*cc02d7e2SAndroid Build Coastguard Worker#
7*cc02d7e2SAndroid Build Coastguard Worker#     http://www.apache.org/licenses/LICENSE-2.0
8*cc02d7e2SAndroid Build Coastguard Worker#
9*cc02d7e2SAndroid Build Coastguard Worker# Unless required by applicable law or agreed to in writing, software
10*cc02d7e2SAndroid Build Coastguard Worker# distributed under the License is distributed on an "AS IS" BASIS,
11*cc02d7e2SAndroid Build Coastguard Worker# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12*cc02d7e2SAndroid Build Coastguard Worker# See the License for the specific language governing permissions and
13*cc02d7e2SAndroid Build Coastguard Worker# limitations under the License.
14*cc02d7e2SAndroid Build Coastguard Worker
15*cc02d7e2SAndroid Build Coastguard Workerrequire_relative '../pb/grpc/testing/metrics_pb.rb'
16*cc02d7e2SAndroid Build Coastguard Workerrequire_relative '../pb/grpc/testing/metrics_services_pb.rb'
17*cc02d7e2SAndroid Build Coastguard Worker
18*cc02d7e2SAndroid Build Coastguard Workerclass Gauge
19*cc02d7e2SAndroid Build Coastguard Worker  def get_name
20*cc02d7e2SAndroid Build Coastguard Worker    raise NoMethodError.new
21*cc02d7e2SAndroid Build Coastguard Worker  end
22*cc02d7e2SAndroid Build Coastguard Worker
23*cc02d7e2SAndroid Build Coastguard Worker  def get_type
24*cc02d7e2SAndroid Build Coastguard Worker    raise NoMethodError.new
25*cc02d7e2SAndroid Build Coastguard Worker  end
26*cc02d7e2SAndroid Build Coastguard Worker
27*cc02d7e2SAndroid Build Coastguard Worker  def get_value
28*cc02d7e2SAndroid Build Coastguard Worker    raise NoMethodError.new
29*cc02d7e2SAndroid Build Coastguard Worker  end
30*cc02d7e2SAndroid Build Coastguard Workerend
31*cc02d7e2SAndroid Build Coastguard Worker
32*cc02d7e2SAndroid Build Coastguard Workerclass MetricsServiceImpl < Grpc::Testing::MetricsService::Service
33*cc02d7e2SAndroid Build Coastguard Worker  include Grpc::Testing
34*cc02d7e2SAndroid Build Coastguard Worker  @gauges
35*cc02d7e2SAndroid Build Coastguard Worker
36*cc02d7e2SAndroid Build Coastguard Worker  def initialize
37*cc02d7e2SAndroid Build Coastguard Worker    @gauges = {}
38*cc02d7e2SAndroid Build Coastguard Worker  end
39*cc02d7e2SAndroid Build Coastguard Worker
40*cc02d7e2SAndroid Build Coastguard Worker  def register_gauge(gauge)
41*cc02d7e2SAndroid Build Coastguard Worker    @gauges[gauge.get_name] = gauge
42*cc02d7e2SAndroid Build Coastguard Worker  end
43*cc02d7e2SAndroid Build Coastguard Worker
44*cc02d7e2SAndroid Build Coastguard Worker  def make_gauge_response(gauge)
45*cc02d7e2SAndroid Build Coastguard Worker    response = GaugeResponse.new(:name => gauge.get_name)
46*cc02d7e2SAndroid Build Coastguard Worker    value = gauge.get_value
47*cc02d7e2SAndroid Build Coastguard Worker    case gauge.get_type
48*cc02d7e2SAndroid Build Coastguard Worker    when 'long'
49*cc02d7e2SAndroid Build Coastguard Worker      response.long_value = value
50*cc02d7e2SAndroid Build Coastguard Worker    when 'double'
51*cc02d7e2SAndroid Build Coastguard Worker      response.double_value = value
52*cc02d7e2SAndroid Build Coastguard Worker    when 'string'
53*cc02d7e2SAndroid Build Coastguard Worker      response.string_value = value
54*cc02d7e2SAndroid Build Coastguard Worker    end
55*cc02d7e2SAndroid Build Coastguard Worker    response
56*cc02d7e2SAndroid Build Coastguard Worker  end
57*cc02d7e2SAndroid Build Coastguard Worker
58*cc02d7e2SAndroid Build Coastguard Worker  def get_all_gauges(_empty, _call)
59*cc02d7e2SAndroid Build Coastguard Worker    @gauges.values.map do |gauge|
60*cc02d7e2SAndroid Build Coastguard Worker      make_gauge_response gauge
61*cc02d7e2SAndroid Build Coastguard Worker    end
62*cc02d7e2SAndroid Build Coastguard Worker  end
63*cc02d7e2SAndroid Build Coastguard Worker
64*cc02d7e2SAndroid Build Coastguard Worker  def get_gauge(gauge_req, _call)
65*cc02d7e2SAndroid Build Coastguard Worker    gauge = @gauges[gauge_req.name]
66*cc02d7e2SAndroid Build Coastguard Worker    make_gauge_response gauge
67*cc02d7e2SAndroid Build Coastguard Worker  end
68*cc02d7e2SAndroid Build Coastguard Workerend
69