improve code quality

2019-12-03 15:25:58 +08:00
parent 4cba1b3ec9
commit e13e329abc
1 changed files with 3 additions and 1 deletions
--- a/grpc.go
+++ b/grpc.go
@@ -222,7 +222,9 @@ func (g *grpcServer) handler(srv interface{}, stream grpc.ServerStream) error {
 	// set the timeout if we have it
 	if len(to) > 0 {
 		if n, err := strconv.ParseUint(to, 10, 64); err == nil {
-			ctx, _ = context.WithTimeout(ctx, time.Duration(n))
+			var cancel context.CancelFunc
+			ctx, cancel = context.WithTimeout(ctx, time.Duration(n))
+			defer cancel()
 		}
 	}