Fix list diffing with root dynamic nodes (#2749)

This commit is contained in:
Evan Almloff 2024-07-31 00:16:42 +02:00 committed by GitHub
parent df8c7e1872
commit 1b0d089d19
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 101 additions and 6 deletions

View file

@ -154,9 +154,11 @@ impl VNode {
pub(crate) fn find_first_element(&self, dom: &VirtualDom) -> ElementId {
let mount = &dom.mounts[self.mount.get().0];
match self.get_dynamic_root_node_and_id(0) {
let first = match self.get_dynamic_root_node_and_id(0) {
// This node is static, just get the root id
None | Some((_, Placeholder(_) | Text(_))) => mount.root_ids[0],
None => mount.root_ids[0],
// If it is dynamic and shallow, grab the id from the mounted dynamic nodes
Some((idx, Placeholder(_) | Text(_))) => ElementId(mount.mounted_dynamic_nodes[idx]),
// The node is a fragment, so we need to find the first element in the fragment
Some((_, Fragment(children))) => {
let child = children.first().unwrap();
@ -170,15 +172,22 @@ impl VNode {
.root_node()
.find_first_element(dom)
}
}
};
// The first element should never be the default element id (the root element)
debug_assert_ne!(first, ElementId::default());
first
}
pub(crate) fn find_last_element(&self, dom: &VirtualDom) -> ElementId {
let mount = &dom.mounts[self.mount.get().0];
let last_root_index = self.template.roots.len() - 1;
match self.get_dynamic_root_node_and_id(last_root_index) {
let last = match self.get_dynamic_root_node_and_id(last_root_index) {
// This node is static, just get the root id
None | Some((_, Placeholder(_) | Text(_))) => mount.root_ids[last_root_index],
None => mount.root_ids[last_root_index],
// If it is dynamic and shallow, grab the id from the mounted dynamic nodes
Some((idx, Placeholder(_) | Text(_))) => ElementId(mount.mounted_dynamic_nodes[idx]),
// The node is a fragment, so we need to find the first element in the fragment
Some((_, Fragment(children))) => {
let child = children.first().unwrap();
@ -192,7 +201,12 @@ impl VNode {
.root_node()
.find_last_element(dom)
}
}
};
// The last element should never be the default element id (the root element)
debug_assert_ne!(last, ElementId::default());
last
}
/// Diff the two text nodes

View file

@ -390,3 +390,84 @@ fn remove_many() {
)
}
}
#[test]
fn replace_and_add_items() {
let mut dom = VirtualDom::new(|| {
let items = (0..generation()).map(|_| {
if generation() % 2 == 0 {
VNode::empty()
} else {
rsx! {
li {
"Fizz"
}
}
}
});
rsx! {
ul {
{items}
}
}
});
// The list starts empty with a placeholder
{
let edits = dom.rebuild_to_vec().sanitize();
assert_eq!(
edits.edits,
[
LoadTemplate { name: "template", index: 0, id: ElementId(1,) },
AssignId { path: &[0], id: ElementId(2,) },
AppendChildren { id: ElementId(0), m: 1 },
]
);
}
// Rerendering adds an a static template
{
dom.mark_dirty(ScopeId::APP);
let edits = dom.render_immediate_to_vec().sanitize();
assert_eq!(
edits.edits,
[
LoadTemplate { name: "template", index: 0, id: ElementId(3,) },
ReplaceWith { id: ElementId(2,), m: 1 },
]
);
}
// Rerendering replaces the old node with a placeholder and adds a new placeholder
{
dom.mark_dirty(ScopeId::APP);
let edits = dom.render_immediate_to_vec().sanitize();
assert_eq!(
edits.edits,
[
CreatePlaceholder { id: ElementId(2,) },
InsertAfter { id: ElementId(3,), m: 1 },
CreatePlaceholder { id: ElementId(4,) },
ReplaceWith { id: ElementId(3,), m: 1 },
]
);
}
// Rerendering replaces both placeholders with the static nodes and add a new static node
{
dom.mark_dirty(ScopeId::APP);
let edits = dom.render_immediate_to_vec().sanitize();
assert_eq!(
edits.edits,
[
LoadTemplate { name: "template", index: 0, id: ElementId(3,) },
InsertAfter { id: ElementId(2,), m: 1 },
LoadTemplate { name: "template", index: 0, id: ElementId(5,) },
ReplaceWith { id: ElementId(4,), m: 1 },
LoadTemplate { name: "template", index: 0, id: ElementId(4,) },
ReplaceWith { id: ElementId(2,), m: 1 },
]
);
}
}