fix(session): prevent task cancellation race condition in stop method

HitaloM · HitaloM · commit a522e653e29c · 2025-04-30T18:33:48.000-03:00
The fix properly cancels the recv_task and suppresses CancelledError when awaiting
it during session shutdown. This resolves the "read() called while another
coroutine is already waiting for incoming data" RuntimeError that occurred when
stopping sessions during reconnection attempts.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -20,7 +20,7 @@ repos:
       - id: check-yaml
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.11.6
+    rev: v0.11.7
     hooks:
       - id: ruff-format
       - id: ruff
diff --git a/hydrogram/session/session.py b/hydrogram/session/session.py
@@ -185,7 +185,9 @@ async def stop(self):
         await self.connection.close()
 
         if self.recv_task:
-            await self.recv_task
+            self.recv_task.cancel()
+            with contextlib.suppress(asyncio.CancelledError):
+                await self.recv_task
 
         if not self.is_media and callable(self.client.disconnect_handler):
             try:
diff --git a/pyproject.toml b/pyproject.toml
@@ -53,7 +53,7 @@ build-backend = "hatchling.build"
 [tool.uv]
 managed = true
 dev-dependencies = [
-    "ruff>=0.11.6",
+    "ruff>=0.11.7",
     "pytest>=7.4.3",
     "pytest-asyncio>=0.23.2",
     "pytest-cov>=4.1.0",
diff --git a/uv.lock b/uv.lock
