From 31801a436cf0c33c15c376bc2c3d41945518b525 Mon Sep 17 00:00:00 2001 From: Javier Cuevas Date: Thu, 23 May 2024 17:06:04 +0200 Subject: [PATCH] Add periodic connection refresh --- mautrix_telegram/abstract_user.py | 14 ++++++++++++++ mautrix_telegram/example-config.yaml | 2 ++ 2 files changed, 16 insertions(+) diff --git a/mautrix_telegram/abstract_user.py b/mautrix_telegram/abstract_user.py index 2eb96260..ef1df456 100644 --- a/mautrix_telegram/abstract_user.py +++ b/mautrix_telegram/abstract_user.py @@ -241,6 +241,20 @@ class AbstractUser(ABC): use_ipv6=self.config["telegram.connection.use_ipv6"], ) self.client.add_event_handler(self._update_catch) + self._schedule_reconnect() + + def _schedule_reconnect(self) -> None: + reconnect_interval = self.config["telegram.force_refresh_interval_seconds"] + if not reconnect_interval or reconnect_interval == 0: + return + refresh_time = time.time() + reconnect_interval + self.log.info("Scheduling forced reconnect in %d seconds. Connection will be refreshed at %s", reconnect_interval, time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(refresh_time))) + self.loop.call_later(reconnect_interval, lambda: asyncio.create_task(self._reconnect())) + + async def _reconnect(self) -> None: + self.log.info("Reconnecting to Telegram...") + await self.stop() + await self.start() @abstractmethod async def on_signed_out(self, err: UnauthorizedError | AuthKeyError) -> None: diff --git a/mautrix_telegram/example-config.yaml b/mautrix_telegram/example-config.yaml index b23083e4..75da8958 100644 --- a/mautrix_telegram/example-config.yaml +++ b/mautrix_telegram/example-config.yaml @@ -581,6 +581,8 @@ telegram: # Should incoming updates be handled sequentially to make sure order is preserved on Matrix? sequential_updates: true exit_on_update_error: false + # Interval to force refresh the connection (full reconnect), default is 1 day. Set 0 to disable force refreshes. + force_refresh_interval_seconds: 10 # Telethon connection options. connection: