diff options
| -rw-r--r-- | lib/bb/utils.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/lib/bb/utils.py b/lib/bb/utils.py index f62709bed..4c894cbb0 100644 --- a/lib/bb/utils.py +++ b/lib/bb/utils.py @@ -884,5 +884,17 @@ def process_profilelog(fn): # Was present to work around multiprocessing pool bugs in python < 2.7.3 # def multiprocessingpool(*args, **kwargs): + + import multiprocessing.pool + #import multiprocessing.util + #multiprocessing.util.log_to_stderr(10) + # Deal with a multiprocessing bug where signals to the processes would be delayed until the work + # completes. Putting in a timeout means the signals (like SIGINT/SIGTERM) get processed. + def wrapper(func): + def wrap(self, timeout=None): + return func(self, timeout=timeout if timeout is not None else 1e100) + return wrap + multiprocessing.pool.IMapIterator.next = wrapper(multiprocessing.pool.IMapIterator.next) + return multiprocessing.Pool(*args, **kwargs) |