src/bin/proxmox-backup-client.rs: implement image/block-device upload
This commit is contained in:
parent
eb1804c567
commit
6af905c18f
@ -28,7 +28,7 @@ use futures::*;
|
|||||||
use tokio::sync::mpsc;
|
use tokio::sync::mpsc;
|
||||||
|
|
||||||
lazy_static! {
|
lazy_static! {
|
||||||
static ref BACKUPSPEC_REGEX: Regex = Regex::new(r"^([a-zA-Z0-9_-]+\.(?:pxar|raw)):(.+)$").unwrap();
|
static ref BACKUPSPEC_REGEX: Regex = Regex::new(r"^([a-zA-Z0-9_-]+\.(?:pxar|img)):(.+)$").unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -139,6 +139,33 @@ fn backup_directory<P: AsRef<Path>>(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn backup_image<P: AsRef<Path>>(
|
||||||
|
client: &BackupClient,
|
||||||
|
image_path: P,
|
||||||
|
archive_name: &str,
|
||||||
|
image_size: u64,
|
||||||
|
chunk_size: Option<u64>,
|
||||||
|
verbose: bool,
|
||||||
|
) -> Result<(), Error> {
|
||||||
|
|
||||||
|
if let Some(_size) = chunk_size {
|
||||||
|
unimplemented!();
|
||||||
|
}
|
||||||
|
|
||||||
|
let path = image_path.as_ref().to_owned();
|
||||||
|
|
||||||
|
let file = tokio::fs::File::open(path).wait()?;
|
||||||
|
|
||||||
|
let stream = tokio::codec::FramedRead::new(file, tokio::codec::BytesCodec::new())
|
||||||
|
.map_err(Error::from);
|
||||||
|
|
||||||
|
let stream = FixedChunkStream::new(stream, 4*1024*1024);
|
||||||
|
|
||||||
|
client.upload_stream(archive_name, stream, "fixed", Some(image_size)).wait()?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
/****
|
/****
|
||||||
fn backup_image(datastore: &DataStore, file: &std::fs::File, size: usize, target: &str, chunk_size: usize) -> Result<(), Error> {
|
fn backup_image(datastore: &DataStore, file: &std::fs::File, size: usize, target: &str, chunk_size: usize) -> Result<(), Error> {
|
||||||
|
|
||||||
@ -408,6 +435,8 @@ fn create_backup(
|
|||||||
|
|
||||||
let mut upload_list = vec![];
|
let mut upload_list = vec![];
|
||||||
|
|
||||||
|
enum BackupType { PXAR, IMAGE };
|
||||||
|
|
||||||
for backupspec in backupspec_list {
|
for backupspec in backupspec_list {
|
||||||
let (target, filename) = parse_backupspec(backupspec.as_str().unwrap())?;
|
let (target, filename) = parse_backupspec(backupspec.as_str().unwrap())?;
|
||||||
|
|
||||||
@ -421,7 +450,7 @@ fn create_backup(
|
|||||||
|
|
||||||
if file_type.is_dir() {
|
if file_type.is_dir() {
|
||||||
|
|
||||||
upload_list.push((filename.to_owned(), target.to_owned()));
|
upload_list.push((BackupType::PXAR, filename.to_owned(), target.to_owned(), 0));
|
||||||
|
|
||||||
} else if file_type.is_file() || file_type.is_block_device() {
|
} else if file_type.is_file() || file_type.is_block_device() {
|
||||||
|
|
||||||
@ -429,12 +458,7 @@ fn create_backup(
|
|||||||
|
|
||||||
if size == 0 { bail!("got zero-sized file '{}'", filename); }
|
if size == 0 { bail!("got zero-sized file '{}'", filename); }
|
||||||
|
|
||||||
panic!("implement me");
|
upload_list.push((BackupType::IMAGE, filename.to_owned(), target.to_owned(), size));
|
||||||
|
|
||||||
//backup_image(&datastore, &file, size, &target, chunk_size)?;
|
|
||||||
|
|
||||||
// let idx = datastore.open_image_reader(target)?;
|
|
||||||
// idx.print_info();
|
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
bail!("unsupported file type (expected a directory, file or block device)");
|
bail!("unsupported file type (expected a directory, file or block device)");
|
||||||
@ -452,9 +476,17 @@ fn create_backup(
|
|||||||
|
|
||||||
let client = client.start_backup(repo.store(), "host", &backup_id, verbose).wait()?;
|
let client = client.start_backup(repo.store(), "host", &backup_id, verbose).wait()?;
|
||||||
|
|
||||||
for (filename, target) in upload_list {
|
for (backup_type, filename, target, size) in upload_list {
|
||||||
println!("Upload '{}' to '{:?}' as {}", filename, repo, target);
|
match backup_type {
|
||||||
backup_directory(&client, &filename, &target, chunk_size_opt, all_file_systems, verbose)?;
|
BackupType::PXAR => {
|
||||||
|
println!("Upload directory '{}' to '{:?}' as {}", filename, repo, target);
|
||||||
|
backup_directory(&client, &filename, &target, chunk_size_opt, all_file_systems, verbose)?;
|
||||||
|
}
|
||||||
|
BackupType::IMAGE => {
|
||||||
|
println!("Upload image '{}' to '{:?}' as {}", filename, repo, target);
|
||||||
|
backup_image(&client, &filename, &target, size, chunk_size_opt, verbose)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
client.finish().wait()?;
|
client.finish().wait()?;
|
||||||
|
Loading…
Reference in New Issue
Block a user