Capture connection errors

kyuupichan · Mar 6, 2021 · 22cd5d8 · 22cd5d8
1 parent ac6ccbf
commit 22cd5d8
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
@@ -440,6 +440,8 @@ async def _process_messages(self, recv_message):
             except MemoryError as e:
                 self.logger.warning(f'{e!r}')
                 continue
+            except ConnectionError:
+                return
 
             self.last_recv = time.time()
             self.recv_count += 1