瀏覽代碼

Merge pull request #5203 from heyitsanthony/fix-lease-leak

clientv3: check stream context in lease keep alive send loop
Anthony Romano 9 年之前
父節點
當前提交
ecb0e2bd38
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      clientv3/lease.go

+ 2 - 0
clientv3/lease.go

@@ -355,6 +355,8 @@ func (l *lessor) sendKeepAliveLoop(stream pb.Lease_LeaseKeepAliveClient) {
 	for {
 		select {
 		case <-time.After(500 * time.Millisecond):
+		case <-stream.Context().Done():
+			return
 		case <-l.donec:
 			return
 		case <-l.stopCtx.Done():