grpcproxy: return nil on receiving snapshot EOF

Gets "code = OutOfRange desc = EOF" errors otherwise.
etcd-io · Jul 18, 2017 · dbd16d5 · dbd16d5
1 parent e6d2b31
commit dbd16d5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/proxy/grpcproxy/maintenance.go b/proxy/grpcproxy/maintenance.go
@@ -15,6 +15,8 @@
 package grpcproxy
 
 import (
+	"io"
+
 	"golang.org/x/net/context"
 
 	"github.com/coreos/etcd/clientv3"
@@ -49,6 +51,9 @@ func (mp *maintenanceProxy) Snapshot(sr *pb.SnapshotRequest, stream pb.Maintenan
 	for {
 		rr, err := sc.Recv()
 		if err != nil {
+			if err == io.EOF {
+				return nil
+			}
 			return err
 		}
 		err = stream.Send(rr)