yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2025-12-29 12:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
)
self.socket.setblocking(False)
self.socket.settimeout(TCP.TIMEOUT)
self.send_queue = asyncio.Queue()
@ -86,9 +85,12 @@ 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)
if self.send_task is not None:
await self.send_task
try: