Update benchmark code to handle both implementations

This commit is contained in:
murgatroid99 2018-04-19 16:10:48 -07:00
parent 8176c70a94
commit cad47e4de8
6 changed files with 39 additions and 21 deletions

View File

@ -14,7 +14,9 @@
}
],
"dependencies": {
"express": "^4.16.3",
"google-auth-library": "^0.9.2",
"lodash": "^4.17.4"
"lodash": "^4.17.4",
"poisson-process": "^1.0.0"
}
}

View File

@ -36,10 +36,16 @@ var Histogram = require('./histogram');
var genericService = require('./generic_service');
// TODO(murgatroid99): use multiple grpc implementations
var grpc = require('grpc');
var serviceProto = grpc.load({
root: __dirname + '/../packages/grpc-native-core/ext/grpc',
file: 'src/proto/grpc/testing/services.proto'}).grpc.testing;
var grpc = require('../any_grpc').client;
var protoLoader = require('../../packages/grpc-protobufjs');
var protoPackage = protoLoader.loadSync(
'src/proto/grpc/testing/services.proto',
{keepCase: true,
defaults: true,
enums: String,
oneofs: true,
include: [__dirname + '/../../packages/grpc-native-core/deps/grpc']});
var serviceProto = grpc.loadPackageDefinition(protoPackage).grpc.testing;
/**
* Create a buffer filled with size zeroes
@ -82,7 +88,7 @@ function BenchmarkClient(server_targets, channels, histogram_params,
if (security_params) {
var ca_path;
if (security_params.use_test_ca) {
ca_path = path.join(__dirname, '../test/data/ca.pem');
ca_path = path.join(__dirname, '../data/ca.pem');
var ca_data = fs.readFileSync(ca_path);
creds = grpc.credentials.createSsl(ca_data);
} else {

View File

@ -60,7 +60,7 @@ function BenchmarkClient(server_targets, channels, histogram_params,
protocol = https;
this.request = _.bind(https.request, https);
if (security_params.use_test_ca) {
ca_path = path.join(__dirname, '../test/data/ca.pem');
ca_path = path.join(__dirname, '../data/ca.pem');
var ca_data = fs.readFileSync(ca_path);
options.ca = ca_data;
}

View File

@ -30,11 +30,16 @@ var util = require('util');
var genericService = require('./generic_service');
// TODO(murgatroid99): use multiple grpc implementations
var grpc = require('grpc');
var serviceProto = grpc.load({
root: __dirname + '/../packages/grpc-native-core/ext/grpc',
file: 'src/proto/grpc/testing/services.proto'}).grpc.testing;
var grpc = require('../any_grpc').server;
var protoLoader = require('../../packages/grpc-protobufjs');
var protoPackage = protoLoader.loadSync(
'src/proto/grpc/testing/services.proto',
{keepCase: true,
defaults: true,
enums: String,
oneofs: true,
include: [__dirname + '/../../packages/grpc-native-core/deps/grpc']});
var serviceProto = grpc.loadPackageDefinition(protoPackage).grpc.testing;
/**
* Create a buffer filled with size zeroes
@ -106,8 +111,8 @@ function BenchmarkServer(host, port, tls, generic, response_size) {
var server_creds;
var host_override;
if (tls) {
var key_path = path.join(__dirname, '../test/data/server1.key');
var pem_path = path.join(__dirname, '../test/data/server1.pem');
var key_path = path.join(__dirname, '../data/server1.key');
var pem_path = path.join(__dirname, '../data/server1.pem');
var key_data = fs.readFileSync(key_path);
var pem_data = fs.readFileSync(pem_path);

View File

@ -47,8 +47,8 @@ function BenchmarkServer(host, port, tls, generic, response_size) {
this.input_port = port;
if (tls) {
var credentials = {};
var key_path = path.join(__dirname, '../test/data/server1.key');
var pem_path = path.join(__dirname, '../test/data/server1.pem');
var key_path = path.join(__dirname, '../data/server1.key');
var pem_path = path.join(__dirname, '../data/server1.pem');
var key_data = fs.readFileSync(key_path);
var pem_data = fs.readFileSync(pem_path);

View File

@ -21,11 +21,16 @@
var console = require('console');
var WorkerServiceImpl = require('./worker_service_impl');
// TODO(murgatroid99): use multiple grpc implementations
var grpc = require('grpc');
var serviceProto = grpc.load({
root: __dirname + '/../packages/grpc-native-core/ext/grpc',
file: 'src/proto/grpc/testing/services.proto'}).grpc.testing;
var grpc = require('../any_grpc').server;
var protoLoader = require('../../packages/grpc-protobufjs');
var protoPackage = protoLoader.loadSync(
'src/proto/grpc/testing/services.proto',
{keepCase: true,
defaults: true,
enums: String,
oneofs: true,
include: [__dirname + '/../../packages/grpc-native-core/deps/grpc']});
var serviceProto = grpc.loadPackageDefinition(protoPackage).grpc.testing;
function runServer(port, benchmark_impl) {
var server_creds = grpc.ServerCredentials.createInsecure();