Replace contextx.ShrinkDeadline with context.WithTimeout (#598)

2021-04-05 21:20:35 +08:00
parent 425430f67c
commit d88da4cc88
3 changed files with 3 additions and 7 deletions
--- a/zrpc/internal/clientinterceptors/timeoutinterceptor.go
+++ b/zrpc/internal/clientinterceptors/timeoutinterceptor.go
@@ -4,7 +4,6 @@ import (
 	"context"
 	"time"

-	"github.com/tal-tech/go-zero/core/contextx"
 	"google.golang.org/grpc"
 )

@@ -16,7 +15,7 @@ func TimeoutInterceptor(timeout time.Duration) grpc.UnaryClientInterceptor {
 			return invoker(ctx, method, req, reply, cc, opts...)
 		}

-		ctx, cancel := contextx.ShrinkDeadline(ctx, timeout)
+		ctx, cancel := context.WithTimeout(ctx, timeout)
 		defer cancel()

 		// create channel with buffer size 1 to avoid goroutine leak