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

Add update event

Browse Source
This commit is contained in:
Jahongir Qurbonov 2024-10-16 17:42:34 +05:00
parent ca97078855
commit d0e2e46c82
1 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -191,6 +191,7 @@ class Sender:
_next_ping: float _next_ping: float
_read_buffer: bytearray _read_buffer: bytearray
_step_lock: Lock _step_lock: Lock
_step_event: Event
_step_generator: AsyncGenerator[None, None] | None = None _step_generator: AsyncGenerator[None, None] | None = None
@classmethod @classmethod
@ -222,6 +223,7 @@ class Sender:
_next_ping=asyncio.get_running_loop().time() + PING_DELAY, _next_ping=asyncio.get_running_loop().time() + PING_DELAY,
_read_buffer=bytearray(), _read_buffer=bytearray(),
_step_lock=Lock(), _step_lock=Lock(),
_step_event=Event(),
) )
async def disconnect(self) -> None: async def disconnect(self) -> None:
@ -249,9 +251,7 @@ class Sender:
async def _step_until_receive(self, rx: Future[bytes]) -> bytes: async def _step_until_receive(self, rx: Future[bytes]) -> bytes:
while True: while True:
step = asyncio.create_task(self.do_step()) await self.do_step()
await asyncio.wait((step, rx), return_when=FIRST_COMPLETED)
if rx.done(): if rx.done():
return rx.result() return rx.result()
@ -262,8 +262,21 @@ class Sender:
return result return result
async def do_step(self) -> None: async def do_step(self) -> None:
if self._step_lock.locked():
await self._step_event.wait()
return
await self._do_step()
async def _do_step(self) -> None:
async with self._step_lock: async with self._step_lock:
await anext(self.step) try:
self._step_event.clear()
await anext(self.step)
except StopAsyncIteration:
pass
finally:
self._step_event.set()
@property @property
def step(self) -> AsyncGenerator[None, None]: def step(self) -> AsyncGenerator[None, None]:
@ -272,7 +285,6 @@ class Sender:
return self._step_generator return self._step_generator
async def _step(self) -> AsyncGenerator[None, None]: async def _step(self) -> AsyncGenerator[None, None]:
recv_req = asyncio.create_task(self._request_event.wait())
recv_data = asyncio.create_task(self._step_recv()) recv_data = asyncio.create_task(self._step_recv())
send_data = asyncio.create_task(self._step_send()) send_data = asyncio.create_task(self._step_send())
@ -288,8 +300,6 @@ class Sender:
yield yield
if recv_req.done():
recv_req = asyncio.create_task(self._request_event.wait())
if recv_data.done(): if recv_data.done():
recv_data = asyncio.create_task(self._step_recv()) recv_data = asyncio.create_task(self._step_recv())
if send_data.done(): if send_data.done():