diff options
author | Charles-Fran?ois Natali <neologix@free.fr> | 2012-04-17 18:45:57 +0200 |
---|---|---|
committer | Charles-Fran?ois Natali <neologix@free.fr> | 2012-04-17 18:45:57 +0200 |
commit | cba36f5380a6f1ca4aa9f07e5c2e3544b9261cdd (patch) | |
tree | 7aa557d68ced4a48f922c43a423833b807abe58b /Lib/multiprocessing/synchronize.py | |
parent | 8d16f2bb39c63af0041cb590d4ad636dc2458759 (diff) | |
download | cpython-cba36f5380a6f1ca4aa9f07e5c2e3544b9261cdd.tar.gz |
Issue #14087: multiprocessing: add Condition.wait_for(). Patch by sbt.
Diffstat (limited to 'Lib/multiprocessing/synchronize.py')
-rw-r--r-- | Lib/multiprocessing/synchronize.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/Lib/multiprocessing/synchronize.py b/Lib/multiprocessing/synchronize.py index e35bbff185..532ac5c1dd 100644 --- a/Lib/multiprocessing/synchronize.py +++ b/Lib/multiprocessing/synchronize.py @@ -43,6 +43,7 @@ import _multiprocessing from multiprocessing.process import current_process from multiprocessing.util import register_after_fork, debug from multiprocessing.forking import assert_spawning, Popen +from time import time as _time # Try to import the mp.synchronize module cleanly, if it fails # raise ImportError for platforms lacking a working sem_open implementation. @@ -290,6 +291,24 @@ class Condition(object): while self._wait_semaphore.acquire(False): pass + def wait_for(self, predicate, timeout=None): + result = predicate() + if result: + return result + if timeout is not None: + endtime = _time() + timeout + else: + endtime = None + waittime = None + while not result: + if endtime is not None: + waittime = endtime - _time() + if waittime <= 0: + break + self.wait(waittime) + result = predicate() + return result + # # Event # |