Correctly rewrite socket timeouts on sending (#842)

neo4j · Oct 28, 2022 · abdbb29 · abdbb29
1 parent 3873dfd
commit abdbb29
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/neo4j/_async/io/_common.py b/neo4j/_async/io/_common.py
@@ -139,7 +139,9 @@ async def flush(self):
         if data:
             try:
                 await self.socket.sendall(data)
-            except (OSError, asyncio.CancelledError) as error:
+            except (
+                OSError, SocketDeadlineExceeded, asyncio.CancelledError
+            ) as error:
                 await AsyncUtil.callback(self.on_error, error)
                 return False
             self._clear()

diff --git a/neo4j/_sync/io/_common.py b/neo4j/_sync/io/_common.py
@@ -139,7 +139,9 @@ def flush(self):
         if data:
             try:
                 self.socket.sendall(data)
-            except (OSError, asyncio.CancelledError) as error:
+            except (
+                OSError, SocketDeadlineExceeded, asyncio.CancelledError
+            ) as error:
                 Util.callback(self.on_error, error)
                 return False
             self._clear()