Pārlūkot izejas kodu

Merge pull request #5176 from xiang90/lease_client

clientv3: keepaliveonce should have a per call ctx
Xiang Li 9 gadi atpakaļ
vecāks
revīzija
9ec176a9b0
1 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  1. 4 1
      clientv3/lease.go

+ 4 - 1
clientv3/lease.go

@@ -250,7 +250,10 @@ func (l *lessor) keepAliveCtxCloser(id LeaseID, ctx context.Context, donec <-cha
 }
 
 func (l *lessor) keepAliveOnce(ctx context.Context, id LeaseID) (*LeaseKeepAliveResponse, error) {
-	stream, err := l.getRemote().LeaseKeepAlive(ctx)
+	cctx, cancel := context.WithCancel(ctx)
+	defer cancel()
+
+	stream, err := l.getRemote().LeaseKeepAlive(cctx)
 	if err != nil {
 		return nil, err
 	}