ENH: Refine connection exception handling

python/xoscar/backends/core.py

@@ -74,15 +74,22 @@ async def _listen(self, client: Client):
             try:
                 try:
                     message: _MessageBase = await client.recv()
-                except (EOFError, ConnectionError, BrokenPipeError):
+                except (
+                    EOFError,
+                    ConnectionError,
+                    BrokenPipeError,
+                    AssertionError,
+                ) as e:
+                    # AssertionError is from get_header
                     # remote server closed, close client and raise ServerClosed
+                    logger.debug(f"{client.dest_address} close due to {e}")
                     try:
                         await client.close()
                     except (ConnectionError, BrokenPipeError):
                         # close failed, ignore it
                         pass
                     raise ServerClosed(
-                        f"Remote server {client.dest_address} closed"
+                        f"Remote server {client.dest_address} closed: {e}"
                     ) from None
                 future = self._client_to_message_futures[client].pop(message.message_id)
                 if not future.done():

python/xoscar/backends/pool.py

@@ -554,7 +554,8 @@ async def on_new_channel(self, channel: Channel):
         while not self._stopped.is_set():
             try:
                 message = await channel.recv()
-            except EOFError:
+            except Exception as e:
+                logger.debug(f"pool: close connection due to {e}")
                 # no data to read, check channel
                 try:
                     await channel.close()