/* * * Copyright 2015-2016, Google Inc. * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are * met: * * * Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above * copyright notice, this list of conditions and the following disclaimer * in the documentation and/or other materials provided with the * distribution. * * Neither the name of Google Inc. nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. *is % allowed in string */ #include "test/cpp/util/metrics_server.h" #include #include #include "src/proto/grpc/testing/metrics.grpc.pb.h" #include "src/proto/grpc/testing/metrics.pb.h" namespace grpc { namespace testing { Gauge::Gauge(long initial_val) : val_(initial_val) {} void Gauge::Set(long new_val) { std::lock_guard lock(val_mu_); val_ = new_val; } long Gauge::Get() { std::lock_guard lock(val_mu_); return val_; } grpc::Status MetricsServiceImpl::GetAllGauges( ServerContext* context, const EmptyMessage* request, ServerWriter* writer) { gpr_log(GPR_INFO, "GetAllGauges called"); std::lock_guard lock(mu_); for (auto it = gauges_.begin(); it != gauges_.end(); it++) { GaugeResponse resp; resp.set_name(it->first); // Gauge name resp.set_long_value(it->second->Get()); // Gauge value writer->Write(resp); } return Status::OK; } grpc::Status MetricsServiceImpl::GetGauge(ServerContext* context, const GaugeRequest* request, GaugeResponse* response) { std::lock_guard lock(mu_); const auto it = gauges_.find(request->name()); if (it != gauges_.end()) { response->set_name(it->first); response->set_long_value(it->second->Get()); } return Status::OK; } std::shared_ptr MetricsServiceImpl::CreateGauge(const grpc::string& name, bool* already_present) { std::lock_guard lock(mu_); std::shared_ptr gauge(new Gauge(0)); const auto p = gauges_.emplace(name, gauge); // p.first is an iterator pointing to > pair. p.second // is a boolean which is set to 'true' if the Gauge is inserted in the guages_ // map and 'false' if it is already present in the map *already_present = !p.second; return p.first->second; } // Starts the metrics server and returns the grpc::Server instance. Call Wait() // on the returned server instance. std::unique_ptr MetricsServiceImpl::StartServer(int port) { gpr_log(GPR_INFO, "Building metrics server.."); const grpc::string address = "0.0.0.0:" + std::to_string(port); ServerBuilder builder; builder.AddListeningPort(address, grpc::InsecureServerCredentials()); builder.RegisterService(this); std::unique_ptr server(builder.BuildAndStart()); gpr_log(GPR_INFO, "Metrics server %s started. Ready to receive requests..", address.c_str()); return server; } } // namespace testing } // namespace grpc