mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-04 01:54:28 +00:00
100 lines
3.0 KiB
Go
100 lines
3.0 KiB
Go
package state
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
|
|
"github.com/holiman/uint256"
|
|
"github.com/ledgerwatch/turbo-geth/common"
|
|
"github.com/ledgerwatch/turbo-geth/common/dbutils"
|
|
"github.com/ledgerwatch/turbo-geth/core/types/accounts"
|
|
"github.com/ledgerwatch/turbo-geth/ethdb"
|
|
)
|
|
|
|
var _ WriterWithChangeSets = (*PlainStateWriter)(nil)
|
|
|
|
type PlainStateWriter struct {
|
|
db ethdb.Database
|
|
csw *ChangeSetWriter
|
|
blockNumber uint64
|
|
}
|
|
|
|
func NewPlainStateWriter(db ethdb.Database, changeSetsDB ethdb.RwTx, blockNumber uint64) *PlainStateWriter {
|
|
return &PlainStateWriter{
|
|
db: db,
|
|
csw: NewChangeSetWriterPlain(changeSetsDB, blockNumber),
|
|
blockNumber: blockNumber,
|
|
}
|
|
}
|
|
|
|
func (w *PlainStateWriter) UpdateAccountData(ctx context.Context, address common.Address, original, account *accounts.Account) error {
|
|
if err := w.csw.UpdateAccountData(ctx, address, original, account); err != nil {
|
|
return err
|
|
}
|
|
value := make([]byte, account.EncodingLengthForStorage())
|
|
account.EncodeForStorage(value)
|
|
return w.db.Put(dbutils.PlainStateBucket, address[:], value)
|
|
}
|
|
|
|
func (w *PlainStateWriter) UpdateAccountCode(address common.Address, incarnation uint64, codeHash common.Hash, code []byte) error {
|
|
if err := w.csw.UpdateAccountCode(address, incarnation, codeHash, code); err != nil {
|
|
return err
|
|
}
|
|
if err := w.db.Put(dbutils.CodeBucket, codeHash[:], code); err != nil {
|
|
return err
|
|
}
|
|
return w.db.Put(dbutils.PlainContractCodeBucket, dbutils.PlainGenerateStoragePrefix(address[:], incarnation), codeHash[:])
|
|
}
|
|
|
|
func (w *PlainStateWriter) DeleteAccount(ctx context.Context, address common.Address, original *accounts.Account) error {
|
|
if err := w.csw.DeleteAccount(ctx, address, original); err != nil {
|
|
return err
|
|
}
|
|
if err := w.db.Delete(dbutils.PlainStateBucket, address[:], nil); err != nil {
|
|
return err
|
|
}
|
|
if original.Incarnation > 0 {
|
|
var b [8]byte
|
|
binary.BigEndian.PutUint64(b[:], original.Incarnation)
|
|
if err := w.db.Put(dbutils.IncarnationMapBucket, address[:], b[:]); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (w *PlainStateWriter) WriteAccountStorage(ctx context.Context, address common.Address, incarnation uint64, key *common.Hash, original, value *uint256.Int) error {
|
|
if err := w.csw.WriteAccountStorage(ctx, address, incarnation, key, original, value); err != nil {
|
|
return err
|
|
}
|
|
if *original == *value {
|
|
return nil
|
|
}
|
|
compositeKey := dbutils.PlainGenerateCompositeStorageKey(address.Bytes(), incarnation, key.Bytes())
|
|
|
|
v := value.Bytes()
|
|
if len(v) == 0 {
|
|
return w.db.Delete(dbutils.PlainStateBucket, compositeKey, nil)
|
|
}
|
|
return w.db.Put(dbutils.PlainStateBucket, compositeKey, v)
|
|
}
|
|
|
|
func (w *PlainStateWriter) CreateContract(address common.Address) error {
|
|
if err := w.csw.CreateContract(address); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (w *PlainStateWriter) WriteChangeSets() error {
|
|
return w.csw.WriteChangeSets()
|
|
}
|
|
|
|
func (w *PlainStateWriter) WriteHistory() error {
|
|
return w.csw.WriteHistory()
|
|
}
|
|
|
|
func (w *PlainStateWriter) ChangeSetWriter() *ChangeSetWriter {
|
|
return w.csw
|
|
}
|