Merge pull request #7402 from heyitsanthony/fix-watchconnerr

grpcproxy: return closing error when stream is canceled from conn close
release-3.2
Anthony Romano 2017-03-01 21:56:36 -08:00 committed by GitHub
commit 2eb8243d94
1 changed files with 6 additions and 1 deletions

View File

@ -73,7 +73,12 @@ func (wp *watchProxy) Watch(stream pb.Watch_WatchServer) (err error) {
select {
case <-wp.ctx.Done():
wp.mu.Unlock()
return
select {
case <-wp.leader.disconnectNotify():
return grpc.ErrClientConnClosing
default:
return wp.ctx.Err()
}
default:
wp.wg.Add(1)
}