mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-22 11:41:19 +00:00
72 lines
2.1 KiB
Go
72 lines
2.1 KiB
Go
package state
|
|
|
|
import (
|
|
"github.com/holiman/uint256"
|
|
"github.com/ledgerwatch/erigon-lib/common"
|
|
|
|
"github.com/ledgerwatch/erigon/core/types/accounts"
|
|
"github.com/ledgerwatch/erigon/turbo/shards"
|
|
)
|
|
|
|
// CachedWriter is a wrapper for an instance of type StateWriter
|
|
type CachedWriter struct {
|
|
w WriterWithChangeSets
|
|
cache *shards.StateCache
|
|
}
|
|
|
|
// NewCachedWriter wraps a given state writer into a cached writer
|
|
func NewCachedWriter(w WriterWithChangeSets, cache *shards.StateCache) *CachedWriter {
|
|
return &CachedWriter{w: w, cache: cache}
|
|
}
|
|
|
|
func (cw *CachedWriter) UpdateAccountData(address common.Address, original, account *accounts.Account) error {
|
|
if err := cw.w.UpdateAccountData(address, original, account); err != nil {
|
|
return err
|
|
}
|
|
cw.cache.SetAccountWrite(address.Bytes(), account)
|
|
return nil
|
|
}
|
|
|
|
func (cw *CachedWriter) UpdateAccountCode(address common.Address, incarnation uint64, codeHash common.Hash, code []byte) error {
|
|
if err := cw.w.UpdateAccountCode(address, incarnation, codeHash, code); err != nil {
|
|
return err
|
|
}
|
|
cw.cache.SetCodeWrite(address.Bytes(), incarnation, code)
|
|
return nil
|
|
}
|
|
|
|
func (cw *CachedWriter) DeleteAccount(address common.Address, original *accounts.Account) error {
|
|
if err := cw.w.DeleteAccount(address, original); err != nil {
|
|
return err
|
|
}
|
|
cw.cache.SetAccountDelete(address.Bytes())
|
|
return nil
|
|
}
|
|
|
|
func (cw *CachedWriter) WriteAccountStorage(address common.Address, incarnation uint64, key *common.Hash, original, value *uint256.Int) error {
|
|
if err := cw.w.WriteAccountStorage(address, incarnation, key, original, value); err != nil {
|
|
return err
|
|
}
|
|
if *original == *value {
|
|
return nil
|
|
}
|
|
if value.IsZero() {
|
|
cw.cache.SetStorageDelete(address.Bytes(), incarnation, key.Bytes())
|
|
} else {
|
|
cw.cache.SetStorageWrite(address.Bytes(), incarnation, key.Bytes(), value.Bytes())
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cw *CachedWriter) CreateContract(address common.Address) error {
|
|
return cw.w.CreateContract(address)
|
|
}
|
|
|
|
func (cw *CachedWriter) WriteChangeSets() error {
|
|
return cw.w.WriteChangeSets()
|
|
}
|
|
|
|
func (cw *CachedWriter) WriteHistory() error {
|
|
return cw.w.WriteHistory()
|
|
}
|