8000 gh-110205: Fix asyncio ThreadedChildWatcher._join_threads() by gvanrossum · Pull Request #110884 · python/cpython · GitHub
[go: up one dir, main page]

Skip to content

gh-110205: Fix asyncio ThreadedChildWatcher._join_threads() #110884

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Oct 27, 2023
Merged
Next Next commit
gh-110205: Fix asyncio ThreadedChildWatcher._join_threads()
ThreadedChildWatcher._join_threads() now clears references to
completed threads.

test_asyncio.utils.TestCase now calls _join_threads() of the watcher,
uses SHORT_TIMEOUT to join a thread, and then raises an exception if
there are still running threads.

Rename also ThreadedChildWatcher threads to add "asyncio-" prefix to
the name.
  • Loading branch information
vstinner committed Oct 13, 2023
commit d414ed266bfc747d03ea9044ea0f662e2b388f1c
10 changes: 7 additions & 3 deletions Lib/asyncio/unix_events.py
Original file line number Diff line number Diff line change
Expand Up @@ -1373,12 +1373,16 @@ def is_active(self):
def close(self):
self._join_threads()

def _join_threads(self):
def _join_threads(self, timeout=None):
"""Internal: Join all non-daemon threads"""
threads = [thread for thread in list(self._threads.values())
if thread.is_alive() and not thread.daemon]
for thread in threads:
thread.join()
thread.join(timeout)

# Clear references to terminated threads
self.threads = [thread for thread in list(self._threads.values())
if thread.is_alive() and not thread.daemon]

def __enter__(self):
return self
Expand All @@ -1397,7 +1401,7 @@ def __del__(self, _warn=warnings.warn):
def add_child_handler(self, pid, callback, *args):
loop = events.get_running_loop()
thread = threading.Thread(target=self._do_waitpid,
name=f"waitpid-{next(self._pid_counter)}",
name=f"asyncio-waitpid-{next(self._pid_counter)}",
args=(loop, pid, callback, args),
daemon=True)
self._threads[pid] = thread
Expand Down
9 changes: 6 additions & 3 deletions Lib/test/test_asyncio/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -546,6 +546,7 @@ def close_loop(loop):
else:
loop._default_executor.shutdown(wait=True)
loop.close()

policy = support.maybe_get_event_loop_policy()
if policy is not None:
try:
Expand All @@ -557,9 +558,11 @@ def close_loop(loop):
pass
else:
if isinstance(watcher, asyncio.ThreadedChildWatcher):
threads = list(watcher._threads.values())
for thread in threads:
thread.join()
watcher._join_threads(timeout=support.SHORT_TIMEOUT)
threads = watcher._threads
if threads:
self.fail(f"watcher still has running threads: "
f"{threads}")

def set_event_loop(self, loop, *, cleanup=True):
if loop is None:
Expand Down
0