diff --git a/mitogen/master.py b/mitogen/master.py index 929e9d3f..9d1f33b2 100644 --- a/mitogen/master.py +++ b/mitogen/master.py @@ -615,6 +615,13 @@ class Router(mitogen.parent.Router): child = parent parent = parent.via + def disconnect_stream(self, stream): + self.broker.defer(stream.on_disconnect, self.broker) + + def disconnect_all(self): + for stream in self._stream_by_id.values(): + self.disconnect_stream(stream) + class IdAllocator(object): def __init__(self, router):