Clean up self references

pull/21904/head
Lidi Zheng 5 years ago
parent 1ee45615e2
commit afb080787e
  1. 11
      src/python/grpcio_tests/tests_aio/benchmark/worker_servicer.py

@ -13,6 +13,7 @@
# limitations under the License. # limitations under the License.
import asyncio import asyncio
import logging
import multiprocessing import multiprocessing
import time import time
from typing import Tuple from typing import Tuple
@ -117,11 +118,11 @@ class WorkerServicer(worker_service_pb2_grpc.WorkerServiceServicer):
server, port = _create_server(config) server, port = _create_server(config)
await server.start() await server.start()
start_time = time.time() start_time = time.time()
yield self._get_server_status(start_time, start_time, port) yield _get_server_status(start_time, start_time, port)
async for request in request_iterator: async for request in request_iterator:
end_time = time.time() end_time = time.time()
status = self._get_server_status(start_time, end_time, port) status = _get_server_status(start_time, end_time, port)
if request.mark.reset: if request.mark.reset:
start_time = end_time start_time = end_time
yield status yield status
@ -138,16 +139,16 @@ class WorkerServicer(worker_service_pb2_grpc.WorkerServiceServicer):
# Create a client for each channel as asyncio.Task # Create a client for each channel as asyncio.Task
for i in range(config.client_channels): for i in range(config.client_channels):
server = config.server_targets[i % len(config.server_targets)] server = config.server_targets[i % len(config.server_targets)]
client = self._create_client(server, config, qps_data) client = _create_client(server, config, qps_data)
running_tasks.append(self._loop.create_task(client.run())) running_tasks.append(self._loop.create_task(client.run()))
end_time = time.time() end_time = time.time()
yield self._get_client_status(start_time, end_time, qps_data) yield _get_client_status(start_time, end_time, qps_data)
# Respond to stat requests # Respond to stat requests
async for request in request_iterator: async for request in request_iterator:
end_time = time.time() end_time = time.time()
status = self._get_client_status(start_time, end_time, qps_data) status = _get_client_status(start_time, end_time, qps_data)
if request.mark.reset: if request.mark.reset:
qps_data.reset() qps_data.reset()
start_time = time.time() start_time = time.time()

Loading…
Cancel
Save