erigon-pulse/turbo/trie/sub_trie_loader.go
alex.sharov e7034c0eba revert
2021-03-21 17:50:05 +07:00

83 lines
2.8 KiB
Go

package trie
import (
"github.com/ledgerwatch/turbo-geth/common"
"github.com/ledgerwatch/turbo-geth/ethdb"
)
var emptyHash [32]byte
// SubTrie is a result of loading sub-trie from either flat db
// or witness db. It encapsulates sub-trie root (which is of the un-exported type `node`)
// If the loading is done for verification and testing purposes, then usually only
// sub-tree root hash would be queried
type SubTries struct {
Hashes []common.Hash // Root hashes of the sub-tries
roots []node // Sub-tries
}
type LoadFunc func(*SubTrieLoader, *RetainList, [][]byte, []int) (SubTries, error)
// Resolver looks up (resolves) some keys and corresponding values from a database.
// One resolver per trie (prefix).
// See also ResolveRequest in trie.go
type SubTrieLoader struct {
blockNr uint64
codeRequests []*LoadRequestForCode
}
func NewSubTrieLoader(blockNr uint64) *SubTrieLoader {
tr := SubTrieLoader{
codeRequests: []*LoadRequestForCode{},
blockNr: blockNr,
}
return &tr
}
func (stl *SubTrieLoader) Reset(blockNr uint64) {
stl.blockNr = blockNr
stl.codeRequests = stl.codeRequests[:0]
}
// AddCodeRequest add a request for code loading
func (stl *SubTrieLoader) AddCodeRequest(req *LoadRequestForCode) {
stl.codeRequests = append(stl.codeRequests, req)
}
// Various values of the account field set
const (
AccountFieldNonceOnly uint32 = 0x01
AccountFieldBalanceOnly uint32 = 0x02
AccountFieldStorageOnly uint32 = 0x04
AccountFieldCodeOnly uint32 = 0x08
AccountFieldSSizeOnly uint32 = 0x10
AccountFieldSetNotAccount uint32 = 0x00
)
// LoadFromDb loads subtries from a state database.
func (stl *SubTrieLoader) LoadSubTries(db ethdb.Database, blockNr uint64, rl RetainDecider, hc HashCollector, dbPrefixes [][]byte, fixedbits []int, trace bool) (SubTries, error) {
return stl.LoadFromFlatDB(db, rl, hc, dbPrefixes, fixedbits, trace)
}
func (stl *SubTrieLoader) LoadFromFlatDB(db ethdb.Database, rl RetainDecider, hc HashCollector, dbPrefixes [][]byte, fixedbits []int, trace bool) (SubTries, error) {
loader := NewFlatDbSubTrieLoader()
if err1 := loader.Reset(db, rl, rl, hc, dbPrefixes, fixedbits, trace); err1 != nil {
return SubTries{}, err1
}
subTries, err := loader.LoadSubTries()
if err != nil {
return subTries, err
}
if err = loader.AttachRequestedCode(db, stl.codeRequests); err != nil {
return subTries, err
}
return subTries, nil
}
// LoadFromWitnessDb loads subtries from a witnesses database instead of a state DB.
func (stl *SubTrieLoader) LoadFromWitnessDb(db WitnessStorage, blockNr uint64, trieLimit uint32, startPos int64, count int) (SubTries, int64, error) {
loader := NewWitnessDbSubTrieLoader()
// we expect CodeNodes to be already attached to the trie when loading from witness db
return loader.LoadSubTries(db, blockNr, trieLimit, startPos, count)
}