mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
3cf385fe91
* unrealized justification API * Add time elapse logging * add unrealized justification checkpoint * Use UnrealizedJustificationCheckpoint * Refactor unrealized checkpoints * Move logic to state package * do not use ctx on a sum * fix ctx * add tests * fix conflicts * unhandled error * Fix ordering in computing checkpoints * gaz * keep finalized checkpoint if nothing justified * gaz * copy checkpoint * fix check for nil * Add state package tests * Add tests * Radek's review * add more tests * Update beacon-chain/core/epoch/precompute/justification_finalization.go Co-authored-by: terencechain <terence@prysmaticlabs.com> * deduplicate to stateutil * missing file * Add stateutil test * Minor refactor, don't export certain things * Fix exports in tests * remove unused error Co-authored-by: terence tsao <terence@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
package v2
|
|
|
|
import (
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/time"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stateutil"
|
|
)
|
|
|
|
// CurrentEpochParticipation corresponding to participation bits on the beacon chain.
|
|
func (b *BeaconState) CurrentEpochParticipation() ([]byte, error) {
|
|
if !b.hasInnerState() {
|
|
return nil, ErrNilInnerState
|
|
}
|
|
if b.state.CurrentEpochParticipation == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
b.lock.RLock()
|
|
defer b.lock.RUnlock()
|
|
|
|
return b.currentEpochParticipation(), nil
|
|
}
|
|
|
|
// PreviousEpochParticipation corresponding to participation bits on the beacon chain.
|
|
func (b *BeaconState) PreviousEpochParticipation() ([]byte, error) {
|
|
if !b.hasInnerState() {
|
|
return nil, ErrNilInnerState
|
|
}
|
|
if b.state.PreviousEpochParticipation == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
b.lock.RLock()
|
|
defer b.lock.RUnlock()
|
|
|
|
return b.previousEpochParticipation(), nil
|
|
}
|
|
|
|
// UnrealizedCheckpointBalances returns the total balances: active, target attested in
|
|
// current epoch and target attested in previous epoch. This function is used to
|
|
// compute the "unrealized justification" that a synced Beacon Block will have.
|
|
func (b *BeaconState) UnrealizedCheckpointBalances() (uint64, uint64, uint64, error) {
|
|
if !b.hasInnerState() {
|
|
return 0, 0, 0, ErrNilInnerState
|
|
}
|
|
b.lock.RLock()
|
|
defer b.lock.RUnlock()
|
|
|
|
cp := b.state.CurrentEpochParticipation
|
|
pp := b.state.PreviousEpochParticipation
|
|
if cp == nil || pp == nil {
|
|
return 0, 0, 0, ErrNilParticipation
|
|
}
|
|
currentEpoch := time.CurrentEpoch(b)
|
|
return stateutil.UnrealizedCheckpointBalances(cp, pp, b.state.Validators, currentEpoch)
|
|
|
|
}
|
|
|
|
// currentEpochParticipation corresponding to participation bits on the beacon chain.
|
|
// This assumes that a lock is already held on BeaconState.
|
|
func (b *BeaconState) currentEpochParticipation() []byte {
|
|
if !b.hasInnerState() {
|
|
return nil
|
|
}
|
|
tmp := make([]byte, len(b.state.CurrentEpochParticipation))
|
|
copy(tmp, b.state.CurrentEpochParticipation)
|
|
return tmp
|
|
}
|
|
|
|
// previousEpochParticipation corresponding to participation bits on the beacon chain.
|
|
// This assumes that a lock is already held on BeaconState.
|
|
func (b *BeaconState) previousEpochParticipation() []byte {
|
|
if !b.hasInnerState() {
|
|
return nil
|
|
}
|
|
tmp := make([]byte, len(b.state.PreviousEpochParticipation))
|
|
copy(tmp, b.state.PreviousEpochParticipation)
|
|
return tmp
|
|
}
|