mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-22 18:07:21 +00:00
Code Issues Releases Wiki Activity

Keep a timeout while connecting and set non-blocking afterwards

Browse Source 
Also fix an await to None value
This commit is contained in:
Dan 2022-12-24 17:15:10 +01:00
parent f30510ab7d
commit 1cb17152f8
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -76,7 +76,6 @@ class TCP:
else socket.AF_INET
)
self.socket.setblocking(False)
self.socket.settimeout(TCP.TIMEOUT)
self.send_queue = asyncio.Queue()
@ -86,10 +85,13 @@ class TCP:
await asyncio.get_event_loop().sock_connect(self.socket, address)
self.reader, self.writer = await asyncio.open_connection(sock=self.socket)
self.send_task = asyncio.create_task(self.send_worker())
self.socket.setblocking(False)
async def close(self):
await self.send_queue.put(None)
await self.send_task
if self.send_task is not None:
await self.send_task
try:
self.writer.close()