Browse Source

grpcproxy: return nil on receiving snapshot EOF

Gets "code = OutOfRange desc = EOF" errors otherwise.
Anthony Romano 8 năm trước cách đây
mục cha
commit
19fe4b0cac
1 tập tin đã thay đổi với 5 bổ sung0 xóa
  1. 5 0
      proxy/grpcproxy/maintenance.go

+ 5 - 0
proxy/grpcproxy/maintenance.go

@@ -15,6 +15,8 @@
 package grpcproxy
 package grpcproxy
 
 
 import (
 import (
+	"io"
+
 	"golang.org/x/net/context"
 	"golang.org/x/net/context"
 
 
 	"github.com/coreos/etcd/clientv3"
 	"github.com/coreos/etcd/clientv3"
@@ -49,6 +51,9 @@ func (mp *maintenanceProxy) Snapshot(sr *pb.SnapshotRequest, stream pb.Maintenan
 	for {
 	for {
 		rr, err := sc.Recv()
 		rr, err := sc.Recv()
 		if err != nil {
 		if err != nil {
+			if err == io.EOF {
+				return nil
+			}
 			return err
 			return err
 		}
 		}
 		err = stream.Send(rr)
 		err = stream.Send(rr)