fix(taskqueue): handle BrokenProcessPool explicitly

talkiq · Jan 24, 2018 · bc07576 · bc07576
1 parent f2a6b0c
commit bc07576
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/taskqueue/gcloud/aio/taskqueue/taskmanager.py b/taskqueue/gcloud/aio/taskqueue/taskmanager.py
@@ -166,7 +166,7 @@ async def process(self, task):
         # and at this point I'm sick of having to update the blacklist. If we
         # get an exception while creating a LeaseManager, that's enough for me
         # to be reasonably sure the pool is broken without needing proof.
-        except Exception as e:  # pylint: disable=broad-except
+        except (concurrent.futures.process.BrokenProcessPool, Exception) as e:  # pylint: disable=broad-except
             log.error('process pool broke, quitting TaskManager')
             log.exception(e)
             self.running = False

diff --git a/taskqueue/setup.py b/taskqueue/setup.py
@@ -13,7 +13,7 @@
 
 setuptools.setup(
     name='gcloud-aio-taskqueue',
-    version='1.2.4',
+    version='1.2.5',
     description='Asyncio Python Client for Google Cloud Task Queue',
     long_description=README,
     namespace_packages=[