1# Copyright 2020 The gRPC Authors 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14"""The Python AsyncIO Benchmark Servicers.""" 15 16import asyncio 17import logging 18import unittest 19 20from grpc.experimental import aio 21 22from src.proto.grpc.testing import benchmark_service_pb2_grpc, messages_pb2 23 24 25class BenchmarkServicer(benchmark_service_pb2_grpc.BenchmarkServiceServicer): 26 27 async def UnaryCall(self, request, unused_context): 28 payload = messages_pb2.Payload(body=b'\0' * request.response_size) 29 return messages_pb2.SimpleResponse(payload=payload) 30 31 async def StreamingFromServer(self, request, unused_context): 32 payload = messages_pb2.Payload(body=b'\0' * request.response_size) 33 # Sends response at full capacity! 34 while True: 35 yield messages_pb2.SimpleResponse(payload=payload) 36 37 async def StreamingCall(self, request_iterator, unused_context): 38 async for request in request_iterator: 39 payload = messages_pb2.Payload(body=b'\0' * request.response_size) 40 yield messages_pb2.SimpleResponse(payload=payload) 41 42 43class GenericBenchmarkServicer( 44 benchmark_service_pb2_grpc.BenchmarkServiceServicer): 45 """Generic (no-codec) Server implementation for the Benchmark service.""" 46 47 def __init__(self, resp_size): 48 self._response = '\0' * resp_size 49 50 async def UnaryCall(self, unused_request, unused_context): 51 return self._response 52 53 async def StreamingCall(self, request_iterator, unused_context): 54 async for _ in request_iterator: 55 yield self._response 56