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

Add timeout on recv loop

Browse source
This commit is contained in:
Dan 2018-06-17 18:33:23 +02:00
parent 39b66b51d6
commit 2b0746a140
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pyrogram/session/session.py
View file

@ -334,7 +334,10 @@ class Session:
log.info("RecvTask started") log.info("RecvTask started")
while True: while True:
packet = await self.connection.recv() try:
packet = await asyncio.wait_for(self.connection.recv(), self.connection.TIMEOUT)
except asyncio.TimeoutError:
packet = None
if packet is None or len(packet) == 4: if packet is None or len(packet) == 4:
if packet: if packet: