diff options
Diffstat (limited to 'src/cmd/backup.rs')
-rw-r--r-- | src/cmd/backup.rs | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/src/cmd/backup.rs b/src/cmd/backup.rs index 71f5aac..926b2b1 100644 --- a/src/cmd/backup.rs +++ b/src/cmd/backup.rs @@ -2,20 +2,38 @@ use crate::client::{BackupClient, ClientConfig}; use crate::fsiter::FsIterator; use crate::generation::Generation; use std::path::Path; +use tempfile::NamedTempFile; pub fn backup(config: &Path, buffer_size: usize) -> anyhow::Result<()> { let config = ClientConfig::read_config(config)?; let client = BackupClient::new(&config.server_url)?; + // Create a named temporary file. We don't meed the open file + // handle, so we discard that. + let dbname = { + let temp = NamedTempFile::new()?; + let (_, dbname) = temp.keep()?; + dbname + }; + { - let mut gen = Generation::create(&config.dbname)?; + // Create the SQLite database using the named temporary file. + // The fetching is in its own block so that the file handles + // get closed and data flushed to disk. + let mut gen = Generation::create(&dbname)?; gen.insert_iter(FsIterator::new(&config.root).map(|entry| match entry { Err(err) => Err(err), Ok(entry) => client.upload_filesystem_entry(entry, buffer_size), }))?; } - let gen_id = client.upload_generation(&config.dbname, buffer_size)?; + + // Upload the SQLite file, i.e., the named temporary file, which + // still exists, since we persisted it above. + let gen_id = client.upload_generation(&dbname, buffer_size)?; println!("gen id: {}", gen_id); + // Delete the temporary file. + std::fs::remove_file(&dbname)?; + Ok(()) } |