Check if the client is still connected

Signed-off-by: Tao He <[email protected]>
v6d-io · Aug 30, 2023 · ca68028 · ca68028
1 parent c2d36ad
commit ca68028
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/rust/vineyard/src/client/ipc_client.rs b/rust/vineyard/src/client/ipc_client.rs
@@ -451,7 +451,9 @@ impl IPCClientManager {
         let mut clients = IPCClientManager::get_clients().lock()?;
         let socket = socket.into();
         if let Some(client) = clients.get(&socket) {
-            return Ok(client.clone());
+            if client.lock()?.connected() {
+                return Ok(client.clone());
+            }
         }
         let client = Arc::new(Mutex::new(IPCClient::connect(&socket)?));
         clients.insert(socket, client.clone());

diff --git a/rust/vineyard/src/client/rpc_client.rs b/rust/vineyard/src/client/rpc_client.rs
@@ -204,7 +204,9 @@ impl RPCClientManager {
         let mut clients: std::sync::MutexGuard<'_, _> = RPCClientManager::get_clients().lock()?;
         let endpoint: String = endpoint.into();
         if let Some(client) = clients.get(endpoint.as_str()) {
-            return Ok(client.clone());
+            if client.lock()?.connected() {
+                return Ok(client.clone());
+            }
         }
         let client = Arc::new(Mutex::new(RPCClient::connect_with_endpoint(&endpoint)?));
         clients.insert(endpoint, client.clone());