issue #249: fix ordering bug masked by previous implementation

6 years ago · 07056b0dd1
parent 4cd9e09130
commit 07056b0dd1
1 changed files with 2 additions and 2 deletions
--- a/mitogen/core.py
+++ b/mitogen/core.py
@ -805,11 +805,11 @@ class BasicStream(object):

    def on_disconnect(self, broker):
        LOG.debug('%r.on_disconnect()', self)
-        broker.stop_receive(self)
-        broker._stop_transmit(self)
        if self.receive_side:
+            broker.stop_receive(self)
            self.receive_side.close()
        if self.transmit_side:
+            broker._stop_transmit(self)
            self.transmit_side.close()
        fire(self, 'disconnect')