mirrors/jj
1
0
Fork 1
mirror of https://github.com/martinvonz/jj.git synced 2024-12-04 05:28:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 12

dag_walk: extract topo-order helper that doesn't reverse the result

Browse source 
I'll add an iterator wrapper that pop()s the sorted result.
This commit is contained in:
Yuya Nishihara 2023-06-09 19:32:34 +09:00
parent 4987c74d4b
commit fb7fff4409
2 changed files with 19 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lib/src/dag_walk.rs
View file

@ -44,8 +44,7 @@ where
})
}
/// Returns neighbors before the node itself.
pub fn topo_order_reverse<T, ID, II, NI>(
pub fn topo_order_forward<T, ID, II, NI>(
start: II,
id_fn: impl Fn(&T) -> ID,
mut neighbors_fn: impl FnMut(&T) -> NI,
@ -75,6 +74,21 @@ where
result.push(node);
}
}
result
}
/// Returns neighbors before the node itself.
pub fn topo_order_reverse<T, ID, II, NI>(
start: II,
id_fn: impl Fn(&T) -> ID,
neighbors_fn: impl FnMut(&T) -> NI,
) -> Vec<T>
where
ID: Hash + Eq + Clone,
II: IntoIterator<Item = T>,
NI: IntoIterator<Item = T>,
{
let mut result = topo_order_forward(start, id_fn, neighbors_fn);
result.reverse();
result
}

7
lib/src/repo.rs
View file

@ -29,7 +29,6 @@ use self::dirty_cell::DirtyCell;
use crate::backend::{Backend, BackendError, BackendResult, ChangeId, CommitId, ObjectId, TreeId};
use crate::commit::Commit;
use crate::commit_builder::CommitBuilder;
use crate::dag_walk::topo_order_reverse;
use crate::default_index_store::DefaultIndexStore;
use crate::git_backend::GitBackend;
use crate::index::{HexPrefix, Index, IndexStore, MutableIndex, PrefixResolution, ReadonlyIndex};
@ -46,7 +45,7 @@ use crate::simple_op_store::SimpleOpStore;
use crate::store::Store;
use crate::transaction::Transaction;
use crate::view::{RefName, View};
use crate::{backend, op_store};
use crate::{backend, dag_walk, op_store};
pub trait Repo {
fn store(&self) -> &Arc<Store>;
@ -849,7 +848,7 @@ impl MutableRepo {
self.view.get_mut().remove_head(parent_id);
}
} else {
let missing_commits = topo_order_reverse(
let missing_commits = dag_walk::topo_order_forward(
vec![head.clone()],
|commit: &Commit| commit.id().clone(),
|commit: &Commit| -> Vec<Commit> {
@ -860,7 +859,7 @@ impl MutableRepo {
.collect()
},
);
for missing_commit in missing_commits.iter().rev() {
for missing_commit in &missing_commits {
self.index.add_commit(missing_commit);
}
self.view.get_mut().add_head(head.id());