gh-121621: clear running loop early in asyncio (#128004)

python · Dec 18, 2024 · 8a433b6 · 8a433b6
1 parent dbd08fb
commit 8a433b6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Modules/_asynciomodule.c b/Modules/_asynciomodule.c
@@ -3723,6 +3723,11 @@ module_clear(PyObject *mod)
     Py_CLEAR(state->iscoroutine_typecache);
 
     Py_CLEAR(state->context_kwname);
+    // Clear the ref to running loop so that finalizers can run early.
+    // If there are other running loops in different threads,
+    // those get cleared in PyThreadState_Clear.
+    _PyThreadStateImpl *ts = (_PyThreadStateImpl *)_PyThreadState_GET();
+    Py_CLEAR(ts->asyncio_running_loop);
 
     return 0;
 }