service: start pool shutdown on broker shutdown.

5 years ago · 23d7a961e7
parent 85cfa3b0f5
commit 23d7a961e7
1 changed files with 4 additions and 0 deletions
--- a/mitogen/service.py
+++ b/mitogen/service.py
@ -82,6 +82,10 @@ def get_or_create_pool(size=None, router=None):
        if _pool_pid != os.getpid():
            _pool = Pool(router, [], size=size or DEFAULT_POOL_SIZE,
                         overwrite=True)
+            # In case of Broker shutdown crash, Pool can cause 'zombie'
+            # processes.
+            mitogen.core.listen(router.broker, 'shutdown',
+                                lambda: _pool.stop(join=False))
            _pool_pid = os.getpid()
        return _pool
    finally: