| # |
| # Module providing the `SyncManager` class for dealing |
| # with shared objects |
| # |
| # multiprocessing/managers.py |
| # |
| # Copyright (c) 2006-2008, R Oudkerk |
| # All rights reserved. |
| # |
| # Redistribution and use in source and binary forms, with or without |
| # modification, are permitted provided that the following conditions |
| # are met: |
| # |
| # 1. Redistributions of source code must retain the above copyright |
| # notice, this list of conditions and the following disclaimer. |
| # 2. Redistributions in binary form must reproduce the above copyright |
| # notice, this list of conditions and the following disclaimer in the |
| # documentation and/or other materials provided with the distribution. |
| # 3. Neither the name of author nor the names of any contributors may be |
| # used to endorse or promote products derived from this software |
| # without specific prior written permission. |
| # |
| # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS "AS IS" AND |
| # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE |
| # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
| # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
| # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
| # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
| # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
| # SUCH DAMAGE. |
| # |
| |
| __all__ = [ 'BaseManager', 'SyncManager', 'BaseProxy', 'Token' ] |
| |
| # |
| # Imports |
| # |
| |
| import sys |
| import threading |
| import array |
| import queue |
| |
| from traceback import format_exc |
| from multiprocessing import Process, current_process, active_children, Pool, util, connection |
| from multiprocessing.process import AuthenticationString |
| from multiprocessing.forking import exit, Popen, ForkingPickler |
| from time import time as _time |
| |
| # |
| # Register some things for pickling |
| # |
| |
| def reduce_array(a): |
| return array.array, (a.typecode, a.tobytes()) |
| ForkingPickler.register(array.array, reduce_array) |
| |
| view_types = [type(getattr({}, name)()) for name in ('items','keys','values')] |
| if view_types[0] is not list: # only needed in Py3.0 |
| def rebuild_as_list(obj): |
| return list, (list(obj),) |
| for view_type in view_types: |
| ForkingPickler.register(view_type, rebuild_as_list) |
| import copyreg |
| copyreg.pickle(view_type, rebuild_as_list) |
| |
| # |
| # Type for identifying shared objects |
| # |
| |
| class Token(object): |
| ''' |
| Type to uniquely indentify a shared object |
| ''' |
| __slots__ = ('typeid', 'address', 'id') |
| |
| def __init__(self, typeid, address, id): |
| (self.typeid, self.address, self.id) = (typeid, address, id) |
| |
| def __getstate__(self): |
| return (self.typeid, self.address, self.id) |
| |
| def __setstate__(self, state): |
| (self.typeid, self.address, self.id) = state |
| |
| def __repr__(self): |
| return 'Token(typeid=%r, address=%r, id=%r)' % \ |
| (self.typeid, self.address, self.id) |
| |
| # |
| # Function for communication with a manager's server process |
| # |
| |
| def dispatch(c, id, methodname, args=(), kwds={}): |
| ''' |
| Send a message to manager using connection `c` and return response |
| ''' |
| c.send((id, methodname, args, kwds)) |
| kind, result = c.recv() |
| if kind == '#RETURN': |
| return result |
| raise convert_to_error(kind, result) |
| |
| def convert_to_error(kind, result): |
| if kind == '#ERROR': |
| return result |
| elif kind == '#TRACEBACK': |
| assert type(result) is str |
| return RemoteError(result) |
| elif kind == '#UNSERIALIZABLE': |
| assert type(result) is str |
| return RemoteError('Unserializable message: %s\n' % result) |
| else: |
| return ValueError('Unrecognized message type') |
| |
| class RemoteError(Exception): |
| def __str__(self): |
| return ('\n' + '-'*75 + '\n' + str(self.args[0]) + '-'*75) |
| |
| # |
| # Functions for finding the method names of an object |
| # |
| |
| def all_methods(obj): |
| ''' |
| Return a list of names of methods of `obj` |
| ''' |
| temp = [] |
| for name in dir(obj): |
| func = getattr(obj, name) |
| if callable(func): |
| temp.append(name) |
| return temp |
| |
| def public_methods(obj): |
| ''' |
| Return a list of names of methods of `obj` which do not start with '_' |
| ''' |
| return [name for name in all_methods(obj) if name[0] != '_'] |
| |
| # |
| # Server which is run in a process controlled by a manager |
| # |
| |
| class Server(object): |
| ''' |
| Server class which runs in a process controlled by a manager object |
| ''' |
| public = ['shutdown', 'create', 'accept_connection', 'get_methods', |
| 'debug_info', 'number_of_objects', 'dummy', 'incref', 'decref'] |
| |
| def __init__(self, registry, address, authkey, serializer): |
| assert isinstance(authkey, bytes) |
| self.registry = registry |
| self.authkey = AuthenticationString(authkey) |
| Listener, Client = listener_client[serializer] |
| |
| # do authentication later |
| self.listener = Listener(address=address, backlog=16) |
| self.address = self.listener.address |
| |
| self.id_to_obj = {'0': (None, ())} |
| self.id_to_refcount = {} |
| self.mutex = threading.RLock() |
| self.stop = 0 |
| |
| def serve_forever(self): |
| ''' |
| Run the server forever |
| ''' |
| current_process()._manager_server = self |
| try: |
| try: |
| while 1: |
| try: |
| c = self.listener.accept() |
| except (OSError, IOError): |
| continue |
| t = threading.Thread(target=self.handle_request, args=(c,)) |
| t.daemon = True |
| t.start() |
| except (KeyboardInterrupt, SystemExit): |
| pass |
| finally: |
| self.stop = 999 |
| self.listener.close() |
| |
| def handle_request(self, c): |
| ''' |
| Handle a new connection |
| ''' |
| funcname = result = request = None |
| try: |
| connection.deliver_challenge(c, self.authkey) |
| connection.answer_challenge(c, self.authkey) |
| request = c.recv() |
| ignore, funcname, args, kwds = request |
| assert funcname in self.public, '%r unrecognized' % funcname |
| func = getattr(self, funcname) |
| except Exception: |
| msg = ('#TRACEBACK', format_exc()) |
| else: |
| try: |
| result = func(c, *args, **kwds) |
| except Exception: |
| msg = ('#TRACEBACK', format_exc()) |
| else: |
| msg = ('#RETURN', result) |
| try: |
| c.send(msg) |
| except Exception as e: |
| try: |
| c.send(('#TRACEBACK', format_exc())) |
| except Exception: |
| pass |
| util.info('Failure to send message: %r', msg) |
| util.info(' ... request was %r', request) |
| util.info(' ... exception was %r', e) |
| |
| c.close() |
| |
| def serve_client(self, conn): |
| ''' |
| Handle requests from the proxies in a particular process/thread |
| ''' |
| util.debug('starting server thread to service %r', |
| threading.current_thread().name) |
| |
| recv = conn.recv |
| send = conn.send |
| id_to_obj = self.id_to_obj |
| |
| while not self.stop: |
| |
| try: |
| methodname = obj = None |
| request = recv() |
| ident, methodname, args, kwds = request |
| obj, exposed, gettypeid = id_to_obj[ident] |
| |
| if methodname not in exposed: |
| raise AttributeError( |
| 'method %r of %r object is not in exposed=%r' % |
| (methodname, type(obj), exposed) |
| ) |
| |
| function = getattr(obj, methodname) |
| |
| try: |
| res = function(*args, **kwds) |
| except Exception as e: |
| msg = ('#ERROR', e) |
| else: |
| typeid = gettypeid and gettypeid.get(methodname, None) |
| if typeid: |
| rident, rexposed = self.create(conn, typeid, res) |
| token = Token(typeid, self.address, rident) |
| msg = ('#PROXY', (rexposed, token)) |
| else: |
| msg = ('#RETURN', res) |
| |
| except AttributeError: |
| if methodname is None: |
| msg = ('#TRACEBACK', format_exc()) |
| else: |
| try: |
| fallback_func = self.fallback_mapping[methodname] |
| result = fallback_func( |
| self, conn, ident, obj, *args, **kwds |
| ) |
| msg = ('#RETURN', result) |
| except Exception: |
| msg = ('#TRACEBACK', format_exc()) |
| |
| except EOFError: |
| util.debug('got EOF -- exiting thread serving %r', |
| threading.current_thread().name) |
| sys.exit(0) |
| |
| except Exception: |
| msg = ('#TRACEBACK', format_exc()) |
| |
| try: |
| try: |
| send(msg) |
| except Exception as e: |
| send(('#UNSERIALIZABLE', repr(msg))) |
| except Exception as e: |
| util.info('exception in thread serving %r', |
| threading.current_thread().name) |
| util.info(' ... message was %r', msg) |
| util.info(' ... exception was %r', e) |
| conn.close() |
| sys.exit(1) |
| |
| def fallback_getvalue(self, conn, ident, obj): |
| return obj |
| |
| def fallback_str(self, conn, ident, obj): |
| return str(obj) |
| |
| def fallback_repr(self, conn, ident, obj): |
| return repr(obj) |
| |
| fallback_mapping = { |
| '__str__':fallback_str, |
| '__repr__':fallback_repr, |
| '#GETVALUE':fallback_getvalue |
| } |
| |
| def dummy(self, c): |
| pass |
| |
| def debug_info(self, c): |
| ''' |
| Return some info --- useful to spot problems with refcounting |
| ''' |
| self.mutex.acquire() |
| try: |
| result = [] |
| keys = list(self.id_to_obj.keys()) |
| keys.sort() |
| for ident in keys: |
| if ident != '0': |
| result.append(' %s: refcount=%s\n %s' % |
| (ident, self.id_to_refcount[ident], |
| str(self.id_to_obj[ident][0])[:75])) |
| return '\n'.join(result) |
| finally: |
| self.mutex.release() |
| |
| def number_of_objects(self, c): |
| ''' |
| Number of shared objects |
| ''' |
| return len(self.id_to_obj) - 1 # don't count ident='0' |
| |
| def shutdown(self, c): |
| ''' |
| Shutdown this process |
| ''' |
| try: |
| try: |
| util.debug('manager received shutdown message') |
| c.send(('#RETURN', None)) |
| |
| if sys.stdout != sys.__stdout__: |
| util.debug('resetting stdout, stderr') |
| sys.stdout = sys.__stdout__ |
| sys.stderr = sys.__stderr__ |
| |
| util._run_finalizers(0) |
| |
| for p in active_children(): |
| util.debug('terminating a child process of manager') |
| p.terminate() |
| |
| for p in active_children(): |
| util.debug('terminating a child process of manager') |
| p.join() |
| |
| util._run_finalizers() |
| util.info('manager exiting with exitcode 0') |
| except: |
| import traceback |
| traceback.print_exc() |
| finally: |
| exit(0) |
| |
| def create(self, c, typeid, *args, **kwds): |
| ''' |
| Create a new shared object and return its id |
| ''' |
| self.mutex.acquire() |
| try: |
| callable, exposed, method_to_typeid, proxytype = \ |
| self.registry[typeid] |
| |
| if callable is None: |
| assert len(args) == 1 and not kwds |
| obj = args[0] |
| else: |
| obj = callable(*args, **kwds) |
| |
| if exposed is None: |
| exposed = public_methods(obj) |
| if method_to_typeid is not None: |
| assert type(method_to_typeid) is dict |
| exposed = list(exposed) + list(method_to_typeid) |
| |
| ident = '%x' % id(obj) # convert to string because xmlrpclib |
| # only has 32 bit signed integers |
| util.debug('%r callable returned object with id %r', typeid, ident) |
| |
| self.id_to_obj[ident] = (obj, set(exposed), method_to_typeid) |
| if ident not in self.id_to_refcount: |
| self.id_to_refcount[ident] = 0 |
| # increment the reference count immediately, to avoid |
| # this object being garbage collected before a Proxy |
| # object for it can be created. The caller of create() |
| # is responsible for doing a decref once the Proxy object |
| # has been created. |
| self.incref(c, ident) |
| return ident, tuple(exposed) |
| finally: |
| self.mutex.release() |
| |
| def get_methods(self, c, token): |
| ''' |
| Return the methods of the shared object indicated by token |
| ''' |
| return tuple(self.id_to_obj[token.id][1]) |
| |
| def accept_connection(self, c, name): |
| ''' |
| Spawn a new thread to serve this connection |
| ''' |
| threading.current_thread().name = name |
| c.send(('#RETURN', None)) |
| self.serve_client(c) |
| |
| def incref(self, c, ident): |
| self.mutex.acquire() |
| try: |
| self.id_to_refcount[ident] += 1 |
| finally: |
| self.mutex.release() |
| |
| def decref(self, c, ident): |
| self.mutex.acquire() |
| try: |
| assert self.id_to_refcount[ident] >= 1 |
| self.id_to_refcount[ident] -= 1 |
| if self.id_to_refcount[ident] == 0: |
| del self.id_to_obj[ident], self.id_to_refcount[ident] |
| util.debug('disposing of obj with id %r', ident) |
| finally: |
| self.mutex.release() |
| |
| # |
| # Class to represent state of a manager |
| # |
| |
| class State(object): |
| __slots__ = ['value'] |
| INITIAL = 0 |
| STARTED = 1 |
| SHUTDOWN = 2 |
| |
| # |
| # Mapping from serializer name to Listener and Client types |
| # |
| |
| listener_client = { |
| 'pickle' : (connection.Listener, connection.Client), |
| 'xmlrpclib' : (connection.XmlListener, connection.XmlClient) |
| } |
| |
| # |
| # Definition of BaseManager |
| # |
| |
| class BaseManager(object): |
| ''' |
| Base class for managers |
| ''' |
| _registry = {} |
| _Server = Server |
| |
| def __init__(self, address=None, authkey=None, serializer='pickle'): |
| if authkey is None: |
| authkey = current_process().authkey |
| self._address = address # XXX not final address if eg ('', 0) |
| self._authkey = AuthenticationString(authkey) |
| self._state = State() |
| self._state.value = State.INITIAL |
| self._serializer = serializer |
| self._Listener, self._Client = listener_client[serializer] |
| |
| def __reduce__(self): |
| return type(self).from_address, \ |
| (self._address, self._authkey, self._serializer) |
| |
| def get_server(self): |
| ''' |
| Return server object with serve_forever() method and address attribute |
| ''' |
| assert self._state.value == State.INITIAL |
| return Server(self._registry, self._address, |
| self._authkey, self._serializer) |
| |
| def connect(self): |
| ''' |
| Connect manager object to the server process |
| ''' |
| Listener, Client = listener_client[self._serializer] |
| conn = Client(self._address, authkey=self._authkey) |
| dispatch(conn, None, 'dummy') |
| self._state.value = State.STARTED |
| |
| def start(self, initializer=None, initargs=()): |
| ''' |
| Spawn a server process for this manager object |
| ''' |
| assert self._state.value == State.INITIAL |
| |
| if initializer is not None and not callable(initializer): |
| raise TypeError('initializer must be a callable') |
| |
| # pipe over which we will retrieve address of server |
| reader, writer = connection.Pipe(duplex=False) |
| |
| # spawn process which runs a server |
| self._process = Process( |
| target=type(self)._run_server, |
| args=(self._registry, self._address, self._authkey, |
| self._serializer, writer, initializer, initargs), |
| ) |
| ident = ':'.join(str(i) for i in self._process._identity) |
| self._process.name = type(self).__name__ + '-' + ident |
| self._process.start() |
| |
| # get address of server |
| writer.close() |
| self._address = reader.recv() |
| reader.close() |
| |
| # register a finalizer |
| self._state.value = State.STARTED |
| self.shutdown = util.Finalize( |
| self, type(self)._finalize_manager, |
| args=(self._process, self._address, self._authkey, |
| self._state, self._Client), |
| exitpriority=0 |
| ) |
| |
| @classmethod |
| def _run_server(cls, registry, address, authkey, serializer, writer, |
| initializer=None, initargs=()): |
| ''' |
| Create a server, report its address and run it |
| ''' |
| if initializer is not None: |
| initializer(*initargs) |
| |
| # create server |
| server = cls._Server(registry, address, authkey, serializer) |
| |
| # inform parent process of the server's address |
| writer.send(server.address) |
| writer.close() |
| |
| # run the manager |
| util.info('manager serving at %r', server.address) |
| server.serve_forever() |
| |
| def _create(self, typeid, *args, **kwds): |
| ''' |
| Create a new shared object; return the token and exposed tuple |
| ''' |
| assert self._state.value == State.STARTED, 'server not yet started' |
| conn = self._Client(self._address, authkey=self._authkey) |
| try: |
| id, exposed = dispatch(conn, None, 'create', (typeid,)+args, kwds) |
| finally: |
| conn.close() |
| return Token(typeid, self._address, id), exposed |
| |
| def join(self, timeout=None): |
| ''' |
| Join the manager process (if it has been spawned) |
| ''' |
| self._process.join(timeout) |
| |
| def _debug_info(self): |
| ''' |
| Return some info about the servers shared objects and connections |
| ''' |
| conn = self._Client(self._address, authkey=self._authkey) |
| try: |
| return dispatch(conn, None, 'debug_info') |
| finally: |
| conn.close() |
| |
| def _number_of_objects(self): |
| ''' |
| Return the number of shared objects |
| ''' |
| conn = self._Client(self._address, authkey=self._authkey) |
| try: |
| return dispatch(conn, None, 'number_of_objects') |
| finally: |
| conn.close() |
| |
| def __enter__(self): |
| return self |
| |
| def __exit__(self, exc_type, exc_val, exc_tb): |
| self.shutdown() |
| |
| @staticmethod |
| def _finalize_manager(process, address, authkey, state, _Client): |
| ''' |
| Shutdown the manager process; will be registered as a finalizer |
| ''' |
| if process.is_alive(): |
| util.info('sending shutdown message to manager') |
| try: |
| conn = _Client(address, authkey=authkey) |
| try: |
| dispatch(conn, None, 'shutdown') |
| finally: |
| conn.close() |
| except Exception: |
| pass |
| |
| process.join(timeout=0.2) |
| if process.is_alive(): |
| util.info('manager still alive') |
| if hasattr(process, 'terminate'): |
| util.info('trying to `terminate()` manager process') |
| process.terminate() |
| process.join(timeout=0.1) |
| if process.is_alive(): |
| util.info('manager still alive after terminate') |
| |
| state.value = State.SHUTDOWN |
| try: |
| del BaseProxy._address_to_local[address] |
| except KeyError: |
| pass |
| |
| address = property(lambda self: self._address) |
| |
| @classmethod |
| def register(cls, typeid, callable=None, proxytype=None, exposed=None, |
| method_to_typeid=None, create_method=True): |
| ''' |
| Register a typeid with the manager type |
| ''' |
| if '_registry' not in cls.__dict__: |
| cls._registry = cls._registry.copy() |
| |
| if proxytype is None: |
| proxytype = AutoProxy |
| |
| exposed = exposed or getattr(proxytype, '_exposed_', None) |
| |
| method_to_typeid = method_to_typeid or \ |
| getattr(proxytype, '_method_to_typeid_', None) |
| |
| if method_to_typeid: |
| for key, value in list(method_to_typeid.items()): |
| assert type(key) is str, '%r is not a string' % key |
| assert type(value) is str, '%r is not a string' % value |
| |
| cls._registry[typeid] = ( |
| callable, exposed, method_to_typeid, proxytype |
| ) |
| |
| if create_method: |
| def temp(self, *args, **kwds): |
| util.debug('requesting creation of a shared %r object', typeid) |
| token, exp = self._create(typeid, *args, **kwds) |
| proxy = proxytype( |
| token, self._serializer, manager=self, |
| authkey=self._authkey, exposed=exp |
| ) |
| conn = self._Client(token.address, authkey=self._authkey) |
| dispatch(conn, None, 'decref', (token.id,)) |
| return proxy |
| temp.__name__ = typeid |
| setattr(cls, typeid, temp) |
| |
| # |
| # Subclass of set which get cleared after a fork |
| # |
| |
| class ProcessLocalSet(set): |
| def __init__(self): |
| util.register_after_fork(self, lambda obj: obj.clear()) |
| def __reduce__(self): |
| return type(self), () |
| |
| # |
| # Definition of BaseProxy |
| # |
| |
| class BaseProxy(object): |
| ''' |
| A base for proxies of shared objects |
| ''' |
| _address_to_local = {} |
| _mutex = util.ForkAwareThreadLock() |
| |
| def __init__(self, token, serializer, manager=None, |
| authkey=None, exposed=None, incref=True): |
| BaseProxy._mutex.acquire() |
| try: |
| tls_idset = BaseProxy._address_to_local.get(token.address, None) |
| if tls_idset is None: |
| tls_idset = util.ForkAwareLocal(), ProcessLocalSet() |
| BaseProxy._address_to_local[token.address] = tls_idset |
| finally: |
| BaseProxy._mutex.release() |
| |
| # self._tls is used to record the connection used by this |
| # thread to communicate with the manager at token.address |
| self._tls = tls_idset[0] |
| |
| # self._idset is used to record the identities of all shared |
| # objects for which the current process owns references and |
| # which are in the manager at token.address |
| self._idset = tls_idset[1] |
| |
| self._token = token |
| self._id = self._token.id |
| self._manager = manager |
| self._serializer = serializer |
| self._Client = listener_client[serializer][1] |
| |
| if authkey is not None: |
| self._authkey = AuthenticationString(authkey) |
| elif self._manager is not None: |
| self._authkey = self._manager._authkey |
| else: |
| self._authkey = current_process().authkey |
| |
| if incref: |
| self._incref() |
| |
| util.register_after_fork(self, BaseProxy._after_fork) |
| |
| def _connect(self): |
| util.debug('making connection to manager') |
| name = current_process().name |
| if threading.current_thread().name != 'MainThread': |
| name += '|' + threading.current_thread().name |
| conn = self._Client(self._token.address, authkey=self._authkey) |
| dispatch(conn, None, 'accept_connection', (name,)) |
| self._tls.connection = conn |
| |
| def _callmethod(self, methodname, args=(), kwds={}): |
| ''' |
| Try to call a method of the referrent and return a copy of the result |
| ''' |
| try: |
| conn = self._tls.connection |
| except AttributeError: |
| util.debug('thread %r does not own a connection', |
| threading.current_thread().name) |
| self._connect() |
| conn = self._tls.connection |
| |
| conn.send((self._id, methodname, args, kwds)) |
| kind, result = conn.recv() |
| |
| if kind == '#RETURN': |
| return result |
| elif kind == '#PROXY': |
| exposed, token = result |
| proxytype = self._manager._registry[token.typeid][-1] |
| proxy = proxytype( |
| token, self._serializer, manager=self._manager, |
| authkey=self._authkey, exposed=exposed |
| ) |
| conn = self._Client(token.address, authkey=self._authkey) |
| dispatch(conn, None, 'decref', (token.id,)) |
| return proxy |
| raise convert_to_error(kind, result) |
| |
| def _getvalue(self): |
| ''' |
| Get a copy of the value of the referent |
| ''' |
| return self._callmethod('#GETVALUE') |
| |
| def _incref(self): |
| conn = self._Client(self._token.address, authkey=self._authkey) |
| dispatch(conn, None, 'incref', (self._id,)) |
| util.debug('INCREF %r', self._token.id) |
| |
| self._idset.add(self._id) |
| |
| state = self._manager and self._manager._state |
| |
| self._close = util.Finalize( |
| self, BaseProxy._decref, |
| args=(self._token, self._authkey, state, |
| self._tls, self._idset, self._Client), |
| exitpriority=10 |
| ) |
| |
| @staticmethod |
| def _decref(token, authkey, state, tls, idset, _Client): |
| idset.discard(token.id) |
| |
| # check whether manager is still alive |
| if state is None or state.value == State.STARTED: |
| # tell manager this process no longer cares about referent |
| try: |
| util.debug('DECREF %r', token.id) |
| conn = _Client(token.address, authkey=authkey) |
| dispatch(conn, None, 'decref', (token.id,)) |
| except Exception as e: |
| util.debug('... decref failed %s', e) |
| |
| else: |
| util.debug('DECREF %r -- manager already shutdown', token.id) |
| |
| # check whether we can close this thread's connection because |
| # the process owns no more references to objects for this manager |
| if not idset and hasattr(tls, 'connection'): |
| util.debug('thread %r has no more proxies so closing conn', |
| threading.current_thread().name) |
| tls.connection.close() |
| del tls.connection |
| |
| def _after_fork(self): |
| self._manager = None |
| try: |
| self._incref() |
| except Exception as e: |
| # the proxy may just be for a manager which has shutdown |
| util.info('incref failed: %s' % e) |
| |
| def __reduce__(self): |
| kwds = {} |
| if Popen.thread_is_spawning(): |
| kwds['authkey'] = self._authkey |
| |
| if getattr(self, '_isauto', False): |
| kwds['exposed'] = self._exposed_ |
| return (RebuildProxy, |
| (AutoProxy, self._token, self._serializer, kwds)) |
| else: |
| return (RebuildProxy, |
| (type(self), self._token, self._serializer, kwds)) |
| |
| def __deepcopy__(self, memo): |
| return self._getvalue() |
| |
| def __repr__(self): |
| return '<%s object, typeid %r at %s>' % \ |
| (type(self).__name__, self._token.typeid, '0x%x' % id(self)) |
| |
| def __str__(self): |
| ''' |
| Return representation of the referent (or a fall-back if that fails) |
| ''' |
| try: |
| return self._callmethod('__repr__') |
| except Exception: |
| return repr(self)[:-1] + "; '__str__()' failed>" |
| |
| # |
| # Function used for unpickling |
| # |
| |
| def RebuildProxy(func, token, serializer, kwds): |
| ''' |
| Function used for unpickling proxy objects. |
| |
| If possible the shared object is returned, or otherwise a proxy for it. |
| ''' |
| server = getattr(current_process(), '_manager_server', None) |
| |
| if server and server.address == token.address: |
| return server.id_to_obj[token.id][0] |
| else: |
| incref = ( |
| kwds.pop('incref', True) and |
| not getattr(current_process(), '_inheriting', False) |
| ) |
| return func(token, serializer, incref=incref, **kwds) |
| |
| # |
| # Functions to create proxies and proxy types |
| # |
| |
| def MakeProxyType(name, exposed, _cache={}): |
| ''' |
| Return an proxy type whose methods are given by `exposed` |
| ''' |
| exposed = tuple(exposed) |
| try: |
| return _cache[(name, exposed)] |
| except KeyError: |
| pass |
| |
| dic = {} |
| |
| for meth in exposed: |
| exec('''def %s(self, *args, **kwds): |
| return self._callmethod(%r, args, kwds)''' % (meth, meth), dic) |
| |
| ProxyType = type(name, (BaseProxy,), dic) |
| ProxyType._exposed_ = exposed |
| _cache[(name, exposed)] = ProxyType |
| return ProxyType |
| |
| |
| def AutoProxy(token, serializer, manager=None, authkey=None, |
| exposed=None, incref=True): |
| ''' |
| Return an auto-proxy for `token` |
| ''' |
| _Client = listener_client[serializer][1] |
| |
| if exposed is None: |
| conn = _Client(token.address, authkey=authkey) |
| try: |
| exposed = dispatch(conn, None, 'get_methods', (token,)) |
| finally: |
| conn.close() |
| |
| if authkey is None and manager is not None: |
| authkey = manager._authkey |
| if authkey is None: |
| authkey = current_process().authkey |
| |
| ProxyType = MakeProxyType('AutoProxy[%s]' % token.typeid, exposed) |
| proxy = ProxyType(token, serializer, manager=manager, authkey=authkey, |
| incref=incref) |
| proxy._isauto = True |
| return proxy |
| |
| # |
| # Types/callables which we will register with SyncManager |
| # |
| |
| class Namespace(object): |
| def __init__(self, **kwds): |
| self.__dict__.update(kwds) |
| def __repr__(self): |
| items = list(self.__dict__.items()) |
| temp = [] |
| for name, value in items: |
| if not name.startswith('_'): |
| temp.append('%s=%r' % (name, value)) |
| temp.sort() |
| return 'Namespace(%s)' % str.join(', ', temp) |
| |
| class Value(object): |
| def __init__(self, typecode, value, lock=True): |
| self._typecode = typecode |
| self._value = value |
| def get(self): |
| return self._value |
| def set(self, value): |
| self._value = value |
| def __repr__(self): |
| return '%s(%r, %r)'%(type(self).__name__, self._typecode, self._value) |
| value = property(get, set) |
| |
| def Array(typecode, sequence, lock=True): |
| return array.array(typecode, sequence) |
| |
| # |
| # Proxy types used by SyncManager |
| # |
| |
| class IteratorProxy(BaseProxy): |
| _exposed_ = ('__next__', 'send', 'throw', 'close') |
| def __iter__(self): |
| return self |
| def __next__(self, *args): |
| return self._callmethod('__next__', args) |
| def send(self, *args): |
| return self._callmethod('send', args) |
| def throw(self, *args): |
| return self._callmethod('throw', args) |
| def close(self, *args): |
| return self._callmethod('close', args) |
| |
| |
| class AcquirerProxy(BaseProxy): |
| _exposed_ = ('acquire', 'release') |
| def acquire(self, blocking=True): |
| return self._callmethod('acquire', (blocking,)) |
| def release(self): |
| return self._callmethod('release') |
| def __enter__(self): |
| return self._callmethod('acquire') |
| def __exit__(self, exc_type, exc_val, exc_tb): |
| return self._callmethod('release') |
| |
| |
| class ConditionProxy(AcquirerProxy): |
| _exposed_ = ('acquire', 'release', 'wait', 'notify', 'notify_all') |
| def wait(self, timeout=None): |
| return self._callmethod('wait', (timeout,)) |
| def notify(self): |
| return self._callmethod('notify') |
| def notify_all(self): |
| return self._callmethod('notify_all') |
| 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 |
| |
| |
| class EventProxy(BaseProxy): |
| _exposed_ = ('is_set', 'set', 'clear', 'wait') |
| def is_set(self): |
| return self._callmethod('is_set') |
| def set(self): |
| return self._callmethod('set') |
| def clear(self): |
| return self._callmethod('clear') |
| def wait(self, timeout=None): |
| return self._callmethod('wait', (timeout,)) |
| |
| class NamespaceProxy(BaseProxy): |
| _exposed_ = ('__getattribute__', '__setattr__', '__delattr__') |
| def __getattr__(self, key): |
| if key[0] == '_': |
| return object.__getattribute__(self, key) |
| callmethod = object.__getattribute__(self, '_callmethod') |
| return callmethod('__getattribute__', (key,)) |
| def __setattr__(self, key, value): |
| if key[0] == '_': |
| return object.__setattr__(self, key, value) |
| callmethod = object.__getattribute__(self, '_callmethod') |
| return callmethod('__setattr__', (key, value)) |
| def __delattr__(self, key): |
| if key[0] == '_': |
| return object.__delattr__(self, key) |
| callmethod = object.__getattribute__(self, '_callmethod') |
| return callmethod('__delattr__', (key,)) |
| |
| |
| class ValueProxy(BaseProxy): |
| _exposed_ = ('get', 'set') |
| def get(self): |
| return self._callmethod('get') |
| def set(self, value): |
| return self._callmethod('set', (value,)) |
| value = property(get, set) |
| |
| |
| BaseListProxy = MakeProxyType('BaseListProxy', ( |
| '__add__', '__contains__', '__delitem__', '__delslice__', |
| '__getitem__', '__getslice__', '__len__', '__mul__', |
| '__reversed__', '__rmul__', '__setitem__', '__setslice__', |
| 'append', 'count', 'extend', 'index', 'insert', 'pop', 'remove', |
| 'reverse', 'sort', '__imul__' |
| )) # XXX __getslice__ and __setslice__ unneeded in Py3.0 |
| class ListProxy(BaseListProxy): |
| def __iadd__(self, value): |
| self._callmethod('extend', (value,)) |
| return self |
| def __imul__(self, value): |
| self._callmethod('__imul__', (value,)) |
| return self |
| |
| |
| DictProxy = MakeProxyType('DictProxy', ( |
| '__contains__', '__delitem__', '__getitem__', '__len__', |
| '__setitem__', 'clear', 'copy', 'get', 'has_key', 'items', |
| 'keys', 'pop', 'popitem', 'setdefault', 'update', 'values' |
| )) |
| |
| |
| ArrayProxy = MakeProxyType('ArrayProxy', ( |
| '__len__', '__getitem__', '__setitem__', '__getslice__', '__setslice__' |
| )) # XXX __getslice__ and __setslice__ unneeded in Py3.0 |
| |
| |
| PoolProxy = MakeProxyType('PoolProxy', ( |
| 'apply', 'apply_async', 'close', 'imap', 'imap_unordered', 'join', |
| 'map', 'map_async', 'starmap', 'starmap_async', 'terminate' |
| )) |
| PoolProxy._method_to_typeid_ = { |
| 'apply_async': 'AsyncResult', |
| 'map_async': 'AsyncResult', |
| 'starmap_async': 'AsyncResult', |
| 'imap': 'Iterator', |
| 'imap_unordered': 'Iterator' |
| } |
| |
| # |
| # Definition of SyncManager |
| # |
| |
| class SyncManager(BaseManager): |
| ''' |
| Subclass of `BaseManager` which supports a number of shared object types. |
| |
| The types registered are those intended for the synchronization |
| of threads, plus `dict`, `list` and `Namespace`. |
| |
| The `multiprocessing.Manager()` function creates started instances of |
| this class. |
| ''' |
| |
| SyncManager.register('Queue', queue.Queue) |
| SyncManager.register('JoinableQueue', queue.Queue) |
| SyncManager.register('Event', threading.Event, EventProxy) |
| SyncManager.register('Lock', threading.Lock, AcquirerProxy) |
| SyncManager.register('RLock', threading.RLock, AcquirerProxy) |
| SyncManager.register('Semaphore', threading.Semaphore, AcquirerProxy) |
| SyncManager.register('BoundedSemaphore', threading.BoundedSemaphore, |
| AcquirerProxy) |
| SyncManager.register('Condition', threading.Condition, ConditionProxy) |
| SyncManager.register('Pool', Pool, PoolProxy) |
| SyncManager.register('list', list, ListProxy) |
| SyncManager.register('dict', dict, DictProxy) |
| SyncManager.register('Value', Value, ValueProxy) |
| SyncManager.register('Array', Array, ArrayProxy) |
| SyncManager.register('Namespace', Namespace, NamespaceProxy) |
| |
| # types returned by methods of PoolProxy |
| SyncManager.register('Iterator', proxytype=IteratorProxy, create_method=False) |
| SyncManager.register('AsyncResult', create_method=False) |