async_generator: keep pushing frames downstream

This commit is contained in:
Aleix Conchillo Flaqué
2024-09-26 16:44:49 -07:00
parent b8ece84c6e
commit d323ea9e95

View File

@@ -26,6 +26,8 @@ class AsyncGeneratorProcessor(FrameProcessor):
async def process_frame(self, frame: Frame, direction: FrameDirection):
await super().process_frame(frame, direction)
await self.push_frame(frame, direction)
if isinstance(frame, (CancelFrame, EndFrame)):
await self._data_queue.put(None)
else: