Do not require explicit Read for zero length

[netstring.git] / r.go

diff --git a/r.go b/r.go
index 9b8f51069e541d86986497e383dec57349f563b8..098ee49be485cb37a10e420c5fce7c03c6405a5d 100644 (file)
--- a/r.go
+++ b/r.go
@@ -58,7 +58,13 @@ func (r *Reader) Next() (uint64, error) {
        }
        r.left = size
        r.eof = false
-       return size, nil
+       if r.left == 0 {
+               err = r.checkTerminator()
+               if err == nil {
+                       r.eof = true
+               }
+       }
+       return size, err
 }
 
 func (r *Reader) checkTerminator() error {