Make dryrun an option to job::run, not a job property
This commit is contained in:
23
src/job.rs
23
src/job.rs
@@ -10,7 +10,6 @@ pub struct Job {
|
|||||||
target: String,
|
target: String,
|
||||||
source_zfs_command: Vec<String>,
|
source_zfs_command: Vec<String>,
|
||||||
target_zfs_command: Vec<String>,
|
target_zfs_command: Vec<String>,
|
||||||
dryrun: bool,
|
|
||||||
retain: usize,
|
retain: usize,
|
||||||
sender: Option<Sender<BackupEvent>>,
|
sender: Option<Sender<BackupEvent>>,
|
||||||
}
|
}
|
||||||
@@ -33,7 +32,6 @@ impl Job {
|
|||||||
println!("Target: {}", self.target);
|
println!("Target: {}", self.target);
|
||||||
println!("Source ZFS Command: {:?}", self.source_zfs_command);
|
println!("Source ZFS Command: {:?}", self.source_zfs_command);
|
||||||
println!("Target ZFS Command: {:?}", self.target_zfs_command);
|
println!("Target ZFS Command: {:?}", self.target_zfs_command);
|
||||||
println!("Dryrun: {}", self.dryrun);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send_event(&self, event: BackupEvent) {
|
fn send_event(&self, event: BackupEvent) {
|
||||||
@@ -191,7 +189,7 @@ impl Job {
|
|||||||
res
|
res
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn run(&self) -> Result<(), String> {
|
pub fn run(&self, execute: bool) -> Result<(), String> {
|
||||||
for (index, source) in self.datasets.iter().enumerate() {
|
for (index, source) in self.datasets.iter().enumerate() {
|
||||||
// Check the source exists
|
// Check the source exists
|
||||||
let mut cmd = self.get_side_command(JobSide::Source);
|
let mut cmd = self.get_side_command(JobSide::Source);
|
||||||
@@ -222,7 +220,7 @@ impl Job {
|
|||||||
|
|
||||||
let snapshot = self.create_snapshot(source, JobSide::Source)?;
|
let snapshot = self.create_snapshot(source, JobSide::Source)?;
|
||||||
let total = self.estimate(&snapshot, Some(&inc_snapshot)).ok();
|
let total = self.estimate(&snapshot, Some(&inc_snapshot)).ok();
|
||||||
if self.dryrun {
|
if !execute {
|
||||||
self.send_event(BackupEvent::DryrunCompleted(source.clone()));
|
self.send_event(BackupEvent::DryrunCompleted(source.clone()));
|
||||||
self.delete_snapshot(Snapshot {
|
self.delete_snapshot(Snapshot {
|
||||||
snapshot: snapshot.clone(),
|
snapshot: snapshot.clone(),
|
||||||
@@ -240,7 +238,7 @@ impl Job {
|
|||||||
});
|
});
|
||||||
let snapshot = self.create_snapshot(source, JobSide::Source)?;
|
let snapshot = self.create_snapshot(source, JobSide::Source)?;
|
||||||
let total = self.estimate(&snapshot, None).ok();
|
let total = self.estimate(&snapshot, None).ok();
|
||||||
if self.dryrun {
|
if !execute {
|
||||||
self.send_event(BackupEvent::DryrunCompleted(source.clone()));
|
self.send_event(BackupEvent::DryrunCompleted(source.clone()));
|
||||||
self.delete_snapshot(Snapshot {
|
self.delete_snapshot(Snapshot {
|
||||||
snapshot: snapshot.clone(),
|
snapshot: snapshot.clone(),
|
||||||
@@ -266,7 +264,6 @@ pub struct JobBuilder {
|
|||||||
target: String,
|
target: String,
|
||||||
source_zfs_command: Vec<String>,
|
source_zfs_command: Vec<String>,
|
||||||
target_zfs_command: Vec<String>,
|
target_zfs_command: Vec<String>,
|
||||||
dryrun: bool,
|
|
||||||
retain: usize,
|
retain: usize,
|
||||||
sender: Option<Sender<BackupEvent>>,
|
sender: Option<Sender<BackupEvent>>,
|
||||||
}
|
}
|
||||||
@@ -324,7 +321,6 @@ impl JobBuilder {
|
|||||||
target,
|
target,
|
||||||
source_zfs_command: vec!["zfs".to_string()],
|
source_zfs_command: vec!["zfs".to_string()],
|
||||||
target_zfs_command: vec!["zfs".to_string()],
|
target_zfs_command: vec!["zfs".to_string()],
|
||||||
dryrun: false,
|
|
||||||
retain: 2,
|
retain: 2,
|
||||||
sender: None,
|
sender: None,
|
||||||
}
|
}
|
||||||
@@ -349,11 +345,6 @@ impl JobBuilder {
|
|||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn dryrun(mut self) -> Self {
|
|
||||||
self.dryrun = true;
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn retain(mut self, retain: usize) -> Self {
|
pub fn retain(mut self, retain: usize) -> Self {
|
||||||
self.retain = retain;
|
self.retain = retain;
|
||||||
self
|
self
|
||||||
@@ -389,13 +380,19 @@ impl JobBuilder {
|
|||||||
target: self.target,
|
target: self.target,
|
||||||
source_zfs_command: self.source_zfs_command,
|
source_zfs_command: self.source_zfs_command,
|
||||||
target_zfs_command: self.target_zfs_command,
|
target_zfs_command: self.target_zfs_command,
|
||||||
dryrun: self.dryrun,
|
|
||||||
retain: self.retain,
|
retain: self.retain,
|
||||||
sender: self.sender,
|
sender: self.sender,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct JobConfig {
|
||||||
|
sources: Vec<String>,
|
||||||
|
target: String,
|
||||||
|
source_zfs_command: Option<String>,
|
||||||
|
target_zfs_command: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|||||||
@@ -35,9 +35,6 @@ struct Args {
|
|||||||
fn main() -> Result<(), Box<dyn Error>> {
|
fn main() -> Result<(), Box<dyn Error>> {
|
||||||
let args = Args::parse();
|
let args = Args::parse();
|
||||||
let mut builder = JobBuilder::new(args.datasets, args.target);
|
let mut builder = JobBuilder::new(args.datasets, args.target);
|
||||||
if args.dry_run {
|
|
||||||
builder = builder.dryrun();
|
|
||||||
}
|
|
||||||
if let Some(cmd) = args.zfs_command {
|
if let Some(cmd) = args.zfs_command {
|
||||||
builder = builder.zfs_command(&cmd);
|
builder = builder.zfs_command(&cmd);
|
||||||
}
|
}
|
||||||
@@ -62,6 +59,6 @@ fn main() -> Result<(), Box<dyn Error>> {
|
|||||||
builder = builder.sender(tx);
|
builder = builder.sender(tx);
|
||||||
|
|
||||||
let job = builder.build()?;
|
let job = builder.build()?;
|
||||||
job.run()?;
|
job.run(!args.dry_run)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user