mirrors/loro
1
0
Fork 0
mirror of https://github.com/loro-dev/loro.git synced 2025-02-02 11:06:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: simplify create level when apply update

Browse source
This commit is contained in:
Zixuan Chen 2022-10-13 16:55:36 +08:00
parent 506350a45a
commit 81b8f2c591
1 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
crates/rle/src/rle_tree/node/internal_impl.rs
View file

@ -334,14 +334,31 @@ impl<'a, T: Rle, A: RleTreeTrait<T>> InternalNode<'a, T, A> {
};
if result.is_err() && self.is_root() {
let mut new = result.unwrap_err();
assert!(new.len() == 1);
let new = new.pop().unwrap();
let inner = new.as_internal_mut().unwrap();
self._balance(inner);
let new_vec = result.unwrap_err();
{
// create level
let origin_root = self.bump.alloc(Node::Internal(InternalNode::new(
self.bump,
Some(self.into()),
)));
let origin_root_internal = origin_root.as_internal_mut().unwrap();
std::mem::swap(&mut self.children, &mut origin_root_internal.children);
let ptr = origin_root_internal.into();
for child in origin_root_internal.children.iter_mut() {
child.set_parent(ptr);
}
A::update_cache_internal(origin_root_internal);
self.children.push(origin_root);
}
let ptr = self.into();
for new_node in new_vec {
new_node.set_parent(ptr);
self.children.push(new_node);
}
A::update_cache_internal(self);
A::update_cache_internal(inner);
self._create_level(new);
Ok(())
} else {
result