From 24a78a026d0efd181f5b645dfb4622c826d9207a Mon Sep 17 00:00:00 2001 From: Roger Peppe Date: Tue, 7 Dec 2010 14:54:15 -0500 Subject: [PATCH] bufio: make Reader.Read implement io.Reader semantics R=rsc CC=golang-dev https://golang.org/cl/3395042 --- src/pkg/bufio/bufio.go | 73 +++++++++++++++++++++--------------------- 1 file changed, 37 insertions(+), 36 deletions(-) diff --git a/src/pkg/bufio/bufio.go b/src/pkg/bufio/bufio.go index 4e9f1cf3e4c..fc4127a9402 100644 --- a/src/pkg/bufio/bufio.go +++ b/src/pkg/bufio/bufio.go @@ -128,43 +128,44 @@ func (b *Reader) Peek(n int) ([]byte, os.Error) { // Read reads data into p. // It returns the number of bytes read into p. -// If nn < len(p), also returns an error explaining -// why the read is short. At EOF, the count will be -// zero and err will be os.EOF. -func (b *Reader) Read(p []byte) (nn int, err os.Error) { - nn = 0 - for len(p) > 0 { - n := len(p) - if b.w == b.r { - if b.err != nil { - return nn, b.err - } - if len(p) >= len(b.buf) { - // Large read, empty buffer. - // Read directly into p to avoid copy. - n, b.err = b.rd.Read(p) - if n > 0 { - b.lastByte = int(p[n-1]) - b.lastRuneSize = -1 - } - p = p[n:] - nn += n - continue - } - b.fill() - continue - } - if n > b.w-b.r { - n = b.w - b.r - } - copy(p[0:n], b.buf[b.r:]) - p = p[n:] - b.r += n - b.lastByte = int(b.buf[b.r-1]) - b.lastRuneSize = -1 - nn += n +// It calls Read at most once on the underlying Reader, +// hence n may be less than len(p). +// At EOF, the count will be zero and err will be os.EOF. +func (b *Reader) Read(p []byte) (n int, err os.Error) { + n = len(p) + if n == 0 { + return 0, b.err } - return nn, nil + if b.w == b.r { + if b.err != nil { + return 0, b.err + } + if len(p) >= len(b.buf) { + // Large read, empty buffer. + // Read directly into p to avoid copy. + n, b.err = b.rd.Read(p) + if n > 0 { + b.lastByte = int(p[n-1]) + b.lastRuneSize = -1 + } + p = p[n:] + return n, b.err + } + b.fill() + if b.w == b.r { + return 0, b.err + } + } + + if n > b.w-b.r { + n = b.w - b.r + } + copy(p[0:n], b.buf[b.r:]) + p = p[n:] + b.r += n + b.lastByte = int(b.buf[b.r-1]) + b.lastRuneSize = -1 + return n, nil } // ReadByte reads and returns a single byte.