Mercurial > cpython-lru_cache-weakref
diff Lib/multiprocessing/pool.py @ 77140:57d6265beaaa
Issue #12091: simplify ApplyResult and MapResult with threading.Event
Patch by Charles-François Natali
| author | Richard Oudkerk <shibturn@gmail.com> |
|---|---|
| date | Fri, 25 May 2012 13:26:53 +0100 |
| parents | 25cc0c8fd4e2 |
| children | 3585cb1388f2 |
line wrap: on
line diff
--- a/Lib/multiprocessing/pool.py Fri May 25 13:04:20 2012 +0100 +++ b/Lib/multiprocessing/pool.py Fri May 25 13:26:53 2012 +0100 @@ -526,32 +526,26 @@ class ApplyResult(object): def __init__(self, cache, callback, error_callback): - self._cond = threading.Condition(threading.Lock()) + self._event = threading.Event() self._job = next(job_counter) self._cache = cache - self._ready = False self._callback = callback self._error_callback = error_callback cache[self._job] = self def ready(self): - return self._ready + return self._event.is_set() def successful(self): - assert self._ready + assert self.ready() return self._success def wait(self, timeout=None): - self._cond.acquire() - try: - if not self._ready: - self._cond.wait(timeout) - finally: - self._cond.release() + self._event.wait(timeout) def get(self, timeout=None): self.wait(timeout) - if not self._ready: + if not self.ready(): raise TimeoutError if self._success: return self._value @@ -564,12 +558,7 @@ self._callback(self._value) if self._error_callback and not self._success: self._error_callback(self._value) - self._cond.acquire() - try: - self._ready = True - self._cond.notify() - finally: - self._cond.release() + self._event.set() del self._cache[self._job] # @@ -586,7 +575,7 @@ self._chunksize = chunksize if chunksize <= 0: self._number_left = 0 - self._ready = True + self._event.set() else: self._number_left = length//chunksize + bool(length % chunksize) @@ -599,24 +588,14 @@ if self._callback: self._callback(self._value) del self._cache[self._job] - self._cond.acquire() - try: - self._ready = True - self._cond.notify() - finally: - self._cond.release() + self._event.set() else: self._success = False self._value = result if self._error_callback: self._error_callback(self._value) del self._cache[self._job] - self._cond.acquire() - try: - self._ready = True - self._cond.notify() - finally: - self._cond.release() + self._event.set() # # Class whose instances are returned by `Pool.imap()`
