feat(memory): Reduce memory usage of restore

rustic-rs · Feb 19, 2024 · 388715b · 388715b
1 parent d20eaaa
commit 388715b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -40,8 +40,8 @@ rustdoc-args = ["--document-private-items", "--generate-link-to-definition"]
 
 [dependencies]
 abscissa_core = { version = "0.7.0", default-features = false, features = ["application"] }
-rustic_backend = { git = "https://github.com/rustic-rs/rustic_core.git", features = ["cli"] }
-rustic_core = { git = "https://github.com/rustic-rs/rustic_core.git", features = ["cli"] }
+rustic_backend = { git = "https://github.com/rustic-rs/rustic_core.git", branch = "drop-index", features = ["cli"] }
+rustic_core = { git = "https://github.com/rustic-rs/rustic_core.git", branch = "drop-index", features = ["cli"] }
 
 # allocators
 jemallocator-global = { version = "0.3.2", optional = true }

diff --git a/src/commands/restore.rs b/src/commands/restore.rs
@@ -64,7 +64,7 @@ impl RestoreCmd {
 
         let dest = LocalDestination::new(&self.dest, true, !node.is_dir())?;
 
-        let restore_infos = repo.prepare_restore(&self.opts, ls.clone(), &dest, dry_run)?;
+        let restore_infos = repo.prepare_restore(&self.opts, ls, &dest, dry_run)?;
 
         let fs = restore_infos.stats.files;
         println!(
@@ -94,6 +94,10 @@ impl RestoreCmd {
         if dry_run {
             repo.warm_up(restore_infos.to_packs().into_iter())?;
         } else {
+            // save some memory
+            let repo = repo.drop_data_from_index();
+
+            let ls = repo.ls(&node, &ls_opts)?;
             repo.restore(restore_infos, &self.opts, ls, &dest)?;
             println!("restore done.");
         }