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

prevent negative timeout settings (#482)

Browse Source 
* prevent negative timeout settings

* fix misleading comment
This commit is contained in:
Kevin Wan
2021-02-19 10:44:39 +08:00
committed by GitHub
parent d239952d2d
commit 086113c843
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
zrpc/config.go
View File

@@ -14,8 +14,7 @@ type (
Auth bool `json:",optional"`
Redis redis.RedisKeyConf `json:",optional"`
StrictControl bool `json:",optional"`
// pending forever is not allowed
// never set it to 0, if zero, the underlying will set to 2s automatically
// setting 0 means no timeout
Timeout int64 `json:",default=2000"`
CpuThreshold int64 `json:",default=900,range=[0:1000]"`
}

3
zrpc/internal/clientinterceptors/timeoutinterceptor.go
View File

@@ -11,9 +11,10 @@ import (
func TimeoutInterceptor(timeout time.Duration) grpc.UnaryClientInterceptor {
return func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn,
invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
if timeout == 0 {
if timeout <= 0 {
return invoker(ctx, method, req, reply, cc, opts...)
}
ctx, cancel := contextx.ShrinkDeadline(ctx, timeout)
defer cancel()
return invoker(ctx, method, req, reply, cc, opts...)