mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-23 04:03:49 +00:00
175 lines
6.2 KiB
Go
175 lines
6.2 KiB
Go
package shards
|
|
|
|
import (
|
|
"context"
|
|
|
|
libcommon "github.com/ledgerwatch/erigon-lib/common"
|
|
"github.com/ledgerwatch/erigon-lib/gointerfaces"
|
|
"github.com/ledgerwatch/erigon-lib/gointerfaces/remote"
|
|
)
|
|
|
|
// Accumulator collects state changes in a form that can then be delivered to the RPC daemon
|
|
type Accumulator struct {
|
|
plainStateID uint64
|
|
changes []*remote.StateChange
|
|
latestChange *remote.StateChange
|
|
accountChangeIndex map[libcommon.Address]int // For the latest changes, allows finding account change by account's address
|
|
storageChangeIndex map[libcommon.Address]map[libcommon.Hash]int
|
|
}
|
|
|
|
func NewAccumulator() *Accumulator {
|
|
return &Accumulator{}
|
|
}
|
|
|
|
type StateChangeConsumer interface {
|
|
SendStateChanges(ctx context.Context, sc *remote.StateChangeBatch)
|
|
}
|
|
|
|
func (a *Accumulator) Reset(plainStateID uint64) {
|
|
a.changes = nil
|
|
a.latestChange = nil
|
|
a.accountChangeIndex = nil
|
|
a.storageChangeIndex = nil
|
|
a.plainStateID = plainStateID
|
|
}
|
|
func (a *Accumulator) SendAndReset(ctx context.Context, c StateChangeConsumer, pendingBaseFee uint64, blockGasLimit uint64) {
|
|
if a == nil || c == nil || len(a.changes) == 0 {
|
|
return
|
|
}
|
|
sc := &remote.StateChangeBatch{StateVersionID: a.plainStateID, ChangeBatch: a.changes, PendingBlockBaseFee: pendingBaseFee, BlockGasLimit: blockGasLimit}
|
|
c.SendStateChanges(ctx, sc)
|
|
a.Reset(0) // reset here for GC, but there will be another Reset with correct viewID
|
|
}
|
|
|
|
func (a *Accumulator) SetStateID(stateID uint64) {
|
|
a.plainStateID = stateID
|
|
}
|
|
|
|
// StartChange begins accumulation of changes for a new block
|
|
func (a *Accumulator) StartChange(blockHeight uint64, blockHash libcommon.Hash, txs [][]byte, unwind bool) {
|
|
a.changes = append(a.changes, &remote.StateChange{})
|
|
a.latestChange = a.changes[len(a.changes)-1]
|
|
a.latestChange.BlockHeight = blockHeight
|
|
a.latestChange.BlockHash = gointerfaces.ConvertHashToH256(blockHash)
|
|
if unwind {
|
|
a.latestChange.Direction = remote.Direction_UNWIND
|
|
} else {
|
|
a.latestChange.Direction = remote.Direction_FORWARD
|
|
}
|
|
a.accountChangeIndex = make(map[libcommon.Address]int)
|
|
a.storageChangeIndex = make(map[libcommon.Address]map[libcommon.Hash]int)
|
|
if txs != nil {
|
|
a.latestChange.Txs = make([][]byte, len(txs))
|
|
for i := range txs {
|
|
a.latestChange.Txs[i] = libcommon.Copy(txs[i])
|
|
}
|
|
}
|
|
}
|
|
|
|
// ChangeAccount adds modification of account balance or nonce (or both) to the latest change
|
|
func (a *Accumulator) ChangeAccount(address libcommon.Address, incarnation uint64, data []byte) {
|
|
i, ok := a.accountChangeIndex[address]
|
|
if !ok || incarnation > a.latestChange.Changes[i].Incarnation {
|
|
// Account has not been changed in the latest block yet
|
|
i = len(a.latestChange.Changes)
|
|
a.latestChange.Changes = append(a.latestChange.Changes, &remote.AccountChange{Address: gointerfaces.ConvertAddressToH160(address)})
|
|
a.accountChangeIndex[address] = i
|
|
delete(a.storageChangeIndex, address)
|
|
}
|
|
accountChange := a.latestChange.Changes[i]
|
|
switch accountChange.Action {
|
|
case remote.Action_STORAGE:
|
|
accountChange.Action = remote.Action_UPSERT
|
|
case remote.Action_CODE:
|
|
accountChange.Action = remote.Action_UPSERT_CODE
|
|
case remote.Action_REMOVE:
|
|
panic("")
|
|
}
|
|
accountChange.Incarnation = incarnation
|
|
accountChange.Data = data
|
|
}
|
|
|
|
// DeleteAccount marks account as deleted
|
|
func (a *Accumulator) DeleteAccount(address libcommon.Address) {
|
|
i, ok := a.accountChangeIndex[address]
|
|
if !ok {
|
|
// Account has not been changed in the latest block yet
|
|
i = len(a.latestChange.Changes)
|
|
a.latestChange.Changes = append(a.latestChange.Changes, &remote.AccountChange{Address: gointerfaces.ConvertAddressToH160(address)})
|
|
a.accountChangeIndex[address] = i
|
|
}
|
|
accountChange := a.latestChange.Changes[i]
|
|
if accountChange.Action != remote.Action_STORAGE {
|
|
panic("")
|
|
}
|
|
accountChange.Data = nil
|
|
accountChange.Code = nil
|
|
accountChange.StorageChanges = nil
|
|
accountChange.Action = remote.Action_REMOVE
|
|
delete(a.storageChangeIndex, address)
|
|
}
|
|
|
|
// ChangeCode adds code to the latest change
|
|
func (a *Accumulator) ChangeCode(address libcommon.Address, incarnation uint64, code []byte) {
|
|
i, ok := a.accountChangeIndex[address]
|
|
if !ok || incarnation > a.latestChange.Changes[i].Incarnation {
|
|
// Account has not been changed in the latest block yet
|
|
i = len(a.latestChange.Changes)
|
|
a.latestChange.Changes = append(a.latestChange.Changes, &remote.AccountChange{Address: gointerfaces.ConvertAddressToH160(address), Action: remote.Action_CODE})
|
|
a.accountChangeIndex[address] = i
|
|
delete(a.storageChangeIndex, address)
|
|
}
|
|
accountChange := a.latestChange.Changes[i]
|
|
switch accountChange.Action {
|
|
case remote.Action_STORAGE:
|
|
accountChange.Action = remote.Action_CODE
|
|
case remote.Action_UPSERT:
|
|
accountChange.Action = remote.Action_UPSERT_CODE
|
|
case remote.Action_REMOVE:
|
|
panic("")
|
|
}
|
|
accountChange.Incarnation = incarnation
|
|
accountChange.Code = code
|
|
}
|
|
|
|
func (a *Accumulator) ChangeStorage(address libcommon.Address, incarnation uint64, location libcommon.Hash, data []byte) {
|
|
i, ok := a.accountChangeIndex[address]
|
|
if !ok || incarnation > a.latestChange.Changes[i].Incarnation {
|
|
// Account has not been changed in the latest block yet
|
|
i = len(a.latestChange.Changes)
|
|
a.latestChange.Changes = append(a.latestChange.Changes, &remote.AccountChange{Address: gointerfaces.ConvertAddressToH160(address), Action: remote.Action_STORAGE})
|
|
a.accountChangeIndex[address] = i
|
|
delete(a.storageChangeIndex, address)
|
|
}
|
|
accountChange := a.latestChange.Changes[i]
|
|
if accountChange.Action == remote.Action_REMOVE {
|
|
panic("")
|
|
}
|
|
accountChange.Incarnation = incarnation
|
|
si, ok1 := a.storageChangeIndex[address]
|
|
if !ok1 {
|
|
si = make(map[libcommon.Hash]int)
|
|
a.storageChangeIndex[address] = si
|
|
}
|
|
j, ok2 := si[location]
|
|
if !ok2 {
|
|
j = len(accountChange.StorageChanges)
|
|
accountChange.StorageChanges = append(accountChange.StorageChanges, &remote.StorageChange{})
|
|
si[location] = j
|
|
}
|
|
storageChange := accountChange.StorageChanges[j]
|
|
storageChange.Location = gointerfaces.ConvertHashToH256(location)
|
|
storageChange.Data = data
|
|
}
|
|
|
|
func (a *Accumulator) CopyAndReset(target *Accumulator) {
|
|
target.changes = a.changes
|
|
a.changes = nil
|
|
target.latestChange = a.latestChange
|
|
a.latestChange = nil
|
|
target.accountChangeIndex = a.accountChangeIndex
|
|
a.accountChangeIndex = nil
|
|
target.storageChangeIndex = a.storageChangeIndex
|
|
a.storageChangeIndex = nil
|
|
}
|