forked from mirror/go-ethereum
core: refactored blockchain.go (#23735)
Marius van der Wijden
3 years ago
committed by
GitHub
parent
464885faaa
commit
c641cff51a
@ -0,0 +1,387 @@ |
||||
// Copyright 2021 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package core |
||||
|
||||
import ( |
||||
"math/big" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/consensus" |
||||
"github.com/ethereum/go-ethereum/core/rawdb" |
||||
"github.com/ethereum/go-ethereum/core/state" |
||||
"github.com/ethereum/go-ethereum/core/state/snapshot" |
||||
"github.com/ethereum/go-ethereum/core/types" |
||||
"github.com/ethereum/go-ethereum/core/vm" |
||||
"github.com/ethereum/go-ethereum/event" |
||||
"github.com/ethereum/go-ethereum/params" |
||||
"github.com/ethereum/go-ethereum/rlp" |
||||
) |
||||
|
||||
// CurrentHeader retrieves the current head header of the canonical chain. The
|
||||
// header is retrieved from the HeaderChain's internal cache.
|
||||
func (bc *BlockChain) CurrentHeader() *types.Header { |
||||
return bc.hc.CurrentHeader() |
||||
} |
||||
|
||||
// CurrentBlock retrieves the current head block of the canonical chain. The
|
||||
// block is retrieved from the blockchain's internal cache.
|
||||
func (bc *BlockChain) CurrentBlock() *types.Block { |
||||
return bc.currentBlock.Load().(*types.Block) |
||||
} |
||||
|
||||
// CurrentFastBlock retrieves the current fast-sync head block of the canonical
|
||||
// chain. The block is retrieved from the blockchain's internal cache.
|
||||
func (bc *BlockChain) CurrentFastBlock() *types.Block { |
||||
return bc.currentFastBlock.Load().(*types.Block) |
||||
} |
||||
|
||||
// HasHeader checks if a block header is present in the database or not, caching
|
||||
// it if present.
|
||||
func (bc *BlockChain) HasHeader(hash common.Hash, number uint64) bool { |
||||
return bc.hc.HasHeader(hash, number) |
||||
} |
||||
|
||||
// GetHeader retrieves a block header from the database by hash and number,
|
||||
// caching it if found.
|
||||
func (bc *BlockChain) GetHeader(hash common.Hash, number uint64) *types.Header { |
||||
return bc.hc.GetHeader(hash, number) |
||||
} |
||||
|
||||
// GetHeaderByHash retrieves a block header from the database by hash, caching it if
|
||||
// found.
|
||||
func (bc *BlockChain) GetHeaderByHash(hash common.Hash) *types.Header { |
||||
return bc.hc.GetHeaderByHash(hash) |
||||
} |
||||
|
||||
// GetHeaderByNumber retrieves a block header from the database by number,
|
||||
// caching it (associated with its hash) if found.
|
||||
func (bc *BlockChain) GetHeaderByNumber(number uint64) *types.Header { |
||||
return bc.hc.GetHeaderByNumber(number) |
||||
} |
||||
|
||||
// GetBody retrieves a block body (transactions and uncles) from the database by
|
||||
// hash, caching it if found.
|
||||
func (bc *BlockChain) GetBody(hash common.Hash) *types.Body { |
||||
// Short circuit if the body's already in the cache, retrieve otherwise
|
||||
if cached, ok := bc.bodyCache.Get(hash); ok { |
||||
body := cached.(*types.Body) |
||||
return body |
||||
} |
||||
number := bc.hc.GetBlockNumber(hash) |
||||
if number == nil { |
||||
return nil |
||||
} |
||||
body := rawdb.ReadBody(bc.db, hash, *number) |
||||
if body == nil { |
||||
return nil |
||||
} |
||||
// Cache the found body for next time and return
|
||||
bc.bodyCache.Add(hash, body) |
||||
return body |
||||
} |
||||
|
||||
// GetBodyRLP retrieves a block body in RLP encoding from the database by hash,
|
||||
// caching it if found.
|
||||
func (bc *BlockChain) GetBodyRLP(hash common.Hash) rlp.RawValue { |
||||
// Short circuit if the body's already in the cache, retrieve otherwise
|
||||
if cached, ok := bc.bodyRLPCache.Get(hash); ok { |
||||
return cached.(rlp.RawValue) |
||||
} |
||||
number := bc.hc.GetBlockNumber(hash) |
||||
if number == nil { |
||||
return nil |
||||
} |
||||
body := rawdb.ReadBodyRLP(bc.db, hash, *number) |
||||
if len(body) == 0 { |
||||
return nil |
||||
} |
||||
// Cache the found body for next time and return
|
||||
bc.bodyRLPCache.Add(hash, body) |
||||
return body |
||||
} |
||||
|
||||
// HasBlock checks if a block is fully present in the database or not.
|
||||
func (bc *BlockChain) HasBlock(hash common.Hash, number uint64) bool { |
||||
if bc.blockCache.Contains(hash) { |
||||
return true |
||||
} |
||||
return rawdb.HasBody(bc.db, hash, number) |
||||
} |
||||
|
||||
// HasFastBlock checks if a fast block is fully present in the database or not.
|
||||
func (bc *BlockChain) HasFastBlock(hash common.Hash, number uint64) bool { |
||||
if !bc.HasBlock(hash, number) { |
||||
return false |
||||
} |
||||
if bc.receiptsCache.Contains(hash) { |
||||
return true |
||||
} |
||||
return rawdb.HasReceipts(bc.db, hash, number) |
||||
} |
||||
|
||||
// GetBlock retrieves a block from the database by hash and number,
|
||||
// caching it if found.
|
||||
func (bc *BlockChain) GetBlock(hash common.Hash, number uint64) *types.Block { |
||||
// Short circuit if the block's already in the cache, retrieve otherwise
|
||||
if block, ok := bc.blockCache.Get(hash); ok { |
||||
return block.(*types.Block) |
||||
} |
||||
block := rawdb.ReadBlock(bc.db, hash, number) |
||||
if block == nil { |
||||
return nil |
||||
} |
||||
// Cache the found block for next time and return
|
||||
bc.blockCache.Add(block.Hash(), block) |
||||
return block |
||||
} |
||||
|
||||
// GetBlockByHash retrieves a block from the database by hash, caching it if found.
|
||||
func (bc *BlockChain) GetBlockByHash(hash common.Hash) *types.Block { |
||||
number := bc.hc.GetBlockNumber(hash) |
||||
if number == nil { |
||||
return nil |
||||
} |
||||
return bc.GetBlock(hash, *number) |
||||
} |
||||
|
||||
// GetBlockByNumber retrieves a block from the database by number, caching it
|
||||
// (associated with its hash) if found.
|
||||
func (bc *BlockChain) GetBlockByNumber(number uint64) *types.Block { |
||||
hash := rawdb.ReadCanonicalHash(bc.db, number) |
||||
if hash == (common.Hash{}) { |
||||
return nil |
||||
} |
||||
return bc.GetBlock(hash, number) |
||||
} |
||||
|
||||
// GetBlocksFromHash returns the block corresponding to hash and up to n-1 ancestors.
|
||||
// [deprecated by eth/62]
|
||||
func (bc *BlockChain) GetBlocksFromHash(hash common.Hash, n int) (blocks []*types.Block) { |
||||
number := bc.hc.GetBlockNumber(hash) |
||||
if number == nil { |
||||
return nil |
||||
} |
||||
for i := 0; i < n; i++ { |
||||
block := bc.GetBlock(hash, *number) |
||||
if block == nil { |
||||
break |
||||
} |
||||
blocks = append(blocks, block) |
||||
hash = block.ParentHash() |
||||
*number-- |
||||
} |
||||
return |
||||
} |
||||
|
||||
// GetReceiptsByHash retrieves the receipts for all transactions in a given block.
|
||||
func (bc *BlockChain) GetReceiptsByHash(hash common.Hash) types.Receipts { |
||||
if receipts, ok := bc.receiptsCache.Get(hash); ok { |
||||
return receipts.(types.Receipts) |
||||
} |
||||
number := rawdb.ReadHeaderNumber(bc.db, hash) |
||||
if number == nil { |
||||
return nil |
||||
} |
||||
receipts := rawdb.ReadReceipts(bc.db, hash, *number, bc.chainConfig) |
||||
if receipts == nil { |
||||
return nil |
||||
} |
||||
bc.receiptsCache.Add(hash, receipts) |
||||
return receipts |
||||
} |
||||
|
||||
// GetUnclesInChain retrieves all the uncles from a given block backwards until
|
||||
// a specific distance is reached.
|
||||
func (bc *BlockChain) GetUnclesInChain(block *types.Block, length int) []*types.Header { |
||||
uncles := []*types.Header{} |
||||
for i := 0; block != nil && i < length; i++ { |
||||
uncles = append(uncles, block.Uncles()...) |
||||
block = bc.GetBlock(block.ParentHash(), block.NumberU64()-1) |
||||
} |
||||
return uncles |
||||
} |
||||
|
||||
// GetCanonicalHash returns the canonical hash for a given block number
|
||||
func (bc *BlockChain) GetCanonicalHash(number uint64) common.Hash { |
||||
return bc.hc.GetCanonicalHash(number) |
||||
} |
||||
|
||||
// GetAncestor retrieves the Nth ancestor of a given block. It assumes that either the given block or
|
||||
// a close ancestor of it is canonical. maxNonCanonical points to a downwards counter limiting the
|
||||
// number of blocks to be individually checked before we reach the canonical chain.
|
||||
//
|
||||
// Note: ancestor == 0 returns the same block, 1 returns its parent and so on.
|
||||
func (bc *BlockChain) GetAncestor(hash common.Hash, number, ancestor uint64, maxNonCanonical *uint64) (common.Hash, uint64) { |
||||
return bc.hc.GetAncestor(hash, number, ancestor, maxNonCanonical) |
||||
} |
||||
|
||||
// GetTransactionLookup retrieves the lookup associate with the given transaction
|
||||
// hash from the cache or database.
|
||||
func (bc *BlockChain) GetTransactionLookup(hash common.Hash) *rawdb.LegacyTxLookupEntry { |
||||
// Short circuit if the txlookup already in the cache, retrieve otherwise
|
||||
if lookup, exist := bc.txLookupCache.Get(hash); exist { |
||||
return lookup.(*rawdb.LegacyTxLookupEntry) |
||||
} |
||||
tx, blockHash, blockNumber, txIndex := rawdb.ReadTransaction(bc.db, hash) |
||||
if tx == nil { |
||||
return nil |
||||
} |
||||
lookup := &rawdb.LegacyTxLookupEntry{BlockHash: blockHash, BlockIndex: blockNumber, Index: txIndex} |
||||
bc.txLookupCache.Add(hash, lookup) |
||||
return lookup |
||||
} |
||||
|
||||
// GetTd retrieves a block's total difficulty in the canonical chain from the
|
||||
// database by hash and number, caching it if found.
|
||||
func (bc *BlockChain) GetTd(hash common.Hash, number uint64) *big.Int { |
||||
return bc.hc.GetTd(hash, number) |
||||
} |
||||
|
||||
// HasState checks if state trie is fully present in the database or not.
|
||||
func (bc *BlockChain) HasState(hash common.Hash) bool { |
||||
_, err := bc.stateCache.OpenTrie(hash) |
||||
return err == nil |
||||
} |
||||
|
||||
// HasBlockAndState checks if a block and associated state trie is fully present
|
||||
// in the database or not, caching it if present.
|
||||
func (bc *BlockChain) HasBlockAndState(hash common.Hash, number uint64) bool { |
||||
// Check first that the block itself is known
|
||||
block := bc.GetBlock(hash, number) |
||||
if block == nil { |
||||
return false |
||||
} |
||||
return bc.HasState(block.Root()) |
||||
} |
||||
|
||||
// TrieNode retrieves a blob of data associated with a trie node
|
||||
// either from ephemeral in-memory cache, or from persistent storage.
|
||||
func (bc *BlockChain) TrieNode(hash common.Hash) ([]byte, error) { |
||||
return bc.stateCache.TrieDB().Node(hash) |
||||
} |
||||
|
||||
// ContractCode retrieves a blob of data associated with a contract hash
|
||||
// either from ephemeral in-memory cache, or from persistent storage.
|
||||
func (bc *BlockChain) ContractCode(hash common.Hash) ([]byte, error) { |
||||
return bc.stateCache.ContractCode(common.Hash{}, hash) |
||||
} |
||||
|
||||
// ContractCodeWithPrefix retrieves a blob of data associated with a contract
|
||||
// hash either from ephemeral in-memory cache, or from persistent storage.
|
||||
//
|
||||
// If the code doesn't exist in the in-memory cache, check the storage with
|
||||
// new code scheme.
|
||||
func (bc *BlockChain) ContractCodeWithPrefix(hash common.Hash) ([]byte, error) { |
||||
type codeReader interface { |
||||
ContractCodeWithPrefix(addrHash, codeHash common.Hash) ([]byte, error) |
||||
} |
||||
return bc.stateCache.(codeReader).ContractCodeWithPrefix(common.Hash{}, hash) |
||||
} |
||||
|
||||
// State returns a new mutable state based on the current HEAD block.
|
||||
func (bc *BlockChain) State() (*state.StateDB, error) { |
||||
return bc.StateAt(bc.CurrentBlock().Root()) |
||||
} |
||||
|
||||
// StateAt returns a new mutable state based on a particular point in time.
|
||||
func (bc *BlockChain) StateAt(root common.Hash) (*state.StateDB, error) { |
||||
return state.New(root, bc.stateCache, bc.snaps) |
||||
} |
||||
|
||||
// Config retrieves the chain's fork configuration.
|
||||
func (bc *BlockChain) Config() *params.ChainConfig { return bc.chainConfig } |
||||
|
||||
// Engine retrieves the blockchain's consensus engine.
|
||||
func (bc *BlockChain) Engine() consensus.Engine { return bc.engine } |
||||
|
||||
// Snapshots returns the blockchain snapshot tree.
|
||||
func (bc *BlockChain) Snapshots() *snapshot.Tree { |
||||
return bc.snaps |
||||
} |
||||
|
||||
// Validator returns the current validator.
|
||||
func (bc *BlockChain) Validator() Validator { |
||||
return bc.validator |
||||
} |
||||
|
||||
// Processor returns the current processor.
|
||||
func (bc *BlockChain) Processor() Processor { |
||||
return bc.processor |
||||
} |
||||
|
||||
// StateCache returns the caching database underpinning the blockchain instance.
|
||||
func (bc *BlockChain) StateCache() state.Database { |
||||
return bc.stateCache |
||||
} |
||||
|
||||
// GasLimit returns the gas limit of the current HEAD block.
|
||||
func (bc *BlockChain) GasLimit() uint64 { |
||||
return bc.CurrentBlock().GasLimit() |
||||
} |
||||
|
||||
// Genesis retrieves the chain's genesis block.
|
||||
func (bc *BlockChain) Genesis() *types.Block { |
||||
return bc.genesisBlock |
||||
} |
||||
|
||||
// GetVMConfig returns the block chain VM config.
|
||||
func (bc *BlockChain) GetVMConfig() *vm.Config { |
||||
return &bc.vmConfig |
||||
} |
||||
|
||||
// SetTxLookupLimit is responsible for updating the txlookup limit to the
|
||||
// original one stored in db if the new mismatches with the old one.
|
||||
func (bc *BlockChain) SetTxLookupLimit(limit uint64) { |
||||
bc.txLookupLimit = limit |
||||
} |
||||
|
||||
// TxLookupLimit retrieves the txlookup limit used by blockchain to prune
|
||||
// stale transaction indices.
|
||||
func (bc *BlockChain) TxLookupLimit() uint64 { |
||||
return bc.txLookupLimit |
||||
} |
||||
|
||||
// SubscribeRemovedLogsEvent registers a subscription of RemovedLogsEvent.
|
||||
func (bc *BlockChain) SubscribeRemovedLogsEvent(ch chan<- RemovedLogsEvent) event.Subscription { |
||||
return bc.scope.Track(bc.rmLogsFeed.Subscribe(ch)) |
||||
} |
||||
|
||||
// SubscribeChainEvent registers a subscription of ChainEvent.
|
||||
func (bc *BlockChain) SubscribeChainEvent(ch chan<- ChainEvent) event.Subscription { |
||||
return bc.scope.Track(bc.chainFeed.Subscribe(ch)) |
||||
} |
||||
|
||||
// SubscribeChainHeadEvent registers a subscription of ChainHeadEvent.
|
||||
func (bc *BlockChain) SubscribeChainHeadEvent(ch chan<- ChainHeadEvent) event.Subscription { |
||||
return bc.scope.Track(bc.chainHeadFeed.Subscribe(ch)) |
||||
} |
||||
|
||||
// SubscribeChainSideEvent registers a subscription of ChainSideEvent.
|
||||
func (bc *BlockChain) SubscribeChainSideEvent(ch chan<- ChainSideEvent) event.Subscription { |
||||
return bc.scope.Track(bc.chainSideFeed.Subscribe(ch)) |
||||
} |
||||
|
||||
// SubscribeLogsEvent registers a subscription of []*types.Log.
|
||||
func (bc *BlockChain) SubscribeLogsEvent(ch chan<- []*types.Log) event.Subscription { |
||||
return bc.scope.Track(bc.logsFeed.Subscribe(ch)) |
||||
} |
||||
|
||||
// SubscribeBlockProcessingEvent registers a subscription of bool where true means
|
||||
// block processing has started while false means it has stopped.
|
||||
func (bc *BlockChain) SubscribeBlockProcessingEvent(ch chan<- bool) event.Subscription { |
||||
return bc.scope.Track(bc.blockProcFeed.Subscribe(ch)) |
||||
} |
||||
Loading…
Reference in new issue