fix(grpc): Chain interceptor (#296)

* fix(grpc): Chain interceptor * fix(grpc): Chain interceptor * fix(grpc): Chain interceptor
ethpandaops · Apr 8, 2024 · 59711a6 · 59711a6
1 parent fdcc570
commit 59711a6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pkg/output/xatu/exporter.go b/pkg/output/xatu/exporter.go
@@ -31,10 +31,8 @@ type ItemExporter struct {
 
 func NewItemExporter(name string, config *Config, log logrus.FieldLogger) (ItemExporter, error) {
 	opts := []grpc.DialOption{
-		grpc.WithUnaryInterceptor(grpc_prometheus.UnaryClientInterceptor),
-		grpc.WithStreamInterceptor(grpc_prometheus.StreamClientInterceptor),
-		grpc.WithStreamInterceptor(retry.StreamClientInterceptor()),
-		grpc.WithUnaryInterceptor(retry.UnaryClientInterceptor()),
+		grpc.WithChainUnaryInterceptor(grpc_prometheus.UnaryClientInterceptor, retry.UnaryClientInterceptor()),
+		grpc.WithChainStreamInterceptor(grpc_prometheus.StreamClientInterceptor, retry.StreamClientInterceptor()),
 	}
 
 	if config.TLS {

diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -227,8 +227,6 @@ func (x *Xatu) startGrpcServer(ctx context.Context) error {
 	grpc_prometheus.EnableHandlingTimeHistogram()
 
 	opts := []grpc.ServerOption{
-		grpc.StreamInterceptor(grpc_prometheus.StreamServerInterceptor),
-		grpc.UnaryInterceptor(grpc_prometheus.UnaryServerInterceptor),
 		grpc.MaxRecvMsgSize(mb100),
 		grpc.KeepaliveParams(keepalive.ServerParameters{
 			MaxConnectionIdle:     5 * time.Minute,
@@ -237,7 +235,11 @@ func (x *Xatu) startGrpcServer(ctx context.Context) error {
 			Time:                  1 * time.Minute,
 			Timeout:               15 * time.Second,
 		}),
+		grpc.ChainStreamInterceptor(
+			grpc.StreamServerInterceptor(grpc_prometheus.StreamServerInterceptor),
+		),
 		grpc.ChainUnaryInterceptor(
+			grpc.UnaryServerInterceptor(grpc_prometheus.UnaryServerInterceptor),
 			func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
 				resp, err := handler(ctx, req)
 				if err != nil {