From cd21bbba83cd5a1642c2b3a4f3fb17104515f543 Mon Sep 17 00:00:00 2001 From: Saathwik Dasari Date: Tue, 30 Dec 2025 16:48:55 +0530 Subject: [PATCH 1/2] fix: replace panic unwrap with error check in connect_leader Signed-off-by: Saathwik Dasari --- src/pd/cluster.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/pd/cluster.rs b/src/pd/cluster.rs index 8c8f0d7c..b77d7c1e 100644 --- a/src/pd/cluster.rs +++ b/src/pd/cluster.rs @@ -319,7 +319,9 @@ impl Connection { // Then try to connect the PD cluster leader. if let Some(resp) = resp { - let leader = resp.leader.as_ref().unwrap(); + let leader = resp.leader.as_ref() + .ok_or_else(|| internal_err!("no leader found in GetMembersResponse"))?; + for ep in &leader.client_urls { if let Ok((client, keyspace_client, members)) = self.try_connect(ep.as_str(), cluster_id, timeout).await From 85ebfa79a8e7235e399ff27a7bb98ba5c56bf214 Mon Sep 17 00:00:00 2001 From: Saathwik Dasari Date: Fri, 2 Jan 2026 16:33:32 +0530 Subject: [PATCH 2/2] style: apply cargo fmt Signed-off-by: Saathwik Dasari --- src/pd/cluster.rs | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/pd/cluster.rs b/src/pd/cluster.rs index b77d7c1e..ef6cce2b 100644 --- a/src/pd/cluster.rs +++ b/src/pd/cluster.rs @@ -319,9 +319,11 @@ impl Connection { // Then try to connect the PD cluster leader. if let Some(resp) = resp { - let leader = resp.leader.as_ref() - .ok_or_else(|| internal_err!("no leader found in GetMembersResponse"))?; - + let leader = resp + .leader + .as_ref() + .ok_or_else(|| internal_err!("no leader found in GetMembersResponse"))?; + for ep in &leader.client_urls { if let Ok((client, keyspace_client, members)) = self.try_connect(ep.as_str(), cluster_id, timeout).await