feat(local-deployer): add per-node readiness wait for manual clusters

This commit is contained in:
andrussal 2026-02-17 10:54:36 +01:00
parent 6c7b66db02
commit 514cae9e89
2 changed files with 35 additions and 0 deletions

View File

@ -73,6 +73,11 @@ impl<E: LocalDeployerEnv> ManualCluster<E> {
pub async fn wait_network_ready(&self) -> Result<(), ReadinessError> {
self.nodes.wait_network_ready().await
}
pub async fn wait_node_ready(&self, name: &str) -> Result<(), ManualClusterError> {
self.nodes.wait_node_ready(name).await?;
Ok(())
}
}
impl<E: LocalDeployerEnv> Drop for ManualCluster<E> {

View File

@ -51,6 +51,11 @@ pub enum NodeManagerError {
#[source]
source: DynError,
},
#[error("failed readiness check: {source}")]
Readiness {
#[source]
source: ReadinessError,
},
}
pub struct NodeManager<E: LocalDeployerEnv> {
@ -191,6 +196,31 @@ impl<E: LocalDeployerEnv> NodeManager<E> {
wait_for_http_ports(&ports, E::readiness_endpoint_path()).await
}
pub async fn wait_node_ready(&self, name: &str) -> Result<(), NodeManagerError> {
let port = {
let state = self.lock_state();
let index =
*state
.indices_by_name
.get(name)
.ok_or_else(|| NodeManagerError::NodeName {
name: name.to_string(),
})?;
state
.nodes
.get(index)
.map(|node| node.endpoints().api.port())
.ok_or_else(|| NodeManagerError::NodeName {
name: name.to_string(),
})?
};
wait_for_http_ports(&[port], E::readiness_endpoint_path())
.await
.map_err(|source| NodeManagerError::Readiness { source })
}
pub async fn start_node_with(
&self,
name: &str,