|
|
|
@ -35,7 +35,7 @@ import time |
|
|
|
|
from concurrent import futures |
|
|
|
|
import grpc |
|
|
|
|
from src.proto.grpc.testing import control_pb2 |
|
|
|
|
from src.proto.grpc.testing import services_pb2 |
|
|
|
|
from src.proto.grpc.testing import services_pb2_grpc |
|
|
|
|
from src.proto.grpc.testing import stats_pb2 |
|
|
|
|
|
|
|
|
|
from tests.qps import benchmark_client |
|
|
|
@ -45,7 +45,7 @@ from tests.qps import histogram |
|
|
|
|
from tests.unit import resources |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class WorkerServer(services_pb2.WorkerServiceServicer): |
|
|
|
|
class WorkerServer(services_pb2_grpc.WorkerServiceServicer): |
|
|
|
|
"""Python Worker Server implementation.""" |
|
|
|
|
|
|
|
|
|
def __init__(self): |
|
|
|
@ -87,8 +87,8 @@ class WorkerServer(services_pb2.WorkerServiceServicer): |
|
|
|
|
futures.ThreadPoolExecutor(max_workers=server_threads)) |
|
|
|
|
if config.server_type == control_pb2.ASYNC_SERVER: |
|
|
|
|
servicer = benchmark_server.BenchmarkServer() |
|
|
|
|
services_pb2.add_BenchmarkServiceServicer_to_server(servicer, |
|
|
|
|
server) |
|
|
|
|
services_pb2_grpc.add_BenchmarkServiceServicer_to_server(servicer, |
|
|
|
|
server) |
|
|
|
|
elif config.server_type == control_pb2.ASYNC_GENERIC_SERVER: |
|
|
|
|
resp_size = config.payload_config.bytebuf_params.resp_size |
|
|
|
|
servicer = benchmark_server.GenericBenchmarkServer(resp_size) |
|
|
|
|