export WithUnaryClientInterceptor

zrpc/client.go

@@ -11,8 +11,9 @@ import (
 )
 
 var (
-	WithDialOption = internal.WithDialOption
+	WithDialOption             = internal.WithDialOption
-	WithTimeout    = internal.WithTimeout
+	WithTimeout                = internal.WithTimeout
+	WithUnaryClientInterceptor = internal.WithUnaryClientInterceptor
 )
 
 type (

zrpc/client_test.go

@@ -60,14 +60,26 @@ func TestDepositServer_Deposit(t *testing.T) {
 		},
 	}
 
-	directClient := MustNewClient(RpcClientConf{
+	directClient := MustNewClient(
-		Endpoints: []string{"foo"},
+		RpcClientConf{
-		App:       "foo",
+			Endpoints: []string{"foo"},
-		Token:     "bar",
+			App:       "foo",
-		Timeout:   1000,
+			Token:     "bar",
-	}, WithDialOption(grpc.WithInsecure()), WithDialOption(grpc.WithContextDialer(dialer())))
+			Timeout:   1000,
+		},
+		WithDialOption(grpc.WithInsecure()),
+		WithDialOption(grpc.WithContextDialer(dialer())),
+		WithUnaryClientInterceptor(func(ctx context.Context, method string, req, reply interface{},
+			cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
+			return invoker(ctx, method, req, reply, cc, opts...)
+		}),
+	)
 	targetClient, err := NewClientWithTarget("foo", WithDialOption(grpc.WithInsecure()),
-		WithDialOption(grpc.WithContextDialer(dialer())))
+		WithDialOption(grpc.WithContextDialer(dialer())), WithUnaryClientInterceptor(
+			func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn,
+				invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
+				return invoker(ctx, method, req, reply, cc, opts...)
+			}))
 	assert.Nil(t, err)
 	clients := []Client{
 		directClient,