mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
cde58f6924
* Update configs * gazelle Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
156 lines
4.2 KiB
Go
156 lines
4.2 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"github.com/prysmaticlabs/prysm/async/event"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/cache/depositcache"
|
|
statefeed "github.com/prysmaticlabs/prysm/beacon-chain/core/feed/state"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/forkchoice"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/operations/attestations"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/operations/slashings"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/operations/voluntaryexits"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/p2p"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/powchain"
|
|
enginev1 "github.com/prysmaticlabs/prysm/beacon-chain/powchain/engine-api-client/v1"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stategen"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
)
|
|
|
|
type Option func(s *Service) error
|
|
|
|
// WithMaxGoroutines to control resource use of the blockchain service.
|
|
func WithMaxGoroutines(x int) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.MaxRoutines = x
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithWeakSubjectivityCheckpoint for checkpoint sync.
|
|
func WithWeakSubjectivityCheckpoint(c *ethpb.Checkpoint) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.WeakSubjectivityCheckpt = c
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithDatabase for head access.
|
|
func WithDatabase(beaconDB db.HeadAccessDatabase) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.BeaconDB = beaconDB
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithChainStartFetcher to retrieve information about genesis.
|
|
func WithChainStartFetcher(f powchain.ChainStartFetcher) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.ChainStartFetcher = f
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithExecutionEngineCaller to call execution engine.
|
|
func WithExecutionEngineCaller(c enginev1.Caller) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.ExecutionEngineCaller = c
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithDepositCache for deposit lifecycle after chain inclusion.
|
|
func WithDepositCache(c *depositcache.DepositCache) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.DepositCache = c
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithAttestationPool for attestation lifecycle after chain inclusion.
|
|
func WithAttestationPool(p attestations.Pool) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.AttPool = p
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithExitPool for exits lifecycle after chain inclusion.
|
|
func WithExitPool(p voluntaryexits.PoolManager) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.ExitPool = p
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithSlashingPool for slashings lifecycle after chain inclusion.
|
|
func WithSlashingPool(p slashings.PoolManager) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.SlashingPool = p
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithP2PBroadcaster to broadcast messages after appropriate processing.
|
|
func WithP2PBroadcaster(p p2p.Broadcaster) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.P2p = p
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithStateNotifier to notify an event feed of state processing.
|
|
func WithStateNotifier(n statefeed.Notifier) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.StateNotifier = n
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithForkChoiceStore to update an optimized fork-choice representation.
|
|
func WithForkChoiceStore(f forkchoice.ForkChoicer) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.ForkChoiceStore = f
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithAttestationService for dealing with attestation lifecycles.
|
|
func WithAttestationService(srv *attestations.Service) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.AttService = srv
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithStateGen for managing state regeneration and replay.
|
|
func WithStateGen(g *stategen.State) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.StateGen = g
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithSlasherAttestationsFeed to forward attestations into slasher if enabled.
|
|
func WithSlasherAttestationsFeed(f *event.Feed) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.SlasherAttestationsFeed = f
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func withStateBalanceCache(c *stateBalanceCache) Option {
|
|
return func(s *Service) error {
|
|
s.justifiedBalances = c
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithFinalizedStateAtStartUp to store finalized state at start up.
|
|
func WithFinalizedStateAtStartUp(st state.BeaconState) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.FinalizedStateAtStartUp = st
|
|
return nil
|
|
}
|
|
}
|