tree: after serializing transform first layer to hashed nodes

Signed-off-by: Ignacio Hagopian <jsign.uy@gmail.com>
ethereum · Jul 4, 2023 · 188a188 · 188a188
1 parent 628f400
commit 188a188
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tree.go b/tree.go
@@ -1538,6 +1538,14 @@ func (n *InternalNode) BatchSerialize() ([]SerializedNode, error) {
 		}
 	}
 
+	for i := range n.children {
+		if ch, ok := n.children[i].(*InternalNode); ok {
+			n.children[i] = ch.toHashedNode()
+		} else if ch, ok := n.children[i].(*LeafNode); ok {
+			n.children[i] = ch.ToHashedNode()
+		}
+	}
+
 	return ret, nil
 }