diff --git a/src/bin/proxmox-backup-manager.rs b/src/bin/proxmox-backup-manager.rs index 228d7ed1..5c70d8c7 100644 --- a/src/bin/proxmox-backup-manager.rs +++ b/src/bin/proxmox-backup-manager.rs @@ -12,6 +12,7 @@ use proxmox_backup::tools; use proxmox_backup::config; use proxmox_backup::api2::{self, types::* }; use proxmox_backup::client::*; +use proxmox_backup::server::wait_for_local_worker; mod proxmox_backup_manager; use proxmox_backup_manager::*; diff --git a/src/client/remote_chunk_reader.rs b/src/client/remote_chunk_reader.rs index 78c79af2..61b6fb05 100644 --- a/src/client/remote_chunk_reader.rs +++ b/src/client/remote_chunk_reader.rs @@ -8,10 +8,10 @@ use anyhow::{bail, Error}; use pbs_datastore::{CryptConfig, CryptMode}; use pbs_datastore::data_blob::DataBlob; use pbs_datastore::read_chunk::ReadChunk; +use pbs_datastore::read_chunk::AsyncReadChunk; use pbs_runtime::block_on; use super::BackupReader; -use crate::backup::AsyncReadChunk; /// Read chunks from remote host using ``BackupReader`` #[derive(Clone)] diff --git a/src/client/task_log.rs b/src/client/task_log.rs index 6350e83a..1d1af141 100644 --- a/src/client/task_log.rs +++ b/src/client/task_log.rs @@ -10,7 +10,6 @@ use proxmox::api::cli::format_and_print_result; use pbs_tools::percent_encoding::percent_encode_component; use super::HttpClient; -use crate::server::{UPID, worker_is_active_local}; /// Display task log on console /// @@ -116,23 +115,3 @@ pub async fn view_task_result( Ok(()) } - -/// Wait for a locally spanned worker task -/// -/// Note: local workers should print logs to stdout, so there is no -/// need to fetch/display logs. We just wait for the worker to finish. -pub async fn wait_for_local_worker(upid_str: &str) -> Result<(), Error> { - - let upid: UPID = upid_str.parse()?; - - let sleep_duration = core::time::Duration::new(0, 100_000_000); - - loop { - if worker_is_active_local(&upid) { - tokio::time::sleep(sleep_duration).await; - } else { - break; - } - } - Ok(()) -} diff --git a/src/server/worker_task.rs b/src/server/worker_task.rs index 13578446..20518ece 100644 --- a/src/server/worker_task.rs +++ b/src/server/worker_task.rs @@ -804,3 +804,23 @@ impl pbs_datastore::task::TaskState for WorkerTask { } } } + +/// Wait for a locally spanned worker task +/// +/// Note: local workers should print logs to stdout, so there is no +/// need to fetch/display logs. We just wait for the worker to finish. +pub async fn wait_for_local_worker(upid_str: &str) -> Result<(), Error> { + + let upid: UPID = upid_str.parse()?; + + let sleep_duration = core::time::Duration::new(0, 100_000_000); + + loop { + if worker_is_active_local(&upid) { + tokio::time::sleep(sleep_duration).await; + } else { + break; + } + } + Ok(()) +}