master: helper functions to force disconnect everything

beb16d0d · David Wilson · 9b5d0a9b · beb16d0d
Commit beb16d0d authored Feb 15, 2018 by David Wilson
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

mitogen/master.py mitogen/master.py +7 -0

No files found.
--- 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):