383ea0326ae103b5d5e0b62ed9c3cb18510c5b9e,distributed/comm/tcp.py,TCP,read,#TCP#Any#,185

Before Change

        try:
            n_frames = await stream.read_bytes(8)
            n_frames = struct.unpack("Q", n_frames)[0]
            lengths = await stream.read_bytes(8 * n_frames)
            lengths = struct.unpack("Q" * n_frames, lengths)

            frames = []
            for length in lengths:
                frame = bytearray(length)
                if length:
                    n = await stream.read_into(frame)
                    assert n == length, (n, length)
                frames.append(frame)
        except StreamClosedError as e:
            self.stream = None
            self._closed = True
            if not shutting_down():

After Change

            raise CommClosedError

        fmt = "Q"
        fmt_size = struct.calcsize(fmt)

        try:
            frames_nbytes = await stream.read_bytes(fmt_size)
            (frames_nbytes,) = struct.unpack(fmt, frames_nbytes)

            frames = bytearray(frames_nbytes)
            n = await stream.read_into(frames)
            assert n == frames_nbytes, (n, frames_nbytes)

Instances