From e44e31e55b752bb48f3c97850e31d818433f7580 Mon Sep 17 00:00:00 2001 From: Alex Haynes Date: Sun, 11 Nov 2018 17:48:31 -0500 Subject: [PATCH] read buffer and write_to_stream return --- muxer/mplex/muxed_connection.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/muxer/mplex/muxed_connection.py b/muxer/mplex/muxed_connection.py index 6c77c9b..98b4d54 100644 --- a/muxer/mplex/muxed_connection.py +++ b/muxer/mplex/muxed_connection.py @@ -33,6 +33,11 @@ class MuxedConn(IMuxedConn): """ pass + def read_buffer(self, stream_id): + data = self.buffers[stream_id] + self.buffers[stream_id] = bytearray() + return data + def open_stream(self, protocol_id, stream_id, peer_id, multi_addr): """ creates a new muxed_stream @@ -67,9 +72,9 @@ class MuxedConn(IMuxedConn): return self.write_to_stream(_bytes) async def write_to_stream(self, _bytes): - to_return = self.raw_conn.writer.write(_bytes) + self.raw_conn.writer.write(_bytes) await self.raw_conn.writer.drain() - return to_return + return len(_bytes) async def handle_incoming(self): data = bytearray()