StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests 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
No known key found for this signature in database
GPG Key ID: EFEE139F5CCB06C2
3 changed files with 10 additions and 6 deletions
Show Stats 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
header = (stream_id << 3) | flag
header = encode_uvarint(header)
data_length = encode_uvarint(len(data))
_bytes = header + data_length + data
if data is None:
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)

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)
def close(self):
async def close(self):
"""
close stream
:return: true if successful
@ -58,7 +58,7 @@ class MuxedStream(IMuxedStream):
if self.local_closed and self.remote_closed:
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.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)
def close(self):
async def close(self):
"""
close stream
:return: true if successful
"""
self.muxed_stream.close()
await self.muxed_stream.close()
return True