summaryrefslogtreecommitdiff
path: root/src/cmd/backup.rs
blob: da7298ffd273f8dc6fa3e6e44524558fe8493a0f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
use crate::backup_run::BackupRun;
use crate::client::ClientConfig;
use crate::fsiter::FsIterator;
use crate::generation::NascentGeneration;
use log::info;
use std::time::SystemTime;
use tempfile::NamedTempFile;

pub fn backup(config: &ClientConfig, buffer_size: usize) -> anyhow::Result<()> {
    let runtime = SystemTime::now();

    let run = BackupRun::new(config, buffer_size)?;

    // Create a named temporary file. We don't meed the open file
    // handle, so we discard that.
    let oldname = {
        let temp = NamedTempFile::new()?;
        let (_, dbname) = temp.keep()?;
        dbname
    };

    // Create a named temporary file. We don't meed the open file
    // handle, so we discard that.
    let newname = {
        let temp = NamedTempFile::new()?;
        let (_, dbname) = temp.keep()?;
        dbname
    };

    let genlist = run.client().list_generations()?;
    let file_count = {
        let iter = FsIterator::new(&config.root);
        let mut new = NascentGeneration::create(&newname)?;

        match genlist.resolve("latest") {
            None => {
                info!("fresh backup without a previous generation");
                new.insert_iter(iter.map(|entry| run.backup_file_initially(entry)))?;
            }
            Some(old) => {
                info!("incremental backup based on {}", old);
                let old = run.client().fetch_generation(&old, &oldname)?;
                run.progress()
                    .files_in_previous_generation(old.file_count()? as u64);
                new.insert_iter(iter.map(|entry| run.backup_file_incrementally(entry, &old)))?;
            }
        }
        run.progress().finish();
        new.file_count()
    };

    // Upload the SQLite file, i.e., the named temporary file, which
    // still exists, since we persisted it above.
    let gen_id = run.client().upload_generation(&newname, buffer_size)?;
    println!("status: OK");
    println!("duration: {}", runtime.elapsed()?.as_secs());
    println!("file-count: {}", file_count);
    println!("generation-id: {}", gen_id);

    // Delete the temporary file.q
    std::fs::remove_file(&newname)?;
    std::fs::remove_file(&oldname)?;

    Ok(())
}