mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
699bfdfdb4
* Rename pow to execution * Fix complain * Fix complain
126 lines
3.7 KiB
Go
126 lines
3.7 KiB
Go
package execution
|
|
|
|
import (
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"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/state"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stategen"
|
|
"github.com/prysmaticlabs/prysm/network"
|
|
"github.com/prysmaticlabs/prysm/network/authorization"
|
|
)
|
|
|
|
type Option func(s *Service) error
|
|
|
|
// WithHttpEndpoints deduplicates and parses http endpoints for the execution chain service to use,
|
|
// and sets the "current" endpoint that will be used first.
|
|
func WithHttpEndpoints(endpointStrings []string) Option {
|
|
return func(s *Service) error {
|
|
stringEndpoints := dedupEndpoints(endpointStrings)
|
|
endpoints := make([]network.Endpoint, len(stringEndpoints))
|
|
for i, e := range stringEndpoints {
|
|
endpoints[i] = HttpEndpoint(e)
|
|
}
|
|
// Select first http endpoint in the provided list.
|
|
var currEndpoint network.Endpoint
|
|
if len(endpointStrings) > 0 {
|
|
currEndpoint = endpoints[0]
|
|
}
|
|
s.cfg.httpEndpoints = endpoints
|
|
s.cfg.currHttpEndpoint = currEndpoint
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithHttpEndpointsAndJWTSecret for authenticating the execution node JSON-RPC endpoint.
|
|
func WithHttpEndpointsAndJWTSecret(endpointStrings []string, secret []byte) Option {
|
|
return func(s *Service) error {
|
|
if len(secret) == 0 {
|
|
return nil
|
|
}
|
|
stringEndpoints := dedupEndpoints(endpointStrings)
|
|
endpoints := make([]network.Endpoint, len(stringEndpoints))
|
|
// Overwrite authorization type for all endpoints to be of a bearer
|
|
// type.
|
|
for i, e := range stringEndpoints {
|
|
hEndpoint := HttpEndpoint(e)
|
|
hEndpoint.Auth.Method = authorization.Bearer
|
|
hEndpoint.Auth.Value = string(secret)
|
|
endpoints[i] = hEndpoint
|
|
}
|
|
// Select first http endpoint in the provided list.
|
|
var currEndpoint network.Endpoint
|
|
if len(endpointStrings) > 0 {
|
|
currEndpoint = endpoints[0]
|
|
}
|
|
s.cfg.httpEndpoints = endpoints
|
|
s.cfg.currHttpEndpoint = currEndpoint
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithDepositContractAddress for the deposit contract.
|
|
func WithDepositContractAddress(addr common.Address) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.depositContractAddr = addr
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithDatabase for the beacon chain database.
|
|
func WithDatabase(database db.HeadAccessDatabase) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.beaconDB = database
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithDepositCache for caching deposits.
|
|
func WithDepositCache(cache *depositcache.DepositCache) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.depositCache = cache
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithStateNotifier for subscribing to state changes.
|
|
func WithStateNotifier(notifier statefeed.Notifier) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.stateNotifier = notifier
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithStateGen to regenerate beacon states from checkpoints.
|
|
func WithStateGen(gen *stategen.State) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.stateGen = gen
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithEth1HeaderRequestLimit to set the upper limit of eth1 header requests.
|
|
func WithEth1HeaderRequestLimit(limit uint64) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.eth1HeaderReqLimit = limit
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithBeaconNodeStatsUpdater to set the beacon node stats updater.
|
|
func WithBeaconNodeStatsUpdater(updater BeaconNodeStatsUpdater) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.beaconNodeStatsUpdater = updater
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithFinalizedStateAtStartup to set the beacon node's finalized state at startup.
|
|
func WithFinalizedStateAtStartup(st state.BeaconState) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.finalizedStateAtStartup = st
|
|
return nil
|
|
}
|
|
}
|