LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-04 04:00:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

Acquire reconnect lock outside the reconnect function

Browse Source
This commit is contained in:
Lonami Exo 2018-02-10 12:56:54 +01:00
parent d8376ee50d
commit 7da092894b
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
telethon/telegram_bare_client.py
View File

@ -248,7 +248,6 @@ class TelegramBareClient:
if new_dc is None:
# Assume we are disconnected due to some error, so connect again
try:
await self._reconnect_lock.acquire()
if self.is_connected():
__log__.info('Reconnection aborted: already connected')
return True
@ -258,8 +257,6 @@ class TelegramBareClient:
except ConnectionResetError as e:
__log__.warning('Reconnection failed due to %s', e)
return False
finally:
self._reconnect_lock.release()
else:
# Since we're reconnecting possibly due to a UserMigrateError,
# we need to first know the Data Centers we can connect to. Do
@ -589,9 +586,10 @@ class TelegramBareClient:
if need_reconnect:
__log__.info('Attempting reconnection from read loop')
need_reconnect = False
while self._user_connected and not await self._reconnect():
# Retry forever, this is instant messaging
await asyncio.sleep(0.1, loop=self._loop)
with await self._reconnect_lock:
while self._user_connected and not await self._reconnect():
# Retry forever, this is instant messaging
await asyncio.sleep(0.1, loop=self._loop)
# Telegram seems to kick us every 1024 items received
# from the network not considering things like bad salt.