Go bytes.Buffer 怎样判断读取位置已经到末尾

我在用 bytes.Buffer 拼接数据后做分段读取时，最容易混淆的是“缓冲区还有容量”和“读取位置还有数据”这两件事。判断读取位置是否已经到末尾，核心不是看 Cap()，而是看当前未读部分：b.Len() == 0。

如果只是想判断 bytes.Buffer 是否还有可读字节，用 b.Len() == 0；如果正在实现 io.Reader 循环，再结合 n 和 errors.Is(err, io.EOF) 处理结束信号。

官方资料：

• https://pkg.go.dev/bytes
• https://go.dev/src/bytes/buffer.go

先分清 Len、Bytes 和 Cap

bytes.Buffer 对外最有用的三个观察点，含义并不相同：

官方实现把缓冲区的未读区域视为 buf[off:]，所以 Len() 统计的是这段区域的长度。即使已经读完，底层仍可能保留一部分容量供后续写入；这不会改变“当前没有可读数据”的结论。

最直接的判断是 Len 等于零

在读取前或每次读取后，只要你的问题是“还剩多少数据”，就直接看 Len()：

package main

import (
	"bytes"
	"fmt"
)

func main() {
	var buf bytes.Buffer
	buf.WriteString("Go") // 写入两字节，之后演示读取位置的变化

	fmt.Println(buf.Len() == 0) // false：还有两个未读字节
	buf.Next(2)                 // 消费全部未读数据，读取位置移动到末尾
	fmt.Println(buf.Len() == 0) // true：当前没有可读字节
}

如果你需要拿到剩余内容再判断，也可以写成 len(buf.Bytes()) == 0。不过，单纯判断是否读完时 Len() 更直接，而且表达的是“未读字节数”这个业务含义。

Read 循环不要只看 EOF

Buffer.Read 遵循 io.Reader 约定：有数据时返回读到的字节数和 nil；没有数据时，目标切片长度大于零才返回 io.EOF。因此循环必须先处理 n，再处理错误：

package main

import (
	"bytes"
	"errors"
	"fmt"
	"io"
)

func main() {
	var buf bytes.Buffer
	buf.WriteString("hello") // 准备一段待读取数据

	chunk := make([]byte, 2) // 非空目标切片，耗尽时可以收到 io.EOF
	for {
		n, err := buf.Read(chunk)
		if n > 0 {
			fmt.Printf("读取 %q，剩余 %d 字节\n", chunk[:n], buf.Len()) // 先消费本次有效数据
		}
		if errors.Is(err, io.EOF) {
			break // 只有确认没有更多数据后才结束循环
		}
		if err != nil {
			panic(err) // 真实项目中应按调用链返回或记录错误
		}
	}
}

这里不能把 err != nil 放在处理 n 之前，也不能假定每次只返回“有数据”或“EOF”二选一。对 bytes.Buffer 来说，正常读取会把有效字节通过 n 返回；循环退出的判断应让 EOF 负责收尾。

按字节读取时直接判断 io.EOF

如果业务是逐字节扫描，ReadByte() 会在没有字节可取时返回 io.EOF。这种场景不必额外调用 Len()，但仍应区分 EOF 和其他错误：

package main

import (
	"bytes"
	"errors"
	"fmt"
	"io"
)

func main() {
	var buf bytes.Buffer
	buf.WriteString("OK") // 逐字节读取的输入

	for {
		ch, err := buf.ReadByte()
		if errors.Is(err, io.EOF) {
			break // 没有下一个字节，读取位置已经到末尾
		}
		if err != nil {
			panic(err) // 保留未来替换读取器时的错误处理空间
		}
		fmt.Printf("%c", ch) // 只处理本次成功读取的字节
	}
}

若代码同时需要展示进度，可以在成功读取后查看 buf.Len()。但不要把 Cap() 当成进度指标，容量只代表底层存储空间，不随读取位置以同样方式表达剩余数据。

Next、ReadString 和 UnreadByte 的边界

Next(n) 会返回最多 n 个未读字节并推进读取位置，但它没有错误返回值。如果一次取完后需要判断是否结束，就在调用后看 Len() == 0：

package main

import (
	"bytes"
	"fmt"
)

func main() {
	buf := bytes.NewBufferString("abc")
	part := buf.Next(10) // n 大于剩余长度时只返回全部剩余数据
	fmt.Printf("%q, done=%t\n", part, buf.Len() == 0) // true 表示已经没有未读字节
}

ReadString(delim) 和 ReadBytes(delim) 则可能返回“已经读到的一段数据”以及 io.EOF。没有找到分隔符时，不能因为错误非空就丢弃返回的数据；应先处理内容，再根据错误决定是否结束。

还有一个容易让调试结果反复变化的情况：调用 UnreadByte() 会把最近一次成功读取的字节放回未读区域。此时 Len() 可能从 0 重新变为 1，所以“到过末尾”不等于“之后永远保持末尾”。写入、Reset() 和 UnreadByte() 都可能改变下一次判断。

三个常见误区对应的修正

1. 把 Cap 当成剩余数据。 修正为 Len()；容量只回答还能否复用底层空间。
2. 只用 len(p) == 0 判断读完。 p 是本次接收数据的目标，不代表 Buffer 当前是否还有数据；应看 Len() 或读取结果。
3. 看到 EOF 就跳过 n。 对带分隔符的读取方法，返回的数据可能同时有意义；先处理数据，再根据 errors.Is(err, io.EOF) 结束。

另外，bytes.Buffer 的读写不能在没有同步保护的情况下被多个 goroutine 并发访问。若“读取位置”在并发场景下看起来不稳定，先解决所有权和同步问题，再讨论 Len 判断。

最后用这张表选择判断方式

一句话记忆：bytes.Buffer 的“末尾”指未读窗口为空，最稳定的状态判断是 Len() == 0；在 Reader 风格循环里，再把 n 和 io.EOF 按约定处理即可。

相关问题

• bytes.Buffer 的 Bytes() 返回值可以长期保存吗？ 不建议跨越下一次读写或重置长期保存，因为它与内部存储可能共享。
• Read 返回 0、nil 是否一定读完了？ 不一定，先看目标切片是否为空以及具体 Reader 的约定；对 Buffer 的常规非空读取目标，耗尽时通常得到 EOF。
• 为什么读取完后 Buffer 还能继续 Write？ 读到末尾只代表未读窗口为空，底层存储仍可被复用，后续写入会重新产生未读数据。