ok/jj
1
0
Fork 0
forked from mirrors/jj
Code Pull requests Activity

rewrite: use MergedTree::diff_stream() when restoring from tree

Browse source
This commit is contained in:
Martin von Zweigbergk 2023-11-03 14:20:11 -07:00 committed by Martin von Zweigbergk
parent 5b02987012
commit 6a5615c933
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
lib/src/rewrite.rs
View file

@ -17,7 +17,9 @@
use std::collections::{HashMap, HashSet};
use std::sync::Arc;
use futures::StreamExt;
use itertools::Itertools;
use pollster::FutureExt;
use tracing::instrument;
use crate::backend::{BackendError, BackendResult, CommitId, MergedTreeId, ObjectId};
@ -83,10 +85,15 @@ pub fn restore_tree(
// TODO: We should be able to not traverse deeper in the diff if the matcher
// matches an entire subtree.
let mut tree_builder = MergedTreeBuilder::new(destination.id().clone());
for (repo_path, diff) in source.diff(destination, matcher) {
let (source_value, _destination_value) = diff?;
tree_builder.set_or_remove(repo_path, source_value);
async {
let mut diff_stream = source.diff_stream(destination, matcher);
while let Some((repo_path, diff)) = diff_stream.next().await {
let (source_value, _destination_value) = diff?;
tree_builder.set_or_remove(repo_path, source_value);
}
Ok::<(), BackendError>(())
}
.block_on()?;
tree_builder.write_tree(destination.store())
}
}