async_generator: keep pushing frames downstream
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user