issue #140: permit mitogen.unix.connect() to accept preconstructed Broker.

Part of an effort to make resource management a little more explicit.
wip-fakessh-exit-status
David Wilson 7 years ago
parent 65df36895e
commit 83c8412474

@ -92,7 +92,7 @@ class Listener(mitogen.core.BasicStream):
sock.close() sock.close()
def connect(path): def connect(path, broker=None):
LOG.debug('unix.connect(path=%r)', path) LOG.debug('unix.connect(path=%r)', path)
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
sock.connect(path) sock.connect(path)
@ -104,7 +104,7 @@ def connect(path):
LOG.debug('unix.connect(): local ID is %r, remote is %r', LOG.debug('unix.connect(): local ID is %r, remote is %r',
mitogen.context_id, remote_id) mitogen.context_id, remote_id)
router = mitogen.master.Router() router = mitogen.master.Router(broker=broker)
stream = mitogen.core.Stream(router, remote_id) stream = mitogen.core.Stream(router, remote_id)
stream.accept(sock.fileno(), sock.fileno()) stream.accept(sock.fileno(), sock.fileno())
stream.name = 'unix_listener.%d' % (pid,) stream.name = 'unix_listener.%d' % (pid,)

Loading…
Cancel
Save