github/reth
1
1
Fork 0
mirror of https://github.com/paradigmxyz/reth.git synced 2026-02-19 03:04:27 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: update sparse trie masks (#21716)

Browse Source
This commit is contained in:
Arsenii Kulikov
2026-02-03 15:01:58 +04:00
committed by GitHub
parent 972f23745e
commit ee224fe20f
2 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
crates/trie/sparse-parallel/src/trie.rs
View File

@@ -913,6 +913,19 @@ impl SparseTrie for ParallelSparseTrie {
fn take_updates(&mut self) -> SparseTrieUpdates {
match self.updates.take() {
Some(updates) => {
// Sync branch_node_masks with what's being committed to DB.
// This ensures that on subsequent root() calls, the masks reflect the actual
// DB state, which is needed for correct removal detection.
for (path, node) in &updates.updated_nodes {
self.branch_node_masks.insert(
*path,
BranchNodeMasks { tree_mask: node.tree_mask, hash_mask: node.hash_mask },
);
}
for path in &updates.removed_nodes {
self.branch_node_masks.remove(path);
}
// NOTE: we need to preserve Some case
self.updates = Some(SparseTrieUpdates::with_capacity(
updates.updated_nodes.len(),

13
crates/trie/sparse/src/trie.rs
View File

@@ -980,6 +980,19 @@ impl SparseTrieTrait for SerialSparseTrie {
fn take_updates(&mut self) -> SparseTrieUpdates {
match self.updates.take() {
Some(updates) => {
// Sync branch_node_masks with what's being committed to DB.
// This ensures that on subsequent root() calls, the masks reflect the actual
// DB state, which is needed for correct removal detection.
for (path, node) in &updates.updated_nodes {
self.branch_node_masks.insert(
*path,
BranchNodeMasks { tree_mask: node.tree_mask, hash_mask: node.hash_mask },
);
}
for path in &updates.removed_nodes {
self.branch_node_masks.remove(path);
}
// NOTE: we need to preserve Some case
self.updates = Some(SparseTrieUpdates::with_capacity(
updates.updated_nodes.len(),