fix

xorbitsai · Nov 25, 2024 · 659c704 · 659c704
1 parent 902f2e0
commit 659c704
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/python/xoscar/backends/core.py b/python/xoscar/backends/core.py
@@ -58,14 +58,20 @@ def _listen_client(self, client: Client):
     async def get_client_via_type(
         self, router: Router, dest_address: str, client_type: Type[Client]
     ) -> Client:
-        client = await router.get_client_via_type(
-            dest_address, client_type, from_who=self
-        )
+        try:
+            client = await router.get_client_via_type(
+                dest_address, client_type, from_who=self
+            )
+        except ConnectionError:
+            raise ServerClosed(f"Remote server {dest_address} closed")
         self._listen_client(client)
         return client
 
     async def get_client(self, router: Router, dest_address: str) -> Client:
-        client = await router.get_client(dest_address, from_who=self)
+        try:
+            client = await router.get_client(dest_address, from_who=self)
+        except ConnectionError:
+            raise ServerClosed(f"Remote server {dest_address} closed")
         self._listen_client(client)
         return client