pd: handle get members with error (tikv#452)

pingyu · Jun 2, 2024 · 54fd720 · 54fd720
1 parent 0db723f
commit 54fd720
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/pd/cluster.rs b/src/pd/cluster.rs
@@ -234,6 +234,19 @@ impl Connection {
             .get_members(pdpb::GetMembersRequest::default())
             .await?
             .into_inner();
+        if let Some(err) = resp
+            .header
+            .as_ref()
+            .and_then(|header| header.error.as_ref())
+        {
+            return Err(internal_err!("failed to get PD members, err {:?}", err));
+        }
+        if resp.leader.is_none() {
+            return Err(internal_err!(
+                "unexpected no PD leader in get member resp: {:?}",
+                resp
+            ));
+        }
         Ok((client, keyspace_client, resp))
     }