erigon-pulse/core/state/plain_state_reader.go
2023-04-19 15:41:12 +07:00

82 lines
2.2 KiB
Go

package state
import (
"bytes"
"encoding/binary"
libcommon "github.com/ledgerwatch/erigon-lib/common"
"github.com/ledgerwatch/erigon-lib/kv"
"github.com/ledgerwatch/erigon/common/dbutils"
"github.com/ledgerwatch/erigon/core/types/accounts"
)
var _ StateReader = (*PlainStateReader)(nil)
// PlainStateReader reads data from so called "plain state".
// Data in the plain state is stored using un-hashed account/storage items
// as opposed to the "normal" state that uses hashes of merkle paths to store items.
type PlainStateReader struct {
db kv.Getter
}
func NewPlainStateReader(db kv.Getter) *PlainStateReader {
return &PlainStateReader{
db: db,
}
}
func (r *PlainStateReader) ReadAccountData(address libcommon.Address) (*accounts.Account, error) {
enc, err := r.db.GetOne(kv.PlainState, address.Bytes())
if err != nil {
return nil, err
}
if len(enc) == 0 {
return nil, nil
}
var a accounts.Account
if err = a.DecodeForStorage(enc); err != nil {
return nil, err
}
return &a, nil
}
func (r *PlainStateReader) ReadAccountStorage(address libcommon.Address, incarnation uint64, key *libcommon.Hash) ([]byte, error) {
compositeKey := dbutils.PlainGenerateCompositeStorageKey(address.Bytes(), incarnation, key.Bytes())
enc, err := r.db.GetOne(kv.PlainState, compositeKey)
if err != nil {
return nil, err
}
if len(enc) == 0 {
return nil, nil
}
return enc, nil
}
func (r *PlainStateReader) ReadAccountCode(address libcommon.Address, incarnation uint64, codeHash libcommon.Hash) ([]byte, error) {
if bytes.Equal(codeHash.Bytes(), emptyCodeHash) {
return nil, nil
}
code, err := r.db.GetOne(kv.Code, codeHash.Bytes())
if len(code) == 0 {
return nil, nil
}
return code, err
}
func (r *PlainStateReader) ReadAccountCodeSize(address libcommon.Address, incarnation uint64, codeHash libcommon.Hash) (int, error) {
code, err := r.ReadAccountCode(address, incarnation, codeHash)
return len(code), err
}
func (r *PlainStateReader) ReadAccountIncarnation(address libcommon.Address) (uint64, error) {
b, err := r.db.GetOne(kv.IncarnationMap, address.Bytes())
if err != nil {
return 0, err
}
if len(b) == 0 {
return 0, nil
}
return binary.BigEndian.Uint64(b), nil
}