TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-23 23:34:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't raise write() and close() exceptions

Browse Source
This commit is contained in:
Dan 2022-12-26 16:23:57 +01:00
parent 7e5d593544
commit 9ee1807e42
1 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pyrogram/connection/transport/tcp/tcp.py
View File

@ -93,19 +93,13 @@ class TCP:
await self.send_task
try:
self.writer.close()
except AttributeError:
try:
self.socket.shutdown(socket.SHUT_RDWR)
except OSError:
pass
finally:
# A tiny sleep placed here helps avoiding .recv(n) hanging until the timeout.
# This is a workaround that seems to fix the occasional delayed stop of a client.
time.sleep(0.001)
self.socket.close()
if self.writer is not None:
self.writer.close()
await asyncio.wait_for(self.writer.wait_closed(), TCP.TIMEOUT)
except Exception as e:
log.info("Close exception: %s %s", type(e).__name__, e)
async def send(self, data: Optional[bytes]):
async def send(self, data: bytes):
await self.send_queue.put(data)
async def send_worker(self):
@ -115,8 +109,11 @@ class TCP:
if data is None:
break
self.writer.write(data)
await self.writer.drain()
try:
self.writer.write(data)
await self.writer.drain()
except Exception as e:
log.info("Send exception: %s %s", type(e).__name__, e)
async def recv(self, length: int = 0):
data = b""