From: Ansgar Burchardt Date: Sat, 26 Mar 2011 11:07:28 +0000 (+0000) Subject: Add a simple multiprocessing wrapper to catch exceptions X-Git-Url: https://git.decadent.org.uk/gitweb/?a=commitdiff_plain;h=defb290b71cff764f4a7574b3131d6fd80db0516;p=dak.git Add a simple multiprocessing wrapper to catch exceptions Signed-off-by: Ansgar Burchardt --- diff --git a/daklib/dakmultiprocessing.py b/daklib/dakmultiprocessing.py new file mode 100644 index 00000000..681755cc --- /dev/null +++ b/daklib/dakmultiprocessing.py @@ -0,0 +1,44 @@ +#!/usr/bin/env python +# vim:set et sw=4: + +""" +multiprocessing for DAK + +@contact: Debian FTP Master +@copyright: 2011 Ansgar Burchardt +@license: GNU General Public License version 2 or later +""" + +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. + +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. + +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +############################################################################### + +import multiprocessing + +class Pool(): + def __init__(self, *args, **kwds): + self.pool = multiprocessing.Pool(*args, **kwds) + self.results = [] + + def apply_async(self, func, args=(), kwds={}, callback=None): + self.results.append(self.pool.apply_async(func, args, kwds, callback)) + + def close(self): + self.pool.close() + + def join(self): + self.pool.join() + for r in self.results: + r.get()