mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-08 18:51:19 +00:00
74 lines
2.6 KiB
Go
74 lines
2.6 KiB
Go
// Package precompute provides gathering of nicely-structured
|
|
// data important to feed into epoch processing, such as attesting
|
|
// records and balances, for faster computation.
|
|
package precompute
|
|
|
|
import (
|
|
"context"
|
|
"math/big"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v5/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/v5/beacon-chain/core/time"
|
|
"github.com/prysmaticlabs/prysm/v5/beacon-chain/state"
|
|
"github.com/prysmaticlabs/prysm/v5/config/params"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// New gets called at the beginning of process epoch cycle to return
|
|
// pre computed instances of validators attesting records and total
|
|
// balances attested in an epoch.
|
|
func New(ctx context.Context, s state.BeaconState) ([]*Validator, *Balance, error) {
|
|
_, span := trace.StartSpan(ctx, "precomputeEpoch.New")
|
|
defer span.End()
|
|
|
|
pValidators := make([]*Validator, s.NumValidators())
|
|
pBal := NewBalance()
|
|
|
|
currentEpoch := time.CurrentEpoch(s)
|
|
prevEpoch := time.PrevEpoch(s)
|
|
|
|
if err := s.ReadFromEveryValidator(func(idx int, val state.ReadOnlyValidator) error {
|
|
// Was validator withdrawable or slashed
|
|
withdrawable := prevEpoch+1 >= val.WithdrawableEpoch()
|
|
effectiveBalance := new(big.Int).SetUint64(val.EffectiveBalance())
|
|
pVal := &Validator{
|
|
IsSlashed: val.Slashed(),
|
|
IsWithdrawableCurrentEpoch: withdrawable,
|
|
CurrentEpochEffectiveBalance: val.EffectiveBalance(),
|
|
}
|
|
// Was validator active current epoch
|
|
if helpers.IsActiveValidatorUsingTrie(val, currentEpoch) {
|
|
pVal.IsActiveCurrentEpoch = true
|
|
pBal.ActiveCurrentEpoch.Add(pBal.ActiveCurrentEpoch, effectiveBalance)
|
|
}
|
|
// Was validator active previous epoch
|
|
if helpers.IsActiveValidatorUsingTrie(val, prevEpoch) {
|
|
pVal.IsActivePrevEpoch = true
|
|
pBal.ActivePrevEpoch.Add(pBal.ActivePrevEpoch, effectiveBalance)
|
|
}
|
|
// Set inclusion slot and inclusion distance to be max, they will be compared and replaced
|
|
// with the lower values
|
|
pVal.InclusionSlot = params.BeaconConfig().FarFutureSlot
|
|
pVal.InclusionDistance = params.BeaconConfig().FarFutureSlot
|
|
|
|
pValidators[idx] = pVal
|
|
return nil
|
|
}); err != nil {
|
|
return nil, nil, errors.Wrap(err, "failed to initialize precompute")
|
|
}
|
|
return pValidators, pBal, nil
|
|
}
|
|
|
|
func NewBalance() *Balance {
|
|
return &Balance{
|
|
ActiveCurrentEpoch: big.NewInt(0),
|
|
ActivePrevEpoch: big.NewInt(0),
|
|
CurrentEpochAttested: big.NewInt(0),
|
|
CurrentEpochTargetAttested: big.NewInt(0),
|
|
PrevEpochAttested: big.NewInt(0),
|
|
PrevEpochHeadAttested: big.NewInt(0),
|
|
PrevEpochTargetAttested: big.NewInt(0),
|
|
}
|
|
}
|