fix: shrink to copying data

probe-lab · Aug 21, 2024 · 79800b4 · 79800b4
1 parent 60554b3
commit 79800b4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kad/trie/trie.go b/kad/trie/trie.go
@@ -90,9 +90,11 @@ func (tr *Trie[K, D]) shrink() {
 		tr.branch[0], tr.branch[1] = nil, nil
 	case b0.IsEmptyLeaf() && b1.IsNonEmptyLeaf():
 		tr.key = b1.key
+		tr.data = b1.data
 		tr.branch[0], tr.branch[1] = nil, nil
 	case b0.IsNonEmptyLeaf() && b1.IsEmptyLeaf():
 		tr.key = b0.key
+		tr.data = b0.data
 		tr.branch[0], tr.branch[1] = nil, nil
 	}
 }