mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 21:38:05 -05:00
2f090c52d9
* Allow custom headers in validator client HTTP requests * changelog <3 * improve flag description * Bastin's review * James' review * add godoc for NodeConnectionOption
71 lines
2.7 KiB
Go
71 lines
2.7 KiB
Go
package rpc
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
api "github.com/OffchainLabs/prysm/v6/api/client"
|
|
grpcutil "github.com/OffchainLabs/prysm/v6/api/grpc"
|
|
ethpb "github.com/OffchainLabs/prysm/v6/proto/prysm/v1alpha1"
|
|
"github.com/OffchainLabs/prysm/v6/validator/client"
|
|
beaconApi "github.com/OffchainLabs/prysm/v6/validator/client/beacon-api"
|
|
beaconChainClientFactory "github.com/OffchainLabs/prysm/v6/validator/client/beacon-chain-client-factory"
|
|
nodeClientFactory "github.com/OffchainLabs/prysm/v6/validator/client/node-client-factory"
|
|
validatorClientFactory "github.com/OffchainLabs/prysm/v6/validator/client/validator-client-factory"
|
|
validatorHelpers "github.com/OffchainLabs/prysm/v6/validator/helpers"
|
|
middleware "github.com/grpc-ecosystem/go-grpc-middleware"
|
|
grpcretry "github.com/grpc-ecosystem/go-grpc-middleware/retry"
|
|
grpcopentracing "github.com/grpc-ecosystem/go-grpc-middleware/tracing/opentracing"
|
|
grpcprometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
|
|
"github.com/pkg/errors"
|
|
"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// Initialize a client connect to a beacon node gRPC or HTTP endpoint.
|
|
func (s *Server) registerBeaconClient() error {
|
|
streamInterceptor := grpc.WithStreamInterceptor(middleware.ChainStreamClient(
|
|
grpcopentracing.StreamClientInterceptor(),
|
|
grpcprometheus.StreamClientInterceptor,
|
|
grpcretry.StreamClientInterceptor(),
|
|
))
|
|
dialOpts := client.ConstructDialOptions(
|
|
s.grpcMaxCallRecvMsgSize,
|
|
s.beaconNodeCert,
|
|
s.grpcRetries,
|
|
s.grpcRetryDelay,
|
|
streamInterceptor,
|
|
)
|
|
if dialOpts == nil {
|
|
return errors.New("no dial options for beacon chain gRPC client")
|
|
}
|
|
|
|
s.ctx = grpcutil.AppendHeaders(s.ctx, s.grpcHeaders)
|
|
|
|
grpcConn, err := grpc.DialContext(s.ctx, s.beaconNodeEndpoint, dialOpts...)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "could not dial endpoint: %s", s.beaconNodeEndpoint)
|
|
}
|
|
if s.beaconNodeCert != "" {
|
|
log.Info("Established secure gRPC connection")
|
|
}
|
|
s.healthClient = ethpb.NewHealthClient(grpcConn)
|
|
|
|
conn := validatorHelpers.NewNodeConnection(
|
|
grpcConn,
|
|
s.beaconApiEndpoint,
|
|
validatorHelpers.WithBeaconApiHeaders(s.beaconApiHeaders),
|
|
validatorHelpers.WithBeaconApiTimeout(s.beaconApiTimeout),
|
|
)
|
|
|
|
headersTransport := api.NewCustomHeadersTransport(http.DefaultTransport, conn.GetBeaconApiHeaders())
|
|
restHandler := beaconApi.NewBeaconApiRestHandler(
|
|
http.Client{Timeout: s.beaconApiTimeout, Transport: otelhttp.NewTransport(headersTransport)},
|
|
s.beaconApiEndpoint,
|
|
)
|
|
|
|
s.chainClient = beaconChainClientFactory.NewChainClient(conn, restHandler)
|
|
s.nodeClient = nodeClientFactory.NewNodeClient(conn, restHandler)
|
|
s.beaconNodeValidatorClient = validatorClientFactory.NewValidatorClient(conn, restHandler)
|
|
return nil
|
|
}
|