diff --git a/beam/copy.go b/beam/copy.go index dd979dc55f..d62c03b8b4 100644 --- a/beam/copy.go +++ b/beam/copy.go @@ -1,6 +1,7 @@ package beam import ( + "io" "sync" ) @@ -22,6 +23,9 @@ func Copy(dst Sender, src Receiver) (int, error) { ) for { msg, err := src.Receive(Ret) + if err == io.EOF { + return n, nil + } if err != nil { return n, err }