Send exceptions across CoroutinePool.wait
This commit is contained in:
@@ -44,6 +44,11 @@ class Cancelled(RuntimeError):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class ExceptionWrapper(object):
|
||||||
|
def __init__(self, e):
|
||||||
|
self.e = e
|
||||||
|
|
||||||
|
|
||||||
NOT_USED = object()
|
NOT_USED = object()
|
||||||
|
|
||||||
|
|
||||||
@@ -311,6 +316,13 @@ class CoroutinePool(pools.Pool):
|
|||||||
traceback.print_exc()
|
traceback.print_exc()
|
||||||
if evt is not None:
|
if evt is not None:
|
||||||
evt.send(exc=e)
|
evt.send(exc=e)
|
||||||
|
if self._tracked_events is not None:
|
||||||
|
if self._next_event is None:
|
||||||
|
self._tracked_events.append(ExceptionWrapper(e))
|
||||||
|
else:
|
||||||
|
ne = self._next_event
|
||||||
|
self._next_event = None
|
||||||
|
ne.send(exc=e)
|
||||||
|
|
||||||
def _execute(self, evt, func, args, kw):
|
def _execute(self, evt, func, args, kw):
|
||||||
""" Private implementation of the execute methods.
|
""" Private implementation of the execute methods.
|
||||||
@@ -384,6 +396,9 @@ class CoroutinePool(pools.Pool):
|
|||||||
return self._next_event.wait()
|
return self._next_event.wait()
|
||||||
|
|
||||||
result = self._tracked_events.pop(0)
|
result = self._tracked_events.pop(0)
|
||||||
|
if isinstance(result, ExceptionWrapper):
|
||||||
|
raise result.e
|
||||||
|
|
||||||
if not self._tracked_events:
|
if not self._tracked_events:
|
||||||
self._next_event = event()
|
self._next_event = event()
|
||||||
return result
|
return result
|
||||||
|
Reference in New Issue
Block a user