From fbc31dc99bd055b756d7895c5da775e61695e022 Mon Sep 17 00:00:00 2001 From: Preston Van Loon Date: Wed, 30 Oct 2019 18:30:10 -0400 Subject: [PATCH] validator client side grpc metrics (#3903) --- validator/client/BUILD.bazel | 1 + validator/client/service.go | 3 +++ 2 files changed, 4 insertions(+) diff --git a/validator/client/BUILD.bazel b/validator/client/BUILD.bazel index 0ed8887760..93569b576b 100644 --- a/validator/client/BUILD.bazel +++ b/validator/client/BUILD.bazel @@ -24,6 +24,7 @@ go_library( "@com_github_gogo_protobuf//types:go_default_library", "@com_github_grpc_ecosystem_go_grpc_middleware//:go_default_library", "@com_github_grpc_ecosystem_go_grpc_middleware//tracing/opentracing:go_default_library", + "@com_github_grpc_ecosystem_go_grpc_prometheus//:go_default_library", "@com_github_pkg_errors//:go_default_library", "@com_github_prysmaticlabs_go_bitfield//:go_default_library", "@com_github_prysmaticlabs_go_ssz//:go_default_library", diff --git a/validator/client/service.go b/validator/client/service.go index 298a30c279..04b8e3373a 100644 --- a/validator/client/service.go +++ b/validator/client/service.go @@ -6,6 +6,7 @@ import ( middleware "github.com/grpc-ecosystem/go-grpc-middleware" grpc_opentracing "github.com/grpc-ecosystem/go-grpc-middleware/tracing/opentracing" + grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus" "github.com/pkg/errors" pb "github.com/prysmaticlabs/prysm/proto/beacon/rpc/v1" "github.com/prysmaticlabs/prysm/shared/bytesutil" @@ -81,9 +82,11 @@ func (v *ValidatorService) Start() { grpc.WithStatsHandler(&ocgrpc.ClientHandler{}), grpc.WithStreamInterceptor(middleware.ChainStreamClient( grpc_opentracing.StreamClientInterceptor(), + grpc_prometheus.StreamClientInterceptor, )), grpc.WithUnaryInterceptor(middleware.ChainUnaryClient( grpc_opentracing.UnaryClientInterceptor(), + grpc_prometheus.UnaryClientInterceptor, )), } conn, err := grpc.DialContext(v.ctx, v.endpoint, opts...)