Merge pull request etcd-io#7402 from heyitsanthony/fix-watchconnerr

grpcproxy: return closing error when stream is canceled from conn close
adamgomulka · Mar 2, 2017 · 2eb8243 · 2eb8243
2 parents 134d1cb + 28e9ba3
commit 2eb8243
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/proxy/grpcproxy/watch.go b/proxy/grpcproxy/watch.go
@@ -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)
 	}