383ea0326ae103b5d5e0b62ed9c3cb18510c5b9e,distributed/comm/tcp.py,TCP,write,#TCP#Any#Any#Any#,229

Before Change

        try:
            nframes = len(frames)
            lengths = [nbytes(frame) for frame in frames]
            length_bytes = struct.pack(f"Q{nframes}Q", nframes, *lengths)

            frames = [length_bytes, *frames]
            lengths = [len(length_bytes), *lengths]

            if sum(lengths) < 2 ** 17:  // 128kiB
                // small enough, send in one go

After Change

        if frames_nbytes < 2 ** 17:  // 128kiB
            // small enough, send in one go
            frames = [b"".join(frames)]

        try:
            // trick to enque all frames for writing beforehand
            for each_frame in frames:
                each_frame_nbytes = nbytes(each_frame)
                if each_frame_nbytes:
                    stream._write_buffer.append(each_frame)
                    stream._total_write_index += each_frame_nbytes

            // start writing frames
            stream.write(b"")
        except StreamClosedError as e:
            self.stream = None
            self._closed = True

Instances