Skip to content

Commit

Permalink
PR comments
Browse files Browse the repository at this point in the history
  • Loading branch information
aviklai committed May 28, 2024
1 parent 952057e commit d0a646f
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -118,12 +118,12 @@ async def new_response_stream_behavior(request_or_iterator, servicer_context):
grpc_service_name,
grpc_method_name)

async for obj in behavior(request_or_iterator, servicer_context):
async for behavior_response in behavior(request_or_iterator, servicer_context):
self._metrics["grpc_server_stream_msg_sent"].labels(
grpc_type=grpc_type,
grpc_service=grpc_service_name,
grpc_method=grpc_method_name).inc()
yield obj
yield behavior_response

self.increase_grpc_server_handled_total_counter(grpc_type,
grpc_service_name,
Expand All @@ -147,8 +147,8 @@ async def new_response_stream_behavior(request_or_iterator, servicer_context):
if self._skip_exceptions:
if self._log_exceptions:
_LOGGER.error(e)
async for obj in behavior(request_or_iterator, servicer_context):
yield obj
async for behavior_response in behavior(request_or_iterator, servicer_context):
yield behavior_response
raise e

if response_streaming:
Expand Down

0 comments on commit d0a646f

Please sign in to comment.