backup info: run rustfmt

Signed-off-by: Thomas Lamprecht <t.lamprecht@proxmox.com>
This commit is contained in:
Thomas Lamprecht 2021-03-14 19:18:35 +01:00
parent b31cdec225
commit cded320e92
1 changed files with 145 additions and 74 deletions

View File

@ -3,17 +3,29 @@ use crate::tools;
use anyhow::{bail, format_err, Error}; use anyhow::{bail, format_err, Error};
use std::os::unix::io::RawFd; use std::os::unix::io::RawFd;
use std::path::{PathBuf, Path}; use std::path::{Path, PathBuf};
use proxmox::const_regex; use proxmox::const_regex;
use super::manifest::MANIFEST_BLOB_NAME; use super::manifest::MANIFEST_BLOB_NAME;
macro_rules! BACKUP_ID_RE { () => (r"[A-Za-z0-9_][A-Za-z0-9._\-]*") } macro_rules! BACKUP_ID_RE {
macro_rules! BACKUP_TYPE_RE { () => (r"(?:host|vm|ct)") } () => {
macro_rules! BACKUP_TIME_RE { () => (r"[0-9]{4}-[0-9]{2}-[0-9]{2}T[0-9]{2}:[0-9]{2}:[0-9]{2}Z") } r"[A-Za-z0-9_][A-Za-z0-9._\-]*"
};
}
macro_rules! BACKUP_TYPE_RE {
() => {
r"(?:host|vm|ct)"
};
}
macro_rules! BACKUP_TIME_RE {
() => {
r"[0-9]{4}-[0-9]{2}-[0-9]{2}T[0-9]{2}:[0-9]{2}:[0-9]{2}Z"
};
}
const_regex!{ const_regex! {
BACKUP_FILE_REGEX = r"^.*\.([fd]idx|blob)$"; BACKUP_FILE_REGEX = r"^.*\.([fd]idx|blob)$";
BACKUP_TYPE_REGEX = concat!(r"^(", BACKUP_TYPE_RE!(), r")$"); BACKUP_TYPE_REGEX = concat!(r"^(", BACKUP_TYPE_RE!(), r")$");
@ -38,7 +50,6 @@ pub struct BackupGroup {
} }
impl std::cmp::Ord for BackupGroup { impl std::cmp::Ord for BackupGroup {
fn cmp(&self, other: &Self) -> std::cmp::Ordering { fn cmp(&self, other: &Self) -> std::cmp::Ordering {
let type_order = self.backup_type.cmp(&other.backup_type); let type_order = self.backup_type.cmp(&other.backup_type);
if type_order != std::cmp::Ordering::Equal { if type_order != std::cmp::Ordering::Equal {
@ -63,9 +74,11 @@ impl std::cmp::PartialOrd for BackupGroup {
} }
impl BackupGroup { impl BackupGroup {
pub fn new<T: Into<String>, U: Into<String>>(backup_type: T, backup_id: U) -> Self { pub fn new<T: Into<String>, U: Into<String>>(backup_type: T, backup_id: U) -> Self {
Self { backup_type: backup_type.into(), backup_id: backup_id.into() } Self {
backup_type: backup_type.into(),
backup_id: backup_id.into(),
}
} }
pub fn backup_type(&self) -> &str { pub fn backup_type(&self) -> &str {
@ -77,7 +90,6 @@ impl BackupGroup {
} }
pub fn group_path(&self) -> PathBuf { pub fn group_path(&self) -> PathBuf {
let mut relative_path = PathBuf::new(); let mut relative_path = PathBuf::new();
relative_path.push(&self.backup_type); relative_path.push(&self.backup_type);
@ -88,46 +100,65 @@ impl BackupGroup {
} }
pub fn list_backups(&self, base_path: &Path) -> Result<Vec<BackupInfo>, Error> { pub fn list_backups(&self, base_path: &Path) -> Result<Vec<BackupInfo>, Error> {
let mut list = vec![]; let mut list = vec![];
let mut path = base_path.to_owned(); let mut path = base_path.to_owned();
path.push(self.group_path()); path.push(self.group_path());
tools::scandir(libc::AT_FDCWD, &path, &BACKUP_DATE_REGEX, |l2_fd, backup_time, file_type| { tools::scandir(
if file_type != nix::dir::Type::Directory { return Ok(()); } libc::AT_FDCWD,
&path,
&BACKUP_DATE_REGEX,
|l2_fd, backup_time, file_type| {
if file_type != nix::dir::Type::Directory {
return Ok(());
}
let backup_dir = BackupDir::with_rfc3339(&self.backup_type, &self.backup_id, backup_time)?; let backup_dir =
BackupDir::with_rfc3339(&self.backup_type, &self.backup_id, backup_time)?;
let files = list_backup_files(l2_fd, backup_time)?; let files = list_backup_files(l2_fd, backup_time)?;
list.push(BackupInfo { backup_dir, files }); list.push(BackupInfo { backup_dir, files });
Ok(()) Ok(())
})?; },
)?;
Ok(list) Ok(list)
} }
pub fn last_successful_backup(&self, base_path: &Path) -> Result<Option<i64>, Error> { pub fn last_successful_backup(&self, base_path: &Path) -> Result<Option<i64>, Error> {
let mut last = None; let mut last = None;
let mut path = base_path.to_owned(); let mut path = base_path.to_owned();
path.push(self.group_path()); path.push(self.group_path());
tools::scandir(libc::AT_FDCWD, &path, &BACKUP_DATE_REGEX, |l2_fd, backup_time, file_type| { tools::scandir(
if file_type != nix::dir::Type::Directory { return Ok(()); } libc::AT_FDCWD,
&path,
&BACKUP_DATE_REGEX,
|l2_fd, backup_time, file_type| {
if file_type != nix::dir::Type::Directory {
return Ok(());
}
let mut manifest_path = PathBuf::from(backup_time); let mut manifest_path = PathBuf::from(backup_time);
manifest_path.push(MANIFEST_BLOB_NAME); manifest_path.push(MANIFEST_BLOB_NAME);
use nix::fcntl::{openat, OFlag}; use nix::fcntl::{openat, OFlag};
match openat(l2_fd, &manifest_path, OFlag::O_RDONLY, nix::sys::stat::Mode::empty()) { match openat(
l2_fd,
&manifest_path,
OFlag::O_RDONLY,
nix::sys::stat::Mode::empty(),
) {
Ok(rawfd) => { Ok(rawfd) => {
/* manifest exists --> assume backup was successful */ /* manifest exists --> assume backup was successful */
/* close else this leaks! */ /* close else this leaks! */
nix::unistd::close(rawfd)?; nix::unistd::close(rawfd)?;
}, }
Err(nix::Error::Sys(nix::errno::Errno::ENOENT)) => { return Ok(()); } Err(nix::Error::Sys(nix::errno::Errno::ENOENT)) => {
return Ok(());
}
Err(err) => { Err(err) => {
bail!("last_successful_backup: unexpected error - {}", err); bail!("last_successful_backup: unexpected error - {}", err);
} }
@ -135,13 +166,16 @@ impl BackupGroup {
let timestamp = proxmox::tools::time::parse_rfc3339(backup_time)?; let timestamp = proxmox::tools::time::parse_rfc3339(backup_time)?;
if let Some(last_timestamp) = last { if let Some(last_timestamp) = last {
if timestamp > last_timestamp { last = Some(timestamp); } if timestamp > last_timestamp {
last = Some(timestamp);
}
} else { } else {
last = Some(timestamp); last = Some(timestamp);
} }
Ok(()) Ok(())
})?; },
)?;
Ok(last) Ok(last)
} }
@ -162,7 +196,8 @@ impl std::str::FromStr for BackupGroup {
/// ///
/// This parses strings like `vm/100". /// This parses strings like `vm/100".
fn from_str(path: &str) -> Result<Self, Self::Err> { fn from_str(path: &str) -> Result<Self, Self::Err> {
let cap = GROUP_PATH_REGEX.captures(path) let cap = GROUP_PATH_REGEX
.captures(path)
.ok_or_else(|| format_err!("unable to parse backup group path '{}'", path))?; .ok_or_else(|| format_err!("unable to parse backup group path '{}'", path))?;
Ok(Self { Ok(Self {
@ -182,11 +217,10 @@ pub struct BackupDir {
/// Backup timestamp /// Backup timestamp
backup_time: i64, backup_time: i64,
// backup_time as rfc3339 // backup_time as rfc3339
backup_time_string: String backup_time_string: String,
} }
impl BackupDir { impl BackupDir {
pub fn new<T, U>(backup_type: T, backup_id: U, backup_time: i64) -> Result<Self, Error> pub fn new<T, U>(backup_type: T, backup_id: U, backup_time: i64) -> Result<Self, Error>
where where
T: Into<String>, T: Into<String>,
@ -196,7 +230,11 @@ impl BackupDir {
BackupDir::with_group(group, backup_time) BackupDir::with_group(group, backup_time)
} }
pub fn with_rfc3339<T,U,V>(backup_type: T, backup_id: U, backup_time_string: V) -> Result<Self, Error> pub fn with_rfc3339<T, U, V>(
backup_type: T,
backup_id: U,
backup_time_string: V,
) -> Result<Self, Error>
where where
T: Into<String>, T: Into<String>,
U: Into<String>, U: Into<String>,
@ -205,12 +243,20 @@ impl BackupDir {
let backup_time_string = backup_time_string.into(); let backup_time_string = backup_time_string.into();
let backup_time = proxmox::tools::time::parse_rfc3339(&backup_time_string)?; let backup_time = proxmox::tools::time::parse_rfc3339(&backup_time_string)?;
let group = BackupGroup::new(backup_type.into(), backup_id.into()); let group = BackupGroup::new(backup_type.into(), backup_id.into());
Ok(Self { group, backup_time, backup_time_string }) Ok(Self {
group,
backup_time,
backup_time_string,
})
} }
pub fn with_group(group: BackupGroup, backup_time: i64) -> Result<Self, Error> { pub fn with_group(group: BackupGroup, backup_time: i64) -> Result<Self, Error> {
let backup_time_string = Self::backup_time_to_string(backup_time)?; let backup_time_string = Self::backup_time_to_string(backup_time)?;
Ok(Self { group, backup_time, backup_time_string }) Ok(Self {
group,
backup_time,
backup_time_string,
})
} }
pub fn group(&self) -> &BackupGroup { pub fn group(&self) -> &BackupGroup {
@ -226,7 +272,6 @@ impl BackupDir {
} }
pub fn relative_path(&self) -> PathBuf { pub fn relative_path(&self) -> PathBuf {
let mut relative_path = self.group.group_path(); let mut relative_path = self.group.group_path();
relative_path.push(self.backup_time_string.clone()); relative_path.push(self.backup_time_string.clone());
@ -247,7 +292,8 @@ impl std::str::FromStr for BackupDir {
/// ///
/// This parses strings like `host/elsa/2020-06-15T05:18:33Z". /// This parses strings like `host/elsa/2020-06-15T05:18:33Z".
fn from_str(path: &str) -> Result<Self, Self::Err> { fn from_str(path: &str) -> Result<Self, Self::Err> {
let cap = SNAPSHOT_PATH_REGEX.captures(path) let cap = SNAPSHOT_PATH_REGEX
.captures(path)
.ok_or_else(|| format_err!("unable to parse backup snapshot path '{}'", path))?; .ok_or_else(|| format_err!("unable to parse backup snapshot path '{}'", path))?;
BackupDir::with_rfc3339( BackupDir::with_rfc3339(
@ -276,7 +322,6 @@ pub struct BackupInfo {
} }
impl BackupInfo { impl BackupInfo {
pub fn new(base_path: &Path, backup_dir: BackupDir) -> Result<BackupInfo, Error> { pub fn new(base_path: &Path, backup_dir: BackupDir) -> Result<BackupInfo, Error> {
let mut path = base_path.to_owned(); let mut path = base_path.to_owned();
path.push(backup_dir.relative_path()); path.push(backup_dir.relative_path());
@ -287,19 +332,24 @@ impl BackupInfo {
} }
/// Finds the latest backup inside a backup group /// Finds the latest backup inside a backup group
pub fn last_backup(base_path: &Path, group: &BackupGroup, only_finished: bool) pub fn last_backup(
-> Result<Option<BackupInfo>, Error> base_path: &Path,
{ group: &BackupGroup,
only_finished: bool,
) -> Result<Option<BackupInfo>, Error> {
let backups = group.list_backups(base_path)?; let backups = group.list_backups(base_path)?;
Ok(backups.into_iter() Ok(backups
.into_iter()
.filter(|item| !only_finished || item.is_finished()) .filter(|item| !only_finished || item.is_finished())
.max_by_key(|item| item.backup_dir.backup_time())) .max_by_key(|item| item.backup_dir.backup_time()))
} }
pub fn sort_list(list: &mut Vec<BackupInfo>, ascendending: bool) { pub fn sort_list(list: &mut Vec<BackupInfo>, ascendending: bool) {
if ascendending { // oldest first if ascendending {
// oldest first
list.sort_unstable_by(|a, b| a.backup_dir.backup_time.cmp(&b.backup_dir.backup_time)); list.sort_unstable_by(|a, b| a.backup_dir.backup_time.cmp(&b.backup_dir.backup_time));
} else { // newest first } else {
// newest first
list.sort_unstable_by(|a, b| b.backup_dir.backup_time.cmp(&a.backup_dir.backup_time)); list.sort_unstable_by(|a, b| b.backup_dir.backup_time.cmp(&a.backup_dir.backup_time));
} }
} }
@ -316,31 +366,52 @@ impl BackupInfo {
pub fn list_backup_groups(base_path: &Path) -> Result<Vec<BackupGroup>, Error> { pub fn list_backup_groups(base_path: &Path) -> Result<Vec<BackupGroup>, Error> {
let mut list = Vec::new(); let mut list = Vec::new();
tools::scandir(libc::AT_FDCWD, base_path, &BACKUP_TYPE_REGEX, |l0_fd, backup_type, file_type| { tools::scandir(
if file_type != nix::dir::Type::Directory { return Ok(()); } libc::AT_FDCWD,
tools::scandir(l0_fd, backup_type, &BACKUP_ID_REGEX, |_, backup_id, file_type| { base_path,
if file_type != nix::dir::Type::Directory { return Ok(()); } &BACKUP_TYPE_REGEX,
|l0_fd, backup_type, file_type| {
if file_type != nix::dir::Type::Directory {
return Ok(());
}
tools::scandir(
l0_fd,
backup_type,
&BACKUP_ID_REGEX,
|_, backup_id, file_type| {
if file_type != nix::dir::Type::Directory {
return Ok(());
}
list.push(BackupGroup::new(backup_type, backup_id)); list.push(BackupGroup::new(backup_type, backup_id));
Ok(()) Ok(())
}) },
})?; )
},
)?;
Ok(list) Ok(list)
} }
pub fn is_finished(&self) -> bool { pub fn is_finished(&self) -> bool {
// backup is considered unfinished if there is no manifest // backup is considered unfinished if there is no manifest
self.files.iter().any(|name| name == super::MANIFEST_BLOB_NAME) self.files
.iter()
.any(|name| name == super::MANIFEST_BLOB_NAME)
} }
} }
fn list_backup_files<P: ?Sized + nix::NixPath>(dirfd: RawFd, path: &P) -> Result<Vec<String>, Error> { fn list_backup_files<P: ?Sized + nix::NixPath>(
dirfd: RawFd,
path: &P,
) -> Result<Vec<String>, Error> {
let mut files = vec![]; let mut files = vec![];
tools::scandir(dirfd, path, &BACKUP_FILE_REGEX, |_, filename, file_type| { tools::scandir(dirfd, path, &BACKUP_FILE_REGEX, |_, filename, file_type| {
if file_type != nix::dir::Type::File { return Ok(()); } if file_type != nix::dir::Type::File {
return Ok(());
}
files.push(filename.to_owned()); files.push(filename.to_owned());
Ok(()) Ok(())
})?; })?;