trie: update comments + err check for preimages (#25672)

This PR includes minor updates to comments in trie/committer that reference insertion to the db, and adds an err != nil check for the return value of preimages.commit.
This commit is contained in:
Darioush Jalali 2022-09-08 03:36:07 -07:00 committed by GitHub
parent 6a575eda6f
commit 8363f79f8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 22 additions and 17 deletions

View File

@ -44,7 +44,8 @@ func newCommitter(owner common.Hash, collectLeaf bool) *committer {
} }
} }
// Commit collapses a node down into a hash node and inserts it into the database // Commit collapses a node down into a hash node and returns it along with
// the modified nodeset.
func (c *committer) Commit(n node) (hashNode, *NodeSet, error) { func (c *committer) Commit(n node) (hashNode, *NodeSet, error) {
h, err := c.commit(nil, n) h, err := c.commit(nil, n)
if err != nil { if err != nil {
@ -53,7 +54,7 @@ func (c *committer) Commit(n node) (hashNode, *NodeSet, error) {
return h.(hashNode), c.nodes, nil return h.(hashNode), c.nodes, nil
} }
// commit collapses a node down into a hash node and inserts it into the database // commit collapses a node down into a hash node and returns it.
func (c *committer) commit(path []byte, n node) (node, error) { func (c *committer) commit(path []byte, n node) (node, error) {
// if this path is clean, use available cached data // if this path is clean, use available cached data
hash, dirty := n.cache() hash, dirty := n.cache()
@ -75,7 +76,8 @@ func (c *committer) commit(path []byte, n node) (node, error) {
} }
collapsed.Val = childV collapsed.Val = childV
} }
// The key needs to be copied, since we're delivering it to database // The key needs to be copied, since we're adding it to the
// modified nodeset.
collapsed.Key = hexToCompact(cn.Key) collapsed.Key = hexToCompact(cn.Key)
hashedNode := c.store(path, collapsed) hashedNode := c.store(path, collapsed)
if hn, ok := hashedNode.(hashNode); ok { if hn, ok := hashedNode.(hashNode); ok {
@ -134,17 +136,16 @@ func (c *committer) commitChildren(path []byte, n *fullNode) ([17]node, error) {
return children, nil return children, nil
} }
// store hashes the node n and if we have a storage layer specified, it writes // store hashes the node n and adds it to the modified nodeset. If leaf collection
// the key/value pair to it and tracks any node->child references as well as any // is enabled, leaf nodes will be tracked in the modified nodeset as well.
// node->external trie references.
func (c *committer) store(path []byte, n node) node { func (c *committer) store(path []byte, n node) node {
// Larger nodes are replaced by their hash and stored in the database. // Larger nodes are replaced by their hash and stored in the database.
var hash, _ = n.cache() var hash, _ = n.cache()
// This was not generated - must be a small node stored in the parent. // This was not generated - must be a small node stored in the parent.
// In theory, we should check if the node is leaf here (embedded node // In theory, we should check if the node is leaf here (embedded node
// usually is leaf node). But small value(less than 32bytes) is not // usually is leaf node). But small value (less than 32bytes) is not
// our target(leaves in account trie only). // our target (leaves in account trie only).
if hash == nil { if hash == nil {
return n return n
} }

View File

@ -562,7 +562,9 @@ func (db *Database) Cap(limit common.StorageSize) error {
// If the preimage cache got large enough, push to disk. If it's still small // If the preimage cache got large enough, push to disk. If it's still small
// leave for later to deduplicate writes. // leave for later to deduplicate writes.
if db.preimages != nil { if db.preimages != nil {
db.preimages.commit(false) if err := db.preimages.commit(false); err != nil {
return err
}
} }
// Keep committing nodes from the flush-list until we're below allowance // Keep committing nodes from the flush-list until we're below allowance
oldest := db.oldest oldest := db.oldest
@ -640,7 +642,9 @@ func (db *Database) Commit(node common.Hash, report bool, callback func(common.H
// Move all of the accumulated preimages into a write batch // Move all of the accumulated preimages into a write batch
if db.preimages != nil { if db.preimages != nil {
db.preimages.commit(true) if err := db.preimages.commit(true); err != nil {
return err
}
} }
// Move the trie itself into the batch, flushing if enough data is accumulated // Move the trie itself into the batch, flushing if enough data is accumulated
nodes, storage := len(db.dirties), db.dirtiesSize nodes, storage := len(db.dirties), db.dirtiesSize

View File

@ -199,10 +199,10 @@ func (t *StateTrie) GetKey(shaKey []byte) []byte {
return t.preimages.preimage(common.BytesToHash(shaKey)) return t.preimages.preimage(common.BytesToHash(shaKey))
} }
// Commit collects all dirty nodes in the trie and replace them with the // Commit collects all dirty nodes in the trie and replaces them with the
// corresponding node hash. All collected nodes(including dirty leaves if // corresponding node hash. All collected nodes (including dirty leaves if
// collectLeaf is true) will be encapsulated into a nodeset for return. // collectLeaf is true) will be encapsulated into a nodeset for return.
// The returned nodeset can be nil if the trie is clean(nothing to commit). // The returned nodeset can be nil if the trie is clean (nothing to commit).
// All cached preimages will be also flushed if preimages recording is enabled. // All cached preimages will be also flushed if preimages recording is enabled.
// Once the trie is committed, it's not usable anymore. A new trie must // Once the trie is committed, it's not usable anymore. A new trie must
// be created with new root and updated trie database for following usage // be created with new root and updated trie database for following usage
@ -218,7 +218,7 @@ func (t *StateTrie) Commit(collectLeaf bool) (common.Hash, *NodeSet, error) {
} }
t.secKeyCache = make(map[string][]byte) t.secKeyCache = make(map[string][]byte)
} }
// Commit the trie to its intermediate node database // Commit the trie and return its modified nodeset.
return t.trie.Commit(collectLeaf) return t.trie.Commit(collectLeaf)
} }

View File

@ -582,10 +582,10 @@ func (t *Trie) Hash() common.Hash {
return common.BytesToHash(hash.(hashNode)) return common.BytesToHash(hash.(hashNode))
} }
// Commit collects all dirty nodes in the trie and replace them with the // Commit collects all dirty nodes in the trie and replaces them with the
// corresponding node hash. All collected nodes(including dirty leaves if // corresponding node hash. All collected nodes (including dirty leaves if
// collectLeaf is true) will be encapsulated into a nodeset for return. // collectLeaf is true) will be encapsulated into a nodeset for return.
// The returned nodeset can be nil if the trie is clean(nothing to commit). // The returned nodeset can be nil if the trie is clean (nothing to commit).
// Once the trie is committed, it's not usable anymore. A new trie must // Once the trie is committed, it's not usable anymore. A new trie must
// be created with new root and updated trie database for following usage // be created with new root and updated trie database for following usage
func (t *Trie) Commit(collectLeaf bool) (common.Hash, *NodeSet, error) { func (t *Trie) Commit(collectLeaf bool) (common.Hash, *NodeSet, error) {