Repository navigation
Remote asyncio stack traces stop at async generators #159620
Description
Bug report
Stacks of tasks suspended in async for, anext() or aclose() end before the async generator, so its frames and everything it awaits are missing
Repro:
import asyncio
import os
async def agen():
await asyncio.sleep(3600)
yield
async def main():
async for _ in agen():
pass
print(os.getpid())
asyncio.run(main())Run (for example i took asyncio pstree)
sudo ./python.exe -m asyncio pstree <pid>
Actual:
└── (T) Task-1
└── main /Users/timofeiivankov/cpython/repro.py:9
Expected:
└── (T) Task-1
└── main /Users/timofeiivankov/cpython/repro.py:9
└── agen /Users/timofeiivankov/cpython/repro.py:5
└── sleep /Users/timofeiivankov/cpython/Lib/asyncio/tasks.py:723
It also affecting those commands:
sudo ./python.exe -m asyncio ps <pid>
sudo ./python.exe -m profiling.sampling dump --async-aware <pid>
sudo ./python.exe -m profiling.sampling run -d 5 --async-aware --async-mode all repro.py
sudo ./python.exe -m profiling.sampling attach -d 5 --async-aware --async-mode all <pid>
But to avoid blowing the issue out of proportion, I decided to just mention them
Proposed fix: follow asend/athrow wrappers to their async generator when walking the coroutine chain
I have a fix ready
CPython versions tested on:
CPython main branch
Operating systems tested on:
macOS
Activity
fixup! pythongh-159620: Fix remote asyncio stacks stopping at async g…
gh-159620: Fix remote asyncio stacks stopping at async generators (GH…
fixup! [3.14] pythongh-159620: Fix remote asyncio stacks stopping at …
[3.15] gh-159620: Fix remote asyncio stacks stopping at async generat…
[3.14] gh-159620: Fix remote asyncio stacks stopping at async generat…