Added the control of unknown exception handling times. If not controlled, I think it is very dangerous and may fall into an infinite loop

This commit is contained in:
super_rocky 2023-08-04 00:19:02 +08:00
parent 6ded164b85
commit caf79da177

View File

@ -51,6 +51,7 @@ class MTProtoSender:
self._loggers = loggers
self._log = loggers[__name__]
self._retries = retries
self._retry_unknown_error = retries
self._delay = delay
self._auto_reconnect = auto_reconnect
self._connect_timeout = connect_timeout
@ -519,6 +520,9 @@ class MTProtoSender:
return
except Exception as e:
self._log.exception('Unhandled error while receiving data')
self._retry_unknown_error = self._retry_unknown_error - 1
if self._retry_unknown_error < 0:
raise
self._start_reconnect(e)
return