pull/20753/head
Richard Belleville 5 years ago
parent 36e8f9ff62
commit 3411202598
  1. 2
      src/python/grpcio/grpc/__init__.py
  2. 15
      src/python/grpcio/grpc/_channel.py
  3. 7
      src/python/grpcio_tests/tests/stress/unary_stream_benchmark.py

@ -1960,7 +1960,7 @@ class ChannelOptions(object):
Attributes:
SingleThreadedUnaryStream: Perform unary-stream RPCs on a single thread.
"""
"""
SingleThreadedUnaryStream = "SingleThreadedUnaryStream"

@ -752,14 +752,13 @@ class _SingleThreadedUnaryStreamMultiCallable(grpc.UnaryStreamMultiCallable):
wait_for_ready)
augmented_metadata = _compression.augment_metadata(
metadata, compression)
operations_and_tags = (
((cygrpc.SendInitialMetadataOperation(augmented_metadata,
initial_metadata_flags),
cygrpc.SendMessageOperation(serialized_request, _EMPTY_FLAGS),
cygrpc.SendCloseFromClientOperation(_EMPTY_FLAGS),
cygrpc.ReceiveStatusOnClientOperation(_EMPTY_FLAGS)), None),
((cygrpc.ReceiveInitialMetadataOperation(_EMPTY_FLAGS),), None),
)
operations_and_tags = ((
(cygrpc.SendInitialMetadataOperation(augmented_metadata,
initial_metadata_flags),
cygrpc.SendMessageOperation(serialized_request, _EMPTY_FLAGS),
cygrpc.SendCloseFromClientOperation(_EMPTY_FLAGS),
cygrpc.ReceiveStatusOnClientOperation(_EMPTY_FLAGS)), None),) + (((
cygrpc.ReceiveInitialMetadataOperation(_EMPTY_FLAGS),), None),)
call = self._channel.segregated_call(
cygrpc.PropagationConstants.GRPC_PROPAGATE_DEFAULTS, self._method,
None, _determine_deadline(deadline), metadata, call_credentials,

@ -43,7 +43,6 @@ try:
(grpc.ChannelOptions.SingleThreadedUnaryStream, 1),
]
@contextlib.contextmanager
def _running_server():
server_process = subprocess.Popen(
@ -60,12 +59,12 @@ try:
sys.stdout.write("stderr: {}".format(server_process.stderr.read()))
sys.stdout.flush()
def profile(message_size, response_count):
request = unary_stream_benchmark_pb2.BenchmarkRequest(
message_size=message_size, response_count=response_count)
with grpc.insecure_channel(
'[::]:{}'.format(_PORT), options=_GRPC_CHANNEL_OPTIONS) as channel:
'[::]:{}'.format(_PORT),
options=_GRPC_CHANNEL_OPTIONS) as channel:
stub = unary_stream_benchmark_pb2_grpc.UnaryStreamBenchmarkServiceStub(
channel)
start = datetime.datetime.now()
@ -75,7 +74,6 @@ try:
end = datetime.datetime.now()
return end - start
def main():
with _running_server():
for i in range(1000):
@ -83,7 +81,6 @@ try:
sys.stdout.write("{}\n".format(latency.total_seconds()))
sys.stdout.flush()
if __name__ == '__main__':
main()

Loading…
Cancel
Save