proxmox-backup/tests/worker-task-abort.rs
Dietmar Maurer b9700a9fe5 move worker_task.rs into proxmox-rest-server crate
Also moved pbs-datastore/src/task.rs to pbs-tools, which now depends on 'log'.
2021-09-24 10:28:17 +02:00

112 lines
3.0 KiB
Rust

use anyhow::{bail, Error};
extern crate proxmox_backup;
extern crate tokio;
extern crate nix;
use proxmox::try_block;
use proxmox::tools::fs::CreateOptions;
use pbs_api_types::{Authid, UPID};
use proxmox_rest_server::{flog, CommandoSocket, WorkerTask};
fn garbage_collection(worker: &WorkerTask) -> Result<(), Error> {
worker.log("start garbage collection");
for i in 0..50 {
worker.fail_on_abort()?;
flog!(worker, "progress {}", i);
std::thread::sleep(std::time::Duration::from_millis(10));
}
worker.log("end garbage collection");
Ok(())
}
#[test]
#[ignore]
fn worker_task_abort() -> Result<(), Error> {
let uid = nix::unistd::Uid::current();
let gid = nix::unistd::Gid::current();
let file_opts = CreateOptions::new().owner(uid).group(gid);
proxmox_rest_server::init_worker_tasks("./target/tasklogtestdir".into(), file_opts.clone())?;
use std::sync::{Arc, Mutex};
let errmsg: Arc<Mutex<Option<String>>> = Arc::new(Mutex::new(None));
let errmsg1 = errmsg.clone();
let rt = tokio::runtime::Runtime::new().unwrap();
rt.block_on(async move {
let mut commando_sock = CommandoSocket::new(
proxmox_rest_server::our_ctrl_sock(), nix::unistd::Gid::current());
let init_result: Result<(), Error> = try_block!({
proxmox_rest_server::register_task_control_commands(&mut commando_sock)?;
proxmox_rest_server::server_state_init()?;
Ok(())
});
if let Err(err) = init_result {
eprintln!("unable to start daemon - {}", err);
return;
}
if let Err(err) = commando_sock.spawn() {
eprintln!("unable to spawn command socket - {}", err);
return;
}
let errmsg = errmsg1.clone();
let res = WorkerTask::new_thread(
"garbage_collection",
None,
Authid::root_auth_id().to_string(),
true,
move |worker| {
println!("WORKER {}", worker);
let result = garbage_collection(&worker);
proxmox_rest_server::request_shutdown();
if let Err(err) = result {
println!("got expected error: {}", err);
} else {
let mut data = errmsg.lock().unwrap();
*data = Some(String::from("thread finished - seems abort did not work as expected"));
}
Ok(())
},
);
match res {
Err(err) => {
println!("unable to start worker - {}", err);
}
Ok(wid) => {
println!("WORKER: {}", wid);
proxmox_rest_server::abort_worker_async(wid.parse::<UPID>().unwrap());
proxmox_rest_server::wait_for_local_worker(&wid).await.unwrap();
}
}
});
let data = errmsg.lock().unwrap();
match *data {
Some(ref err) => bail!("Error: {}", err),
None => {},
}
Ok(())
}