mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 13:28:01 -05:00
15ae71c0da
* wip * refactoring functions for easier readability * allow ssz for register validator * changelog * adding in blinded block submission tests * adding in tests for header ssz * fixing linting and tests * adding in custom errors and fixing ssz validator registration * Update api/client/builder/client.go Co-authored-by: Manu NALEPA <enalepa@offchainlabs.com> * Update api/client/builder/client.go Co-authored-by: Manu NALEPA <enalepa@offchainlabs.com> * manu's feedback * linting * adding in info log to notify that this setting was turned on * fixing linting * manu's feedback * fixing import --------- Co-authored-by: Manu NALEPA <enalepa@offchainlabs.com>
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package builder
|
|
|
|
import (
|
|
"github.com/prysmaticlabs/prysm/v5/api/client/builder"
|
|
"github.com/prysmaticlabs/prysm/v5/beacon-chain/blockchain"
|
|
"github.com/prysmaticlabs/prysm/v5/beacon-chain/cache"
|
|
"github.com/prysmaticlabs/prysm/v5/beacon-chain/db"
|
|
"github.com/prysmaticlabs/prysm/v5/cmd/beacon-chain/flags"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
type Option func(s *Service) error
|
|
|
|
// FlagOptions for builder service flag configurations.
|
|
func FlagOptions(c *cli.Context) ([]Option, error) {
|
|
endpoint := c.String(flags.MevRelayEndpoint.Name)
|
|
sszEnabled := c.Bool(flags.EnableBuilderSSZ.Name)
|
|
var client *builder.Client
|
|
if endpoint != "" {
|
|
var opts []builder.ClientOpt
|
|
if sszEnabled {
|
|
log.Info("Using APIs with SSZ enabled")
|
|
opts = append(opts, builder.WithSSZ())
|
|
}
|
|
var err error
|
|
client, err = builder.NewClient(endpoint, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
opts := []Option{
|
|
WithBuilderClient(client),
|
|
}
|
|
return opts, nil
|
|
}
|
|
|
|
// WithBuilderClient sets the builder client for the beacon chain builder service.
|
|
func WithBuilderClient(client builder.BuilderClient) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.builderClient = client
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithHeadFetcher gets the head info from chain service.
|
|
func WithHeadFetcher(svc blockchain.HeadFetcher) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.headFetcher = svc
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithDatabase for head access.
|
|
func WithDatabase(beaconDB db.HeadAccessDatabase) Option {
|
|
return func(s *Service) error {
|
|
s.cfg.beaconDB = beaconDB
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithRegistrationCache uses a cache for the validator registrations instead of a persistent db.
|
|
func WithRegistrationCache() Option {
|
|
return func(s *Service) error {
|
|
s.registrationCache = cache.NewRegistrationCache()
|
|
return nil
|
|
}
|
|
}
|