Capture connection errors

kyuupichan · Mar 6, 2021 · 65b67d2 · 65b67d2
1 parent ac6ccbf
commit 65b67d2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/aiorpcx/rawsocket.py b/aiorpcx/rawsocket.py
@@ -36,7 +36,7 @@
 from aiorpcx.util import NetAddress
 
 
-class ConnectionLostError(Exception):
+class ConnectionLostError(ConnectionError):
     pass
 
 

diff --git a/aiorpcx/session.py b/aiorpcx/session.py
@@ -448,6 +448,8 @@ async def _process_messages(self, recv_message):
 
             try:
                 requests = self.connection.receive_message(message)
+            except ConnectionError:
+                return
             except ProtocolError as e:
                 self.logger.debug(str(e))
                 if e.code == JSONRPC.PARSE_ERROR: