mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-24 12:37:16 +00:00
106 lines
3.6 KiB
Go
106 lines
3.6 KiB
Go
package rpchelper
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/ledgerwatch/erigon-lib/kv"
|
|
"github.com/ledgerwatch/erigon-lib/kv/kvcache"
|
|
"github.com/ledgerwatch/erigon/cmd/rpcdaemon/filters"
|
|
"github.com/ledgerwatch/erigon/common"
|
|
"github.com/ledgerwatch/erigon/core/rawdb"
|
|
"github.com/ledgerwatch/erigon/core/state"
|
|
"github.com/ledgerwatch/erigon/core/types/accounts"
|
|
"github.com/ledgerwatch/erigon/eth/stagedsync/stages"
|
|
"github.com/ledgerwatch/erigon/rpc"
|
|
"github.com/ledgerwatch/erigon/turbo/adapter"
|
|
)
|
|
|
|
// unable to decode supplied params, or an invalid number of parameters
|
|
type nonCanonocalHashError struct{ hash common.Hash }
|
|
|
|
func (e nonCanonocalHashError) ErrorCode() int { return -32603 }
|
|
|
|
func (e nonCanonocalHashError) Error() string {
|
|
return fmt.Sprintf("hash %x is not currently canonical", e.hash)
|
|
}
|
|
|
|
func GetBlockNumber(blockNrOrHash rpc.BlockNumberOrHash, tx kv.Tx, filters *filters.Filters) (uint64, common.Hash, bool, error) {
|
|
return _GetBlockNumber(blockNrOrHash.RequireCanonical, blockNrOrHash, tx, filters)
|
|
}
|
|
|
|
func GetCanonicalBlockNumber(blockNrOrHash rpc.BlockNumberOrHash, tx kv.Tx, filters *filters.Filters) (uint64, common.Hash, bool, error) {
|
|
return _GetBlockNumber(true, blockNrOrHash, tx, filters)
|
|
}
|
|
|
|
func _GetBlockNumber(requireCanonical bool, blockNrOrHash rpc.BlockNumberOrHash, tx kv.Tx, filters *filters.Filters) (blockNumber uint64, hash common.Hash, latest bool, err error) {
|
|
var latestBlockNumber uint64
|
|
if latestBlockNumber, err = stages.GetStageProgress(tx, stages.Execution); err != nil {
|
|
return 0, common.Hash{}, false, fmt.Errorf("getting latest block number: %w", err)
|
|
}
|
|
var ok bool
|
|
hash, ok = blockNrOrHash.Hash()
|
|
if !ok {
|
|
number := *blockNrOrHash.BlockNumber
|
|
if number == rpc.LatestBlockNumber {
|
|
blockNumber = latestBlockNumber
|
|
} else if number == rpc.EarliestBlockNumber {
|
|
blockNumber = 0
|
|
} else if number == rpc.PendingBlockNumber {
|
|
pendingBlock := filters.LastPendingBlock()
|
|
if pendingBlock == nil {
|
|
blockNumber, err = stages.GetStageProgress(tx, stages.Execution)
|
|
if err != nil {
|
|
return 0, common.Hash{}, false, fmt.Errorf("getting latest block number: %w", err)
|
|
}
|
|
} else {
|
|
return pendingBlock.NumberU64(), pendingBlock.Hash(), false, nil
|
|
}
|
|
} else {
|
|
blockNumber = uint64(number.Int64())
|
|
}
|
|
hash, err = rawdb.ReadCanonicalHash(tx, blockNumber)
|
|
if err != nil {
|
|
return 0, common.Hash{}, false, err
|
|
}
|
|
} else {
|
|
number := rawdb.ReadHeaderNumber(tx, hash)
|
|
if number == nil {
|
|
return 0, common.Hash{}, false, fmt.Errorf("block %x not found", hash)
|
|
}
|
|
blockNumber = *number
|
|
|
|
ch, err := rawdb.ReadCanonicalHash(tx, blockNumber)
|
|
if err != nil {
|
|
return 0, common.Hash{}, false, err
|
|
}
|
|
if requireCanonical && ch != hash {
|
|
return 0, common.Hash{}, false, nonCanonocalHashError{hash}
|
|
}
|
|
}
|
|
return blockNumber, hash, blockNumber == latestBlockNumber, nil
|
|
}
|
|
|
|
func GetAccount(tx kv.Tx, blockNumber uint64, address common.Address) (*accounts.Account, error) {
|
|
reader := adapter.NewStateReader(tx, blockNumber)
|
|
return reader.ReadAccountData(address)
|
|
}
|
|
|
|
func CreateStateReader(ctx context.Context, tx kv.Tx, blockNrOrHash rpc.BlockNumberOrHash, filters *filters.Filters, stateCache kvcache.Cache) (state.StateReader, error) {
|
|
blockNumber, _, latest, err := _GetBlockNumber(true, blockNrOrHash, tx, filters)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var stateReader state.StateReader
|
|
if latest {
|
|
cacheView, err := stateCache.View(ctx, tx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
stateReader = state.NewCachedReader2(cacheView, tx)
|
|
} else {
|
|
stateReader = state.NewPlainState(tx, blockNumber+1)
|
|
}
|
|
return stateReader, nil
|
|
}
|