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): async def process_frame(self, frame: Frame, direction: FrameDirection):
await super().process_frame(frame, direction) await super().process_frame(frame, direction)
await self.push_frame(frame, direction)
if isinstance(frame, (CancelFrame, EndFrame)): if isinstance(frame, (CancelFrame, EndFrame)):
await self._data_queue.put(None) await self._data_queue.put(None)
else: else: