Fixed decoder, if io.Reader returns n > 0 and err != nil in a same time

1 files changed, 4 insertions, 3 deletions

diff --git a/decoder.go b/decoder.go
index dfd892e..0509c52 100644
--- a/decoder.go
+++ b/decoder.go
@@ -304,13 +304,14 @@ func (md *messageDecoder) read(data []byte) error {
 
 	for len(data) > 0 {
 		n, err := md.in.Read(data)
-		if err != nil {
+		if n > 0 {
+			md.cnt += n
+			data = data[n:]
+		} else if err != nil {
 			md.off = md.cnt
 			return err
 		}
 
-		md.cnt += n
-		data = data[n:]
 	}
 
 	return nil