Skip to content

Commit aa6a73d

Browse files
committed
fix graphql send cancel on break from subscription loop
1 parent d14710b commit aa6a73d

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

rsocket/graphql/rsocket_transport.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -121,17 +121,17 @@ def request(self, n: int):
121121
self._rsocket_client.request_stream(rsocket_payload).subscribe(subscriber)
122122

123123
while True:
124-
response = await received_queue.get()
124+
try:
125+
response = await received_queue.get()
125126

126-
if isinstance(response, Exception):
127-
raise response
127+
if isinstance(response, Exception):
128+
raise response
128129

129-
if response is complete_object:
130-
break
130+
if response is complete_object:
131+
break
131132

132-
execution_result = self._response_to_execution_result(response)
133+
execution_result = self._response_to_execution_result(response)
133134

134-
try:
135135
yield execution_result
136136
except GeneratorExit:
137137
logger().debug('Generator exited')

0 commit comments

Comments
 (0)