Merge pull request #7402 from heyitsanthony/fix-watchconnerr

grpcproxy: return closing error when stream is canceled from conn close
This commit is contained in:
Anthony Romano
2017-03-01 21:56:36 -08:00
committed by GitHub

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)
}