From bbcaca2d83c06aa548fbe587f8ff16fe298f78d8 Mon Sep 17 00:00:00 2001 From: Lidi Zheng Date: Wed, 19 Feb 2020 13:09:46 -0800 Subject: [PATCH] Add uvloop argument --- src/python/grpcio_tests/tests_aio/benchmark/worker.py | 7 +++++++ .../grpcio_tests/tests_aio/benchmark/worker_servicer.py | 2 +- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/src/python/grpcio_tests/tests_aio/benchmark/worker.py b/src/python/grpcio_tests/tests_aio/benchmark/worker.py index ad6d3e7b1c2..94651b55057 100644 --- a/src/python/grpcio_tests/tests_aio/benchmark/worker.py +++ b/src/python/grpcio_tests/tests_aio/benchmark/worker.py @@ -46,6 +46,13 @@ if __name__ == '__main__': type=int, dest='port', help='The port the worker should listen on') + parser.add_argument('--uvloop', + action='store_true', + help='Use uvloop or not') args = parser.parse_args() + if args.uvloop: + import uvloop + uvloop.install() + asyncio.get_event_loop().run_until_complete(run_worker_server(args.port)) diff --git a/src/python/grpcio_tests/tests_aio/benchmark/worker_servicer.py b/src/python/grpcio_tests/tests_aio/benchmark/worker_servicer.py index 720af70ea3d..4f80095cd20 100644 --- a/src/python/grpcio_tests/tests_aio/benchmark/worker_servicer.py +++ b/src/python/grpcio_tests/tests_aio/benchmark/worker_servicer.py @@ -33,7 +33,7 @@ from tests_aio.benchmark import benchmark_client, benchmark_servicer _NUM_CORES = multiprocessing.cpu_count() _WORKER_ENTRY_FILE = os.path.join( - os.path.split(os.path.abspath(__file__))[0], '/worker.py') + os.path.split(os.path.abspath(__file__))[0], 'worker.py') _LOGGER = logging.getLogger(__name__)