소스 검색

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():