varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 08:00:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

make stream close async

Browse Source
This commit is contained in:
Christophe de Carvalho Pereira Martins
2018-11-18 17:22:17 +01:00
parent 01daf8ae8a
commit 96edf96e76
3 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
muxer/mplex/muxed_connection.py
View File

@ -83,8 +83,12 @@ class MuxedConn(IMuxedConn):
# << by 3, then or with flag # << by 3, then or with flag
header = (stream_id << 3) | flag header = (stream_id << 3) | flag
header = encode_uvarint(header) header = encode_uvarint(header)
data_length = encode_uvarint(len(data)) if data is None:
_bytes = header + data_length + data data_length = encode_uvarint(0)
_bytes = header + data_length
else:
data_length = encode_uvarint(len(data))
_bytes = header + data_length + data
return await self.write_to_stream(_bytes) return await self.write_to_stream(_bytes)

4
muxer/mplex/muxed_stream.py
View File

@ -49,7 +49,7 @@ class MuxedStream(IMuxedStream):
""" """
return await self.muxed_conn.send_message(self.get_flag("MESSAGE"), data, self.stream_id) return await self.muxed_conn.send_message(self.get_flag("MESSAGE"), data, self.stream_id)
def close(self): async def close(self):
""" """
close stream close stream
:return: true if successful :return: true if successful
@ -58,7 +58,7 @@ class MuxedStream(IMuxedStream):
if self.local_closed and self.remote_closed: if self.local_closed and self.remote_closed:
return True return True
self.muxed_conn.send_message(self.get_flag("CLOSE"), None, self.stream_id) await self.muxed_conn.send_message(self.get_flag("CLOSE"), None, self.stream_id)
self.muxed_conn.streams.pop(self.stream_id) self.muxed_conn.streams.pop(self.stream_id)
self.local_closed = True self.local_closed = True

4
network/stream/net_stream.py
View File

@ -34,10 +34,10 @@ class NetStream(INetStream):
""" """
return await self.muxed_stream.write(data) return await self.muxed_stream.write(data)
def close(self): async def close(self):
""" """
close stream close stream
:return: true if successful :return: true if successful
""" """
self.muxed_stream.close() await self.muxed_stream.close()
return True return True