mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-23 02:17: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

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

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