From ebfe733914a4a0685c634b90e29ea60f4850536f Mon Sep 17 00:00:00 2001 From: David Wilson Date: Sun, 18 Mar 2018 06:23:29 +0545 Subject: [PATCH] core: tidy up Stream.on_receive() branches. --- mitogen/core.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/mitogen/core.py b/mitogen/core.py index 3a56971a..ad299dea 100644 --- a/mitogen/core.py +++ b/mitogen/core.py @@ -741,17 +741,18 @@ class Stream(BasicStream): _vv and IOLOG.debug('%r.on_receive()', self) buf = self.receive_side.read() - if buf: - if self._input_buf and self._input_buf_len < 128: - self._input_buf[0] += buf - else: - self._input_buf.append(buf) - self._input_buf_len += len(buf) - while self._receive_one(broker): - pass - else: + if not buf: return self.on_disconnect(broker) + if self._input_buf and self._input_buf_len < 128: + self._input_buf[0] += buf + else: + self._input_buf.append(buf) + + self._input_buf_len += len(buf) + while self._receive_one(broker): + pass + HEADER_FMT = '>hhhLLL' HEADER_LEN = struct.calcsize(HEADER_FMT)