erigon-pulse/cmd/rpcdaemon/commands/erigon_block.go

242 lines
5.9 KiB
Go
Raw Normal View History

package commands
import (
"bytes"
"context"
2022-07-01 14:36:44 +00:00
"errors"
"fmt"
"sort"
"github.com/holiman/uint256"
"github.com/ledgerwatch/erigon-lib/kv"
"github.com/ledgerwatch/erigon/common"
"github.com/ledgerwatch/erigon/common/changeset"
"github.com/ledgerwatch/erigon/common/dbutils"
"github.com/ledgerwatch/erigon/common/hexutil"
"github.com/ledgerwatch/erigon/core/rawdb"
"github.com/ledgerwatch/erigon/core/types"
"github.com/ledgerwatch/erigon/core/types/accounts"
"github.com/ledgerwatch/erigon/internal/ethapi"
"github.com/ledgerwatch/erigon/rpc"
"github.com/ledgerwatch/erigon/turbo/rpchelper"
)
// GetHeaderByNumber implements erigon_getHeaderByNumber. Returns a block's header given a block number ignoring the block's transaction and uncle list (may be faster).
func (api *ErigonImpl) GetHeaderByNumber(ctx context.Context, blockNumber rpc.BlockNumber) (*types.Header, error) {
// Pending block is only known by the miner
if blockNumber == rpc.PendingBlockNumber {
block := api.pendingBlock()
if block == nil {
return nil, nil
}
return block.Header(), nil
}
2021-04-03 06:26:00 +00:00
tx, err := api.db.BeginRo(ctx)
if err != nil {
return nil, err
}
defer tx.Rollback()
blockNum, _, _, err := rpchelper.GetBlockNumber(rpc.BlockNumberOrHashWithNumber(blockNumber), tx, api.filters)
if err != nil {
return nil, err
}
2022-07-01 14:36:44 +00:00
header, err := api._blockReader.HeaderByNumber(ctx, tx, blockNum)
if err != nil {
return nil, err
}
if header == nil {
return nil, fmt.Errorf("block header not found: %d", blockNum)
}
return header, nil
}
// GetHeaderByHash implements erigon_getHeaderByHash. Returns a block's header given a block's hash.
func (api *ErigonImpl) GetHeaderByHash(ctx context.Context, hash common.Hash) (*types.Header, error) {
2021-04-03 06:26:00 +00:00
tx, err := api.db.BeginRo(ctx)
if err != nil {
return nil, err
}
defer tx.Rollback()
2022-07-01 14:36:44 +00:00
header, err := api._blockReader.HeaderByHash(ctx, tx, hash)
if err != nil {
return nil, err
}
if header == nil {
return nil, fmt.Errorf("block header not found: %s", hash.String())
}
return header, nil
}
func (api *ErigonImpl) GetBlockByTimestamp(ctx context.Context, timeStamp rpc.Timestamp, fullTx bool) (map[string]interface{}, error) {
tx, err := api.db.BeginRo(ctx)
if err != nil {
return nil, err
}
defer tx.Rollback()
uintTimestamp := timeStamp.TurnIntoUint64()
currentHeader := rawdb.ReadCurrentHeader(tx)
currenttHeaderTime := currentHeader.Time
highestNumber := currentHeader.Number.Uint64()
2022-07-01 14:36:44 +00:00
firstHeader, err := api._blockReader.HeaderByNumber(ctx, tx, 0)
if err != nil {
return nil, err
}
if firstHeader == nil {
return nil, errors.New("no genesis header found")
}
firstHeaderTime := firstHeader.Time
if currenttHeaderTime <= uintTimestamp {
blockResponse, err := buildBlockResponse(tx, highestNumber, fullTx)
if err != nil {
return nil, err
}
return blockResponse, nil
}
if firstHeaderTime >= uintTimestamp {
blockResponse, err := buildBlockResponse(tx, 0, fullTx)
if err != nil {
return nil, err
}
return blockResponse, nil
}
blockNum := sort.Search(int(currentHeader.Number.Uint64()), func(blockNum int) bool {
2022-07-01 14:36:44 +00:00
currentHeader, err := api._blockReader.HeaderByNumber(ctx, tx, uint64(blockNum))
if err != nil {
return false
}
if currentHeader == nil {
return false
}
return currentHeader.Time >= uintTimestamp
})
2022-07-01 14:36:44 +00:00
resultingHeader, err := api._blockReader.HeaderByNumber(ctx, tx, uint64(blockNum))
if err != nil {
return nil, err
}
if resultingHeader == nil {
return nil, fmt.Errorf("no header found with header number: %d", blockNum)
}
if resultingHeader.Time > uintTimestamp {
response, err := buildBlockResponse(tx, uint64(blockNum)-1, fullTx)
if err != nil {
return nil, err
}
return response, nil
}
response, err := buildBlockResponse(tx, uint64(blockNum), fullTx)
if err != nil {
return nil, err
}
return response, nil
}
func buildBlockResponse(db kv.Tx, blockNum uint64, fullTx bool) (map[string]interface{}, error) {
block, err := rawdb.ReadBlockByNumber(db, blockNum)
if err != nil {
return nil, err
}
if block == nil {
return nil, nil
}
response, err := ethapi.RPCMarshalBlock(block, true, fullTx)
if err == nil && rpc.BlockNumber(block.NumberU64()) == rpc.PendingBlockNumber {
// Pending blocks need to nil out a few fields
for _, field := range []string{"hash", "nonce", "miner"} {
response[field] = nil
}
}
return response, err
}
func (api *ErigonImpl) GetBalanceChangesInBlock(ctx context.Context, blockNrOrHash rpc.BlockNumberOrHash) (map[common.Address]*hexutil.Big, error) {
tx, err := api.db.BeginRo(ctx)
if err != nil {
return nil, err
}
defer tx.Rollback()
blockNumber, _, _, err := rpchelper.GetBlockNumber(blockNrOrHash, tx, api.filters)
if err != nil {
return nil, err
}
c, err := tx.Cursor(kv.AccountChangeSet)
if err != nil {
return nil, err
}
defer c.Close()
startkey := dbutils.EncodeBlockNumber(blockNumber)
decodeFn := changeset.Mapper[kv.AccountChangeSet].Decode
balancesMapping := make(map[common.Address]*hexutil.Big)
newReader, err := rpchelper.CreateStateReader(ctx, tx, blockNrOrHash, api.filters, api.stateCache, api.historyV2(tx), api._agg, api._txNums)
if err != nil {
return nil, err
}
for dbKey, dbValue, err := c.Seek(startkey); bytes.Equal(dbKey, startkey) && dbKey != nil; dbKey, dbValue, err = c.Next() {
if err != nil {
return nil, err
}
_, addressBytes, v, err := decodeFn(dbKey, dbValue)
if err != nil {
return nil, err
}
var oldAcc accounts.Account
if err = oldAcc.DecodeForStorage(v); err != nil {
return nil, err
}
oldBalance := oldAcc.Balance
address := common.BytesToAddress(addressBytes)
newAcc, err := newReader.ReadAccountData(address)
if err != nil {
return nil, err
}
newBalance := uint256.NewInt(0)
if newAcc != nil {
newBalance = &newAcc.Balance
}
if !oldBalance.Eq(newBalance) {
newBalanceDesc := (*hexutil.Big)(newBalance.ToBig())
balancesMapping[address] = newBalanceDesc
}
}
return balancesMapping, nil
}