Raise and handle send errors in order to immediately act upon

delivrance · delivrance · commit 1daa05a35cd2 · 2022-12-28T00:21:05.000+01:00
diff --git a/pyrogram/connection/transport/tcp/tcp.py b/pyrogram/connection/transport/tcp/tcp.py
@@ -101,6 +101,7 @@ async def send(self, data: bytes):
                     await self.writer.drain()
             except Exception as e:
                 log.warning("Send exception: %s %s", type(e).__name__, e)
+                raise OSError(e)
 
     async def recv(self, length: int = 0):
         data = b""
diff --git a/pyrogram/session/session.py b/pyrogram/session/session.py
@@ -297,7 +297,11 @@ async def send(self, data: TLObject, wait_response: bool = True, timeout: float
             self.auth_key_id
         )
 
-        await self.connection.send(payload)
+        try:
+            await self.connection.send(payload)
+        except OSError as e:
+            self.results.pop(msg_id, None)
+            raise e
 
         if wait_response:
             try:
