varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change async def write

Browse Source 
To return `None` instead of `int. `Writer.write` *does* write all data
in all use case.
This commit is contained in:
mhchia
2020-02-07 18:33:15 +08:00
parent 1152f9b703
commit fb53edbc04
7 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libp2p/security/noise/transport.py
View File

@ -27,8 +27,8 @@ class NoiseConnection(BaseSession):
async def read(self, n: int = None) -> bytes:
return await self.conn.read(n)
async def write(self, data: bytes) -> int:
return await self.conn.write(data)
async def write(self, data: bytes) -> None:
await self.conn.write(data)
async def close(self) -> None:
await self.conn.close()

3
libp2p/security/secio/transport.py
View File

@ -135,9 +135,8 @@ class SecureSession(BaseSession):
raise DecryptionFailedException() from e
return decrypted_msg
async def write(self, data: bytes) -> int:
async def write(self, data: bytes) -> None:
await self.write_msg(data)
return len(data)
async def write_msg(self, msg: bytes) -> None:
encrypted_data = self.local_encrypter.encrypt(msg)