mirror of https://github.com/grpc/grpc.git
Craig Tiller
10 years ago
commit
c5b5290bc6
57 changed files with 872 additions and 246 deletions
@ -0,0 +1,128 @@ |
||||
# Copyright 2015, 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. |
||||
|
||||
""" |
||||
Bazel macros to declare gRPC libraries automatically generated from proto files. |
||||
|
||||
This file declares two macros: |
||||
- objc_proto_library |
||||
- objc_grpc_library |
||||
""" |
||||
|
||||
def _lower_underscore_to_upper_camel(str): |
||||
humps = [] |
||||
for hump in str.split('_'): |
||||
humps += [hump[0].upper() + hump[1:]] |
||||
return "".join(humps) |
||||
|
||||
def _file_to_upper_camel(src): |
||||
elements = src.rpartition('/') |
||||
upper_camel = _lower_underscore_to_upper_camel(elements[-1]) |
||||
return "".join(elements[:-1] + [upper_camel]) |
||||
|
||||
def _file_with_extension(src, ext): |
||||
elements = src.rpartition('/') |
||||
basename = elements[-1].partition('.')[0] |
||||
return "".join(elements[:-1] + [basename, ext]) |
||||
|
||||
def _protoc_invocation(srcs, flags): |
||||
"""Returns a command line to invoke protoc from a genrule, on the given |
||||
sources, using the given flags. |
||||
""" |
||||
protoc_command = "$(location //external:protoc) -I . " |
||||
srcs_params = "" |
||||
for src in srcs: |
||||
srcs_params += " $(location %s)" % (src) |
||||
return protoc_command + flags + srcs_params |
||||
|
||||
def objc_proto_library(name, srcs, visibility=None): |
||||
"""Declares an objc_library for the code generated by protoc from the given |
||||
proto sources. This generated code doesn't include proto services. |
||||
""" |
||||
h_files = [] |
||||
m_files = [] |
||||
for src in srcs: |
||||
src = _file_to_upper_camel(src) |
||||
h_files += [_file_with_extension(src, ".pbobjc.h")] |
||||
m_files += [_file_with_extension(src, ".pbobjc.m")] |
||||
|
||||
protoc_flags = "--objc_out=$(GENDIR)" |
||||
|
||||
native.genrule( |
||||
name = name + "_codegen", |
||||
srcs = srcs + ["//external:protoc"], |
||||
outs = h_files + m_files, |
||||
cmd = _protoc_invocation(srcs, protoc_flags), |
||||
) |
||||
native.objc_library( |
||||
name = name, |
||||
hdrs = h_files, |
||||
includes = ["."], |
||||
non_arc_srcs = m_files, |
||||
deps = ["//external:protobuf_objc"], |
||||
visibility = visibility, |
||||
) |
||||
|
||||
def objc_grpc_library(name, services, other_messages, visibility=None): |
||||
"""Declares an objc_library for the code generated by gRPC and protoc from the |
||||
given proto sources (services and other_messages). The generated code doesn't |
||||
include proto services of the files passed as other_messages. |
||||
""" |
||||
objc_proto_library(name + "_messages", services + other_messages) |
||||
|
||||
h_files = [] |
||||
m_files = [] |
||||
for src in services: |
||||
src = _file_to_upper_camel(src) |
||||
h_files += [_file_with_extension(src, ".pbrpc.h")] |
||||
m_files += [_file_with_extension(src, ".pbrpc.m")] |
||||
|
||||
protoc_flags = ("--grpc_out=$(GENDIR) --plugin=" + |
||||
"protoc-gen-grpc=$(location //external:grpc_protoc_plugin_objc)") |
||||
|
||||
native.genrule( |
||||
name = name + "_codegen", |
||||
srcs = services + [ |
||||
"//external:grpc_protoc_plugin_objc", |
||||
"//external:protoc", |
||||
], |
||||
outs = h_files + m_files, |
||||
cmd = _protoc_invocation(services, protoc_flags), |
||||
) |
||||
native.objc_library( |
||||
name = name, |
||||
hdrs = h_files, |
||||
includes = ["."], |
||||
srcs = m_files, |
||||
deps = [ |
||||
":" + name + "_messages", |
||||
"//external:proto_objc_rpc", |
||||
], |
||||
visibility = visibility, |
||||
) |
||||
@ -1,40 +0,0 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, 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. |
||||
* |
||||
*/ |
||||
|
||||
#import <Foundation/Foundation.h> |
||||
|
||||
#import "GRPCClient/GRPCMethodName.h" |
||||
|
||||
@interface GRPCMethodName (HTTP2Encoding) |
||||
- (NSString *)HTTP2Path; |
||||
@end |
||||
@ -1,44 +0,0 @@ |
||||
/* |
||||
* |
||||
* Copyright 2015, 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. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCMethodName+HTTP2Encoding.h" |
||||
|
||||
@implementation GRPCMethodName (HTTP2Encoding) |
||||
- (NSString *)HTTP2Path { |
||||
if (self.package) { |
||||
return [NSString stringWithFormat:@"/%@.%@/%@", self.package, self.interface, self.method]; |
||||
} else { |
||||
return [NSString stringWithFormat:@"/%@/%@", self.interface, self.method]; |
||||
} |
||||
} |
||||
@end |
||||
@ -0,0 +1,71 @@ |
||||
// Copyright 2015, 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. |
||||
|
||||
syntax = "proto3"; |
||||
|
||||
import "test/cpp/qps/qpstest.proto"; |
||||
|
||||
package grpc.testing; |
||||
|
||||
service PerfDbTransfer { |
||||
// Sends client info |
||||
rpc RecordSingleClientData(SingleUserRecordRequest) |
||||
returns (SingleUserRecordReply) { |
||||
} |
||||
} |
||||
|
||||
// Metrics to be stored |
||||
message Metrics { |
||||
double qps = 1; |
||||
double qps_per_core = 2; |
||||
double perc_lat_50 = 3; |
||||
double perc_lat_90 = 4; |
||||
double perc_lat_95 = 5; |
||||
double perc_lat_99 = 6; |
||||
double perc_lat_99_point_9 = 7; |
||||
double server_system_time = 8; |
||||
double server_user_time = 9; |
||||
double client_system_time = 10; |
||||
double client_user_time = 11; |
||||
} |
||||
|
||||
// Request for storing a single user's data |
||||
message SingleUserRecordRequest { |
||||
string hashed_id = 1; |
||||
string test_name = 2; |
||||
string sys_info = 3; |
||||
string tag = 4; |
||||
Metrics metrics = 5; |
||||
ClientConfig client_config = 6; |
||||
ServerConfig server_config = 7; |
||||
} |
||||
|
||||
// Reply to request for storing single user's data |
||||
message SingleUserRecordReply { |
||||
} |
||||
@ -0,0 +1,143 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, 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. |
||||
* |
||||
*/ |
||||
|
||||
#include "test/cpp/qps/perf_db_client.h" |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
// sets the client and server config information
|
||||
void PerfDbClient::setConfigs(const ClientConfig& client_config, |
||||
const ServerConfig& server_config) { |
||||
client_config_ = client_config; |
||||
server_config_ = server_config; |
||||
} |
||||
|
||||
// sets the QPS
|
||||
void PerfDbClient::setQps(double qps) { |
||||
qps_ = qps; |
||||
} |
||||
|
||||
// sets the QPS per core
|
||||
void PerfDbClient::setQpsPerCore(double qps_per_core) { |
||||
qps_per_core_ = qps_per_core; |
||||
} |
||||
|
||||
// sets the 50th, 90th, 95th, 99th and 99.9th percentile latency
|
||||
void PerfDbClient::setLatencies(double perc_lat_50, |
||||
double perc_lat_90, |
||||
double perc_lat_95, |
||||
double perc_lat_99, |
||||
double perc_lat_99_point_9) { |
||||
perc_lat_50_ = perc_lat_50; |
||||
perc_lat_90_ = perc_lat_90; |
||||
perc_lat_95_ = perc_lat_95; |
||||
perc_lat_99_ = perc_lat_99; |
||||
perc_lat_99_point_9_ = perc_lat_99_point_9; |
||||
} |
||||
|
||||
// sets the server and client, user and system times
|
||||
void PerfDbClient::setTimes(double server_system_time, double server_user_time, |
||||
double client_system_time, double client_user_time) { |
||||
server_system_time_ = server_system_time; |
||||
server_user_time_ = server_user_time; |
||||
client_system_time_ = client_system_time; |
||||
client_user_time_ = client_user_time; |
||||
} |
||||
|
||||
// sends the data to the performance database server
|
||||
bool PerfDbClient::sendData(std::string hashed_id, std::string test_name, |
||||
std::string sys_info, std::string tag) { |
||||
// Data record request object
|
||||
SingleUserRecordRequest single_user_record_request; |
||||
|
||||
// setting access token, name of the test and the system information
|
||||
single_user_record_request.set_hashed_id(hashed_id); |
||||
single_user_record_request.set_test_name(test_name); |
||||
single_user_record_request.set_sys_info(sys_info); |
||||
single_user_record_request.set_tag(tag); |
||||
|
||||
// setting configs
|
||||
*(single_user_record_request.mutable_client_config()) = client_config_; |
||||
*(single_user_record_request.mutable_server_config()) = server_config_; |
||||
|
||||
Metrics* metrics = single_user_record_request.mutable_metrics(); |
||||
|
||||
// setting metrcs in data record request
|
||||
if (qps_ != DBL_MIN) { |
||||
metrics->set_qps(qps_); |
||||
} |
||||
if (qps_per_core_ != DBL_MIN) { |
||||
metrics->set_qps_per_core(qps_per_core_); |
||||
} |
||||
if (perc_lat_50_ != DBL_MIN) { |
||||
metrics->set_perc_lat_50(perc_lat_50_); |
||||
} |
||||
if (perc_lat_90_ != DBL_MIN) { |
||||
metrics->set_perc_lat_90(perc_lat_90_); |
||||
} |
||||
if (perc_lat_95_ != DBL_MIN) { |
||||
metrics->set_perc_lat_95(perc_lat_95_); |
||||
} |
||||
if (perc_lat_99_ != DBL_MIN) { |
||||
metrics->set_perc_lat_99(perc_lat_99_); |
||||
} |
||||
if (perc_lat_99_point_9_ != DBL_MIN) { |
||||
metrics->set_perc_lat_99_point_9(perc_lat_99_point_9_); |
||||
} |
||||
if (server_system_time_ != DBL_MIN) { |
||||
metrics->set_server_system_time(server_system_time_); |
||||
} |
||||
if (server_user_time_ != DBL_MIN) { |
||||
metrics->set_server_user_time(server_user_time_); |
||||
} |
||||
if (client_system_time_ != DBL_MIN) { |
||||
metrics->set_client_system_time(client_system_time_); |
||||
} |
||||
if (client_user_time_ != DBL_MIN) { |
||||
metrics->set_client_user_time(client_user_time_); |
||||
} |
||||
|
||||
SingleUserRecordReply single_user_record_reply; |
||||
ClientContext context; |
||||
|
||||
Status status = stub_->RecordSingleClientData( |
||||
&context, single_user_record_request, &single_user_record_reply); |
||||
if (status.ok()) { |
||||
return true; // data sent to database successfully
|
||||
} else { |
||||
return false; // error in data sending
|
||||
} |
||||
} |
||||
} // testing
|
||||
} // grpc
|
||||
@ -0,0 +1,115 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, 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. |
||||
* |
||||
*/ |
||||
|
||||
#include <iostream> |
||||
#include <memory> |
||||
#include <string> |
||||
#include <cfloat> |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc++/channel_arguments.h> |
||||
#include <grpc++/channel_interface.h> |
||||
#include <grpc++/client_context.h> |
||||
#include <grpc++/create_channel.h> |
||||
#include <grpc++/credentials.h> |
||||
#include <grpc++/status.h> |
||||
#include "test/cpp/qps/perf_db.grpc.pb.h" |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
// Manages data sending to performance database server
|
||||
class PerfDbClient { |
||||
public: |
||||
PerfDbClient() { |
||||
qps_ = DBL_MIN; |
||||
qps_per_core_ = DBL_MIN; |
||||
perc_lat_50_ = DBL_MIN; |
||||
perc_lat_90_ = DBL_MIN; |
||||
perc_lat_95_ = DBL_MIN; |
||||
perc_lat_99_ = DBL_MIN; |
||||
perc_lat_99_point_9_ = DBL_MIN; |
||||
server_system_time_ = DBL_MIN; |
||||
server_user_time_ = DBL_MIN; |
||||
client_system_time_ = DBL_MIN; |
||||
client_user_time_ = DBL_MIN; |
||||
} |
||||
|
||||
void init(std::shared_ptr<ChannelInterface> channel) { |
||||
stub_ = PerfDbTransfer::NewStub(channel); |
||||
} |
||||
|
||||
~PerfDbClient() {} |
||||
|
||||
// sets the client and server config information
|
||||
void setConfigs(const ClientConfig& client_config, |
||||
const ServerConfig& server_config); |
||||
|
||||
// sets the qps
|
||||
void setQps(double qps); |
||||
|
||||
// sets the qps per core
|
||||
void setQpsPerCore(double qps_per_core); |
||||
|
||||
// sets the 50th, 90th, 95th, 99th and 99.9th percentile latency
|
||||
void setLatencies(double perc_lat_50, double perc_lat_90, |
||||
double perc_lat_95, double perc_lat_99, |
||||
double perc_lat_99_point_9); |
||||
|
||||
// sets the server and client, user and system times
|
||||
void setTimes(double server_system_time, double server_user_time, |
||||
double client_system_time, double client_user_time); |
||||
|
||||
// sends the data to the performance database server
|
||||
bool sendData(std::string hashed_id, std::string test_name, |
||||
std::string sys_info, std::string tag); |
||||
|
||||
private: |
||||
std::unique_ptr<PerfDbTransfer::Stub> stub_; |
||||
ClientConfig client_config_; |
||||
ServerConfig server_config_; |
||||
double qps_; |
||||
double qps_per_core_; |
||||
double perc_lat_50_; |
||||
double perc_lat_90_; |
||||
double perc_lat_95_; |
||||
double perc_lat_99_; |
||||
double perc_lat_99_point_9_; |
||||
double server_system_time_; |
||||
double server_user_time_; |
||||
double client_system_time_; |
||||
double client_user_time_; |
||||
}; |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
Loading…
Reference in new issue