From e4bc3e0e8d0920065c973f21deec4b0b20ce3d0c Mon Sep 17 00:00:00 2001 From: Dietmar Maurer Date: Wed, 3 Nov 2021 13:56:06 +0100 Subject: [PATCH] proxmox-backup-client: add rate/burst parameter to backup CLI Signed-off-by: Dietmar Maurer --- proxmox-backup-client/src/main.rs | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/proxmox-backup-client/src/main.rs b/proxmox-backup-client/src/main.rs index cb083006..d81271d0 100644 --- a/proxmox-backup-client/src/main.rs +++ b/proxmox-backup-client/src/main.rs @@ -45,7 +45,7 @@ use pbs_client::tools::{ complete_archive_name, complete_auth_id, complete_backup_group, complete_backup_snapshot, complete_backup_source, complete_chunk_size, complete_group_or_snapshot, complete_img_archive_name, complete_pxar_archive_name, complete_repository, connect, - extract_repository_from_value, + connect_rate_limited, extract_repository_from_value, key_source::{ crypto_parameters, format_key_source, get_encryption_key_password, KEYFD_SCHEMA, KEYFILE_SCHEMA, MASTER_PUBKEY_FD_SCHEMA, MASTER_PUBKEY_FILE_SCHEMA, @@ -582,6 +582,18 @@ fn spawn_catalog_upload( schema: CHUNK_SIZE_SCHEMA, 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": { type: Array, description: "List of paths or patterns for matching files to exclude.", @@ -630,6 +642,9 @@ async fn create_backup( verify_chunk_size(size)?; } + let rate_limit = param["rate"].as_u64(); + let bucket_size = param["burst"].as_u64(); + let crypto = crypto_parameters(¶m)?; 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 client = connect(&repo)?; + let client = connect_rate_limited(&repo, rate_limit, bucket_size)?; record_repository(&repo); println!("Starting backup: {}/{}/{}", backup_type, backup_id, BackupDir::backup_time_to_string(backup_time)?);