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