PR comments

pull/8880/head
David Garcia Quintas 8 years ago
parent a0a3c411d1
commit 523a4aa0a0
  1. 31
      test/cpp/qps/json_run_localhost.cc

@ -46,10 +46,11 @@
#include "test/cpp/util/subprocess.h" #include "test/cpp/util/subprocess.h"
using grpc::SubProcess; using grpc::SubProcess;
typedef std::unique_ptr<SubProcess> SubProcessPtr;
SubProcessPtr g_driver;
constexpr auto kNumWorkers = 2; constexpr auto kNumWorkers = 2;
std::vector<SubProcessPtr> g_workers(2);
static SubProcess* g_driver;
static SubProcess* g_workers[kNumWorkers];
template <class T> template <class T>
std::string as_string(const T& val) { std::string as_string(const T& val) {
@ -61,8 +62,9 @@ std::string as_string(const T& val) {
static void sighandler(int sig) { static void sighandler(int sig) {
const int errno_saved = errno; const int errno_saved = errno;
g_driver->Interrupt(); g_driver->Interrupt();
for (const auto& worker : g_workers) for (int i = 0; i < kNumWorkers; ++i) {
if (worker) worker->Interrupt(); if (g_workers[i]) g_workers[i]->Interrupt();
}
errno = errno_saved; errno = errno_saved;
} }
@ -100,7 +102,7 @@ int main(int argc, char** argv) {
const auto port = grpc_pick_unused_port_or_die(); const auto port = grpc_pick_unused_port_or_die();
std::vector<std::string> args = {bin_dir + "/qps_worker", "-driver_port", std::vector<std::string> args = {bin_dir + "/qps_worker", "-driver_port",
as_string(port)}; as_string(port)};
g_workers[i].reset(new SubProcess(args)); g_workers[i] = new SubProcess(args);
if (!first) env << ","; if (!first) env << ",";
env << "localhost:" << port; env << "localhost:" << port;
first = false; first = false;
@ -112,20 +114,25 @@ int main(int argc, char** argv) {
args.push_back(argv[i]); args.push_back(argv[i]);
} }
g_driver.reset(new SubProcess(args)); g_driver = new SubProcess(args);
const int driver_join_status = g_driver->Join(); const int driver_join_status = g_driver->Join();
if (driver_join_status != 0) { if (driver_join_status != 0) {
LogStatus(driver_join_status, "driver"); LogStatus(driver_join_status, "driver");
} }
for (const auto& worker : g_workers) { for (int i = 0; i < kNumWorkers; ++i) {
if (worker) worker->Interrupt(); if (g_workers[i]) g_workers[i]->Interrupt();
} }
for (const auto& worker : g_workers) {
if (worker) { for (int i = 0; i < kNumWorkers; ++i) {
const int worker_status = worker->Join(); if (g_workers[i]) {
const int worker_status = g_workers[i]->Join();
if (worker_status != 0) { if (worker_status != 0) {
LogStatus(worker_status, "worker"); LogStatus(worker_status, "worker");
} }
} }
} }
delete g_driver;
for (int i = 0; i < kNumWorkers; ++i) delete g_workers[i];
GPR_ASSERT(driver_join_status == 0);
} }

Loading…
Cancel
Save