hivectl: batch QueueNodes polling by id set, drop remaining dag wording
This commit is contained in:
parent
309c1c91c6
commit
0c4d56a585
8 changed files with 182 additions and 125 deletions
|
|
@ -13,7 +13,7 @@ use anyhow::{Context as _, Result, bail};
|
|||
use hive_host_sock::HostRequest;
|
||||
|
||||
use crate::cli::{AgentCmd, AgentQuotaCmd};
|
||||
use crate::dag_progress::wait_for_dags;
|
||||
use crate::dag_progress::wait_for_nodes;
|
||||
use crate::util::render;
|
||||
|
||||
async fn agents_restart(socket: &Path, name: &str, no_wait: bool) -> Result<()> {
|
||||
|
|
@ -27,7 +27,7 @@ async fn agents_restart(socket: &Path, name: &str, no_wait: bool) -> Result<()>
|
|||
.with_context(|| format!("connect to daemon socket {}", socket.display()))?;
|
||||
if resp.ok {
|
||||
println!("restart queued: {name}");
|
||||
wait_for_dags(socket, resp.queued_dags.unwrap_or_default(), no_wait).await
|
||||
wait_for_nodes(socket, resp.queued_dags.unwrap_or_default(), no_wait).await
|
||||
} else {
|
||||
bail!(
|
||||
"restart {name}: {}",
|
||||
|
|
|
|||
Loading…
Reference in a new issue