zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Revert "Revert "feat: enable retry for zrpc (#1237)"" (#1246)

Browse Source
This commit is contained in:
Kevin Wan
2021-11-16 10:29:31 +08:00
committed by GitHub
parent 67db40ed4f
commit 2d4c29ea7c
4 changed files with 26 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
zrpc/client_test.go
View File

@@ -96,21 +96,20 @@ func TestDepositServer_Deposit(t *testing.T) {
return invoker(ctx, method, req, reply, cc, opts...)
}),
)
// TODO: enable it in v1.2.4
// retryClient := MustNewClient(
// RpcClientConf{
// Endpoints: []string{"foo"},
// App: "foo",
// Token: "bar",
// Timeout: 1000,
// Retry: true,
// },
// 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...)
// }),
// )
retryClient := MustNewClient(
RpcClientConf{
Endpoints: []string{"foo"},
App: "foo",
Token: "bar",
Timeout: 1000,
Retry: true,
},
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...)
}),
)
tarConfClient := MustNewClient(
RpcClientConf{
Target: "foo",
@@ -135,8 +134,7 @@ func TestDepositServer_Deposit(t *testing.T) {
clients := []Client{
directClient,
nonBlockClient,
// TODO: enable it in v1.2.4
// retryClient,
retryClient,
tarConfClient,
targetClient,
}