feat: support the specified timeout of rpc methods (#2742)
Co-authored-by: hanzijian <hanzijian@52tt.com> Co-authored-by: Kevin Wan <wanjunfeng@gmail.com>
This commit is contained in:
@@ -40,6 +40,12 @@ func TestServer_setupInterceptors(t *testing.T) {
|
||||
Prometheus: true,
|
||||
Breaker: true,
|
||||
},
|
||||
SpecifiedTimeouts: []ServerSpecifiedTimeoutConf{
|
||||
{
|
||||
FullMethod: "/foo",
|
||||
Timeout: 5 * time.Second,
|
||||
},
|
||||
},
|
||||
}
|
||||
err = setupInterceptors(server, conf, new(stat.Metrics))
|
||||
assert.Nil(t, err)
|
||||
@@ -75,6 +81,12 @@ func TestServer(t *testing.T) {
|
||||
Prometheus: true,
|
||||
Breaker: true,
|
||||
},
|
||||
SpecifiedTimeouts: []ServerSpecifiedTimeoutConf{
|
||||
{
|
||||
FullMethod: "/foo",
|
||||
Timeout: time.Second,
|
||||
},
|
||||
},
|
||||
}, func(server *grpc.Server) {
|
||||
})
|
||||
svr.AddOptions(grpc.ConnectionTimeout(time.Hour))
|
||||
@@ -105,6 +117,7 @@ func TestServerError(t *testing.T) {
|
||||
Prometheus: true,
|
||||
Breaker: true,
|
||||
},
|
||||
SpecifiedTimeouts: []ServerSpecifiedTimeoutConf{},
|
||||
}, func(server *grpc.Server) {
|
||||
})
|
||||
assert.NotNil(t, err)
|
||||
@@ -131,6 +144,7 @@ func TestServer_HasEtcd(t *testing.T) {
|
||||
Prometheus: true,
|
||||
Breaker: true,
|
||||
},
|
||||
SpecifiedTimeouts: []ServerSpecifiedTimeoutConf{},
|
||||
}, func(server *grpc.Server) {
|
||||
})
|
||||
svr.AddOptions(grpc.ConnectionTimeout(time.Hour))
|
||||
|
||||
Reference in New Issue
Block a user