Skip to content

Commit 9d19adb

Browse files
Handle StreamReset, and acknowledge received data (#81)
1 parent d8950ef commit 9d19adb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

httpcore/dispatch/http2.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ async def send(
3131
#  Start sending the request.
3232
if not self.initialized:
3333
self.initiate_connection()
34+
3435
stream_id = await self.send_headers(request, timeout)
3536
self.events[stream_id] = []
3637

@@ -117,8 +118,9 @@ async def body_iter(
117118
while True:
118119
event = await self.receive_event(stream_id, timeout)
119120
if isinstance(event, h2.events.DataReceived):
121+
self.h2_state.acknowledge_received_data(len(event.data), stream_id)
120122
yield event.data
121-
elif isinstance(event, h2.events.StreamEnded):
123+
elif isinstance(event, (h2.events.StreamEnded, h2.events.StreamReset)):
122124
break
123125

124126
async def receive_event(

0 commit comments

Comments
 (0)