Export client UnaryInterceptors

This commits exports a function that returns the unary interceptors
that should be configured for a client so they can be used in other
dialing code as well.

1 files changed, 9 insertions, 4 deletions

diff --git a/internal/gitaly/client/dial.go b/internal/gitaly/client/dial.go
index 5705d9d14..5b6fe226d 100644
--- a/internal/gitaly/client/dial.go
+++ b/internal/gitaly/client/dial.go
@@ -127,10 +127,7 @@ func Dial(ctx context.Context, rawAddress string, connOpts []grpc.DialOption, ha
 			Time:                20 * time.Second,
 			PermitWithoutStream: true,
 		}),
-		grpc.WithChainUnaryInterceptor(
-			grpctracing.UnaryClientTracingInterceptor(),
-			grpccorrelation.UnaryClientCorrelationInterceptor(),
-		),
+		UnaryInterceptor(),
 		grpc.WithChainStreamInterceptor(
 			grpctracing.StreamClientTracingInterceptor(),
 			grpccorrelation.StreamClientCorrelationInterceptor(),
@@ -144,3 +141,11 @@ func Dial(ctx context.Context, rawAddress string, connOpts []grpc.DialOption, ha
 
 	return conn, nil
 }
+
+// UnaryInterceptor returns the unary interceptors that should be configured for a client.
+func UnaryInterceptor() grpc.DialOption {
+	return grpc.WithChainUnaryInterceptor(
+		grpctracing.UnaryClientTracingInterceptor(),
+		grpccorrelation.UnaryClientCorrelationInterceptor(),
+	)
+}