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

Ping every 60 s

Browse Source
This commit is contained in:
Jahongir Qurbonov 2024-10-23 12:09:44 +05:00
parent e5adfe203e
commit 276738dfc6
1 changed files with 20 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
client/src/telethon/_impl/mtsender/sender.py
View File

@ -173,6 +173,7 @@ class Sender:
_updates: list[Updates] _updates: list[Updates]
_requests: list[Request[object]] _requests: list[Request[object]]
_response_event: Event _response_event: Event
_next_ping: float
_read_buffer: bytearray _read_buffer: bytearray
@classmethod @classmethod
@ -203,6 +204,7 @@ class Sender:
_updates=[], _updates=[],
_requests=[], _requests=[],
_response_event=Event(), _response_event=Event(),
_next_ping=asyncio.get_running_loop().time() + PING_DELAY,
_read_buffer=bytearray(), _read_buffer=bytearray(),
) )
@ -244,7 +246,7 @@ class Sender:
self._reading = True self._reading = True
self._response_event.clear() self._response_event.clear()
await self._try_read() await self._try_read()
self._response_event.set() self._try_timeout_ping()
self._reading = False self._reading = False
else: else:
await self._response_event.wait() await self._response_event.wait()
@ -259,7 +261,7 @@ class Sender:
async with asyncio.timeout(PING_DELAY): async with asyncio.timeout(PING_DELAY):
recv_data = await self._reader.read(MAXIMUM_DATA) recv_data = await self._reader.read(MAXIMUM_DATA)
except TimeoutError: except TimeoutError:
self._on_ping_timeout() pass
else: else:
self._on_net_read(recv_data) self._on_net_read(recv_data)
@ -280,6 +282,22 @@ class Sender:
if isinstance(request.state, Serialized): if isinstance(request.state, Serialized):
request.state = Sent(request.state.msg_id, container_msg_id) request.state = Sent(request.state.msg_id, container_msg_id)
self._response_event.set()
def _try_timeout_ping(self) -> None:
current_time = asyncio.get_running_loop().time()
if current_time >= self._next_ping:
ping_id = generate_random_id()
self._enqueue_body(
bytes(
ping_delay_disconnect(
ping_id=ping_id, disconnect_delay=NO_PING_DISCONNECT
)
)
)
self._next_ping = current_time + PING_DELAY
def _on_net_read(self, read_buffer: bytes) -> None: def _on_net_read(self, read_buffer: bytes) -> None:
if not read_buffer: if not read_buffer:
raise ConnectionResetError("read 0 bytes") raise ConnectionResetError("read 0 bytes")
@ -296,16 +314,6 @@ class Sender:
del self._read_buffer[:n] del self._read_buffer[:n]
self._process_mtp_buffer() self._process_mtp_buffer()
def _on_ping_timeout(self) -> None:
ping_id = generate_random_id()
self._enqueue_body(
bytes(
ping_delay_disconnect(
ping_id=ping_id, disconnect_delay=NO_PING_DISCONNECT
)
)
)
def _process_mtp_buffer(self) -> None: def _process_mtp_buffer(self) -> None:
results = self._mtp.deserialize(self._mtp_buffer) results = self._mtp.deserialize(self._mtp_buffer)