tyler/proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

proxmox-backup-client: add rate/burst parameter to backup CLI

Browse Source 
Signed-off-by: Dietmar Maurer <dietmar@proxmox.com>
This commit is contained in:
Dietmar Maurer 2021-11-03 13:56:06 +01:00
parent 2419dc0de9
commit e4bc3e0e8d
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
proxmox-backup-client/src/main.rs
View File

@ -45,7 +45,7 @@ use pbs_client::tools::{
complete_archive_name, complete_auth_id, complete_backup_group, complete_backup_snapshot, complete_archive_name, complete_auth_id, complete_backup_group, complete_backup_snapshot,
complete_backup_source, complete_chunk_size, complete_group_or_snapshot, complete_backup_source, complete_chunk_size, complete_group_or_snapshot,
complete_img_archive_name, complete_pxar_archive_name, complete_repository, connect, complete_img_archive_name, complete_pxar_archive_name, complete_repository, connect,
extract_repository_from_value, connect_rate_limited, extract_repository_from_value,
key_source::{ key_source::{
crypto_parameters, format_key_source, get_encryption_key_password, KEYFD_SCHEMA, crypto_parameters, format_key_source, get_encryption_key_password, KEYFD_SCHEMA,
KEYFILE_SCHEMA, MASTER_PUBKEY_FD_SCHEMA, MASTER_PUBKEY_FILE_SCHEMA, KEYFILE_SCHEMA, MASTER_PUBKEY_FD_SCHEMA, MASTER_PUBKEY_FILE_SCHEMA,
@ -582,6 +582,18 @@ fn spawn_catalog_upload(
schema: CHUNK_SIZE_SCHEMA, schema: CHUNK_SIZE_SCHEMA,
optional: true, optional: true,
}, },
rate: {
type: u64,
description: "Rate limit for TBF in bytes/second.",
optional: true,
minimum: 1,
},
burst: {
type: u64,
description: "Size of the TBF bucket, in bytes.",
optional: true,
minimum: 1,
},
"exclude": { "exclude": {
type: Array, type: Array,
description: "List of paths or patterns for matching files to exclude.", description: "List of paths or patterns for matching files to exclude.",
@ -630,6 +642,9 @@ async fn create_backup(
verify_chunk_size(size)?; verify_chunk_size(size)?;
} }
let rate_limit = param["rate"].as_u64();
let bucket_size = param["burst"].as_u64();
let crypto = crypto_parameters(&param)?; let crypto = crypto_parameters(&param)?;
let backup_id = param["backup-id"].as_str().unwrap_or(&proxmox::tools::nodename()); let backup_id = param["backup-id"].as_str().unwrap_or(&proxmox::tools::nodename());
@ -724,7 +739,7 @@ async fn create_backup(
let backup_time = backup_time_opt.unwrap_or_else(epoch_i64); let backup_time = backup_time_opt.unwrap_or_else(epoch_i64);
let client = connect(&repo)?; let client = connect_rate_limited(&repo, rate_limit, bucket_size)?;
record_repository(&repo); record_repository(&repo);
println!("Starting backup: {}/{}/{}", backup_type, backup_id, BackupDir::backup_time_to_string(backup_time)?); println!("Starting backup: {}/{}/{}", backup_type, backup_id, BackupDir::backup_time_to_string(backup_time)?);