mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-24 12:27:18 +00:00
d077483577
* v3 import renamings * tidy * fmt * rev * Update beacon-chain/core/epoch/precompute/reward_penalty_test.go * Update beacon-chain/core/helpers/validators_test.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/iface/BUILD.bazel * Update beacon-chain/db/kv/kv.go * Update beacon-chain/db/kv/state.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/sync/initial-sync/service.go * fix deps * fix bad replacements * fix bad replacements * change back * gohashtree version * fix deps Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Potuz <potuz@prysmaticlabs.com>
156 lines
5.3 KiB
Go
156 lines
5.3 KiB
Go
package v2
|
|
|
|
import (
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/state/stateutil"
|
|
)
|
|
|
|
// SetPreviousParticipationBits for the beacon state. Updates the entire
|
|
// list to a new value by overwriting the previous one.
|
|
func (b *BeaconState) SetPreviousParticipationBits(val []byte) error {
|
|
if !b.hasInnerState() {
|
|
return ErrNilInnerState
|
|
}
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
b.sharedFieldReferences[previousEpochParticipationBits].MinusRef()
|
|
b.sharedFieldReferences[previousEpochParticipationBits] = stateutil.NewRef(1)
|
|
|
|
b.state.PreviousEpochParticipation = val
|
|
b.markFieldAsDirty(previousEpochParticipationBits)
|
|
b.rebuildTrie[previousEpochParticipationBits] = true
|
|
return nil
|
|
}
|
|
|
|
// SetCurrentParticipationBits for the beacon state. Updates the entire
|
|
// list to a new value by overwriting the previous one.
|
|
func (b *BeaconState) SetCurrentParticipationBits(val []byte) error {
|
|
if !b.hasInnerState() {
|
|
return ErrNilInnerState
|
|
}
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
b.sharedFieldReferences[currentEpochParticipationBits].MinusRef()
|
|
b.sharedFieldReferences[currentEpochParticipationBits] = stateutil.NewRef(1)
|
|
|
|
b.state.CurrentEpochParticipation = val
|
|
b.markFieldAsDirty(currentEpochParticipationBits)
|
|
b.rebuildTrie[currentEpochParticipationBits] = true
|
|
return nil
|
|
}
|
|
|
|
// AppendCurrentParticipationBits for the beacon state. Appends the new value
|
|
// to the the end of list.
|
|
func (b *BeaconState) AppendCurrentParticipationBits(val byte) error {
|
|
if !b.hasInnerState() {
|
|
return ErrNilInnerState
|
|
}
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
participation := b.state.CurrentEpochParticipation
|
|
if b.sharedFieldReferences[currentEpochParticipationBits].Refs() > 1 {
|
|
// Copy elements in underlying array by reference.
|
|
participation = make([]byte, len(b.state.CurrentEpochParticipation))
|
|
copy(participation, b.state.CurrentEpochParticipation)
|
|
b.sharedFieldReferences[currentEpochParticipationBits].MinusRef()
|
|
b.sharedFieldReferences[currentEpochParticipationBits] = stateutil.NewRef(1)
|
|
}
|
|
|
|
b.state.CurrentEpochParticipation = append(participation, val)
|
|
b.markFieldAsDirty(currentEpochParticipationBits)
|
|
b.addDirtyIndices(currentEpochParticipationBits, []uint64{uint64(len(b.state.CurrentEpochParticipation) - 1)})
|
|
return nil
|
|
}
|
|
|
|
// AppendPreviousParticipationBits for the beacon state. Appends the new value
|
|
// to the the end of list.
|
|
func (b *BeaconState) AppendPreviousParticipationBits(val byte) error {
|
|
if !b.hasInnerState() {
|
|
return ErrNilInnerState
|
|
}
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
bits := b.state.PreviousEpochParticipation
|
|
if b.sharedFieldReferences[previousEpochParticipationBits].Refs() > 1 {
|
|
bits = make([]byte, len(b.state.PreviousEpochParticipation))
|
|
copy(bits, b.state.PreviousEpochParticipation)
|
|
b.sharedFieldReferences[previousEpochParticipationBits].MinusRef()
|
|
b.sharedFieldReferences[previousEpochParticipationBits] = stateutil.NewRef(1)
|
|
}
|
|
|
|
b.state.PreviousEpochParticipation = append(bits, val)
|
|
b.markFieldAsDirty(previousEpochParticipationBits)
|
|
b.addDirtyIndices(previousEpochParticipationBits, []uint64{uint64(len(b.state.PreviousEpochParticipation) - 1)})
|
|
|
|
return nil
|
|
}
|
|
|
|
// ModifyPreviousParticipationBits modifies the previous participation bitfield via
|
|
// the provided mutator function.
|
|
func (b *BeaconState) ModifyPreviousParticipationBits(mutator func(val []byte) ([]byte, error)) error {
|
|
if !b.hasInnerState() {
|
|
return ErrNilInnerState
|
|
}
|
|
b.lock.Lock()
|
|
|
|
participation := b.state.PreviousEpochParticipation
|
|
if b.sharedFieldReferences[previousEpochParticipationBits].Refs() > 1 {
|
|
// Copy elements in underlying array by reference.
|
|
participation = make([]byte, len(b.state.PreviousEpochParticipation))
|
|
copy(participation, b.state.PreviousEpochParticipation)
|
|
b.sharedFieldReferences[previousEpochParticipationBits].MinusRef()
|
|
b.sharedFieldReferences[previousEpochParticipationBits] = stateutil.NewRef(1)
|
|
}
|
|
// Lock is released so that mutator can
|
|
// acquire it.
|
|
b.lock.Unlock()
|
|
|
|
var err error
|
|
participation, err = mutator(participation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
b.state.PreviousEpochParticipation = participation
|
|
b.markFieldAsDirty(previousEpochParticipationBits)
|
|
b.rebuildTrie[previousEpochParticipationBits] = true
|
|
return nil
|
|
}
|
|
|
|
// ModifyCurrentParticipationBits modifies the current participation bitfield via
|
|
// the provided mutator function.
|
|
func (b *BeaconState) ModifyCurrentParticipationBits(mutator func(val []byte) ([]byte, error)) error {
|
|
if !b.hasInnerState() {
|
|
return ErrNilInnerState
|
|
}
|
|
b.lock.Lock()
|
|
|
|
participation := b.state.CurrentEpochParticipation
|
|
if b.sharedFieldReferences[currentEpochParticipationBits].Refs() > 1 {
|
|
// Copy elements in underlying array by reference.
|
|
participation = make([]byte, len(b.state.CurrentEpochParticipation))
|
|
copy(participation, b.state.CurrentEpochParticipation)
|
|
b.sharedFieldReferences[currentEpochParticipationBits].MinusRef()
|
|
b.sharedFieldReferences[currentEpochParticipationBits] = stateutil.NewRef(1)
|
|
}
|
|
// Lock is released so that mutator can
|
|
// acquire it.
|
|
b.lock.Unlock()
|
|
|
|
var err error
|
|
participation, err = mutator(participation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
b.state.CurrentEpochParticipation = participation
|
|
b.markFieldAsDirty(currentEpochParticipationBits)
|
|
b.rebuildTrie[currentEpochParticipationBits] = true
|
|
return nil
|
|
}
|