登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go教程

Go bytes.Buffer 与 bytes.Reader 怎么选:写入、读取和复用游标的差异

来源:17golang原创

时间:2026-08-27 21:15:45 279浏览 收藏

做文件上传、签名计算或协议编码时,输入往往已经是一段 []byte,但下游只认 io.Reader;另一类场景则需要先拼接多段内容,再交给网络层一次性读走。前者更适合 bytes.Reader,后者更适合 bytes.Buffer。关键差异不在名字,而在“谁负责产生字节”和“游标能不能回退”。

已有字节切片、只读并可能定位或重读,用 bytes.Reader;需要持续写入、读取后自然消耗,用 bytes.Buffer。不要把 Buffer 当成可随意回退的 Reader。

要点速览
  • bytes.Buffer 的零值即可写入,读取会消耗当前内容。
  • bytes.Reader 只读底层切片,支持 SeekReadAtReset
  • 需要复用同一份输入时优先使用 Reader;需要拼装输出时使用 Buffer。
  • 传给异步任务前要明确底层切片是否还会被修改。

先把两种字节流放进同一个小实验

实验只使用标准库 bytesio。准备一段短文本,分别让 Buffer 负责写入,让 Reader 包装既有切片。为了看清状态变化,代码会打印 Buffer 的 Len,再打印 Reader 的游标位置。

package main

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

func main() {
    var buf bytes.Buffer
    buf.WriteString("go-buffer")
    p := make([]byte, 2)
    n, _ := buf.Read(p)
    fmt.Printf("buffer=%q n=%d len=%d\n", p[:n], n, buf.Len())

    src := []byte("go-reader")
    r := bytes.NewReader(src)
    q := make([]byte, 2)
    n, _ = r.Read(q)
    pos, _ := r.Seek(0, io.SeekCurrent)
    fmt.Printf("reader=%q n=%d pos=%d len=%d\n", q[:n], n, pos, r.Len())
}

运行结果应显示 Buffer 的长度从 9 变成 7,Reader 的当前位置从 0 变成 2,剩余长度变成 7。这里的数字来自字符串本身,不是容量:Buffer 的 Len 和 Reader 的 Len 都表示当前还能读多少字节。

Buffer 的主路径:WriteString 之后由 Read 消耗

bytes.Buffer 的零值可以直接使用。WriteStringWriteReadFrom 把数据追加进去,ReadNextReadBytes 则从当前读位置取走数据。读过的部分不会因为再次调用 Bytes 而自动回到开头。

Go bytes.Buffer 中 WriteString 写入后由 Read 消耗并更新 Len 的数据路径

这条路径适合“边收到边拼起来”的代码,例如把多个小片段写成请求体。若要再次读取完整内容,应在消耗之前复制 Bytes,或从原始输入重新构造;不要把 Buffer.Bytes() 返回的切片当成永久快照。

var buf bytes.Buffer
buf.WriteString("header:")
buf.Write([]byte("payload"))
part := make([]byte, 7)
if _, err := buf.Read(part); err != nil && err != io.EOF {
    panic(err)
}
fmt.Println(string(part), buf.Len()) // header: 7

Reader 的主路径:Seek 定位,ReadAt 不移动游标

bytes.Reader 包装已有切片,底层数据不会因为普通读取而被移除。Read 会前进游标,Seek 可以按绝对或相对位置定位,ReadAt 从指定偏移读取且不改变当前游标,Reset 则把 Reader 绑定到另一段切片。

Go bytes.Reader 通过 Seek 定位、ReadAt 按偏移读取并用 Reset 换回新切片
src := []byte("0123456789")
r := bytes.NewReader(src)
head := make([]byte, 3)
_, _ = r.Read(head) // 游标到 3

at := make([]byte, 2)
_, _ = r.ReadAt(at, 7) // 读取 78,游标仍为 3
_, _ = r.Seek(-1, io.SeekCurrent) // 回到 2
fmt.Println(string(head), string(at), r.Len())

r.Reset([]byte("new-data"))
fmt.Println(r.Len()) // 8

这个能力对重复解析头部、计算签名或实现分段读取很有用。要注意 ReadAt 的偏移是相对底层切片的绝对位置,不是相对当前游标的位置;如果偏移超过可读范围,返回的字节数可能小于目标切片长度,并伴随 io.EOF

按任务选择:拼装输出还是复用输入

任务更合适的类型原因
逐段生成请求体bytes.Buffer写入 API 直接,读完后自然消耗
已有切片交给 JSON 或哈希读取bytes.Reader只读且可重新定位
按偏移读取文件头bytes.ReaderReadAt 不改变主游标
暂存上游流并转交下游bytes.Buffer实现 io.Reader,可持续追加

如果下游只需要一次读取,两者都能满足 io.Reader。选择依据应放在数据生命周期:Buffer 的内容由它逐步拥有和消耗,Reader 的内容通常仍由调用方持有,调用方必须保证底层切片在读取期间不被并发修改。

三个容易踩到的边界

不要用 Buffer 的 Bytes 返回值跨越下一次写入

Bytes 返回的是当前缓冲区视图,下一次写入可能让底层存储变化。需要长期保存时复制一份;只是立刻计算长度或同步交给读取函数,则可以直接使用。

不要把 Reader.Len 当成底层切片总长度

Reader 的 Len 是“当前位置到结尾”的剩余长度。总长度应使用 Size,或者在构造 Reader 前保存 len(src)

复用前先确认所有读取已经结束

Reset 会改变 Reader 的数据和位置,Buffer 的 Reset 会清空内容。它们都适合串行复用,不适合在仍有下游读取时抢先调用。

相关问题:什么时候换成 bufio.Reader

bytes.Reader 能替代文件读取吗?

它适合内存中的既有字节,不会替代需要从磁盘或网络持续取数据的文件对象。数据来源是流时,应让真正的流实现 io.Reader

Buffer 能不能 Seek?

不能把 Buffer 当作带定位能力的 Reader 使用。需要回退或按偏移读取时,保留原始切片并构造 bytes.Reader,或者重新建立 Buffer。

Reader 读取后还能再读一遍吗?

可以通过 Seek(0, io.SeekStart) 回到开头;如果只是某一段旁路读取,优先使用不移动游标的 ReadAt

什么时候应该选 bufio.Reader?

当数据来自文件、连接等持续流,并且需要按行或按小块缓冲时选 bufio.Reader。它解决的是流式缓冲,不是已有切片的复用定位。

把选择写成一条可复查的规则

先问数据是否已经在内存里:没有,就让上游流直接接入或使用 bufio.Reader;有,就问是否还要追加内容。要追加,选择 bytes.Buffer;只读且可能回退、按偏移读取,选择 bytes.Reader。最后再检查生命周期:谁持有底层切片、谁负责复用,以及读取是否已经结束。这个判断比记住某个方法名更不容易出错。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>