use grpc.WithStatsHandler(otelgrpc.NewClientHandler ...

Signed-off-by: Jörn Friedrich Dreyer <[email protected]>
cs3org · Feb 8, 2024 · 515dad9 · 515dad9
1 parent 604b101
commit 515dad9
Showing 1 changed file with 1 addition and 11 deletions.
diff --git a/pkg/rgrpc/todo/pool/connection.go b/pkg/rgrpc/todo/pool/connection.go
@@ -90,24 +90,14 @@ func NewConn(address string, opts ...Option) (*grpc.ClientConn, error) {
 		grpc.WithDefaultCallOptions(
 			grpc.MaxCallRecvMsgSize(maxRcvMsgSize),
 		),
-		grpc.WithStreamInterceptor(otelgrpc.StreamClientInterceptor(
+		grpc.WithStatsHandler(otelgrpc.NewClientHandler(
 			otelgrpc.WithTracerProvider(
 				options.tracerProvider,
 			),
 			otelgrpc.WithPropagators(
 				rtrace.Propagator,
 			),
 		)),
-		grpc.WithUnaryInterceptor(
-			otelgrpc.UnaryClientInterceptor(
-				otelgrpc.WithTracerProvider(
-					options.tracerProvider,
-				),
-				otelgrpc.WithPropagators(
-					rtrace.Propagator,
-				),
-			),
-		),
 	)
 	if err != nil {
 		return nil, err