diff --git a/zenith/src/main.rs b/zenith/src/main.rs index cccc2ffc3b..745e1fee97 100644 --- a/zenith/src/main.rs +++ b/zenith/src/main.rs @@ -182,19 +182,7 @@ fn main() -> Result<()> { /// Connects to the pageserver to query this information. fn get_branch_infos(env: &local_env::LocalEnv) -> Result> { let page_server = PageServerNode::from_env(env); - let mut client = page_server.page_server_psql_client()?; - let branches_msgs = client.simple_query("pg_list")?; - - let branches_json = branches_msgs - .first() - .map(|msg| match msg { - postgres::SimpleQueryMessage::Row(row) => row.get(0), - _ => None, - }) - .flatten() - .ok_or_else(|| anyhow!("missing branches"))?; - - let branch_infos: Vec = serde_json::from_str(branches_json)?; + let branch_infos: Vec = page_server.branches_list()?; let branch_infos: Result> = branch_infos .into_iter() .map(|branch_info| {