fix(trie): descendants count for clear prefix

ChainSafe · Jun 15, 2022 · 6c50153 · 6c50153
1 parent 7f55bec
commit 6c50153
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/trie/trie.go b/lib/trie/trie.go
@@ -877,10 +877,11 @@ func (t *Trie) clearPrefix(parent *Node, prefix []byte) (
 			return parent, nodesRemoved
 		}
 
-		nodesRemoved = 1
+		nodesRemoved = 1 + child.Descendants
 		copySettings := node.DefaultCopySettings
 		branch = t.prepBranchForMutation(branch, copySettings)
 		branch.Children[childIndex] = nil
+		branch.Descendants -= nodesRemoved
 		var branchChildMerged bool
 		newParent, branchChildMerged = handleDeletion(branch, prefix)
 		if branchChildMerged {
@@ -892,7 +893,7 @@ func (t *Trie) clearPrefix(parent *Node, prefix []byte) (
 	noPrefixForNode := len(prefix) <= len(branch.Key) ||
 		lenCommonPrefix(branch.Key, prefix) < len(branch.Key)
 	if noPrefixForNode {
-		const nodesRemoved = 0
+		const nodesRemoved = 0 // XXX
 		return parent, nodesRemoved
 	}