Go Reader Peek 后读取长度不足怎么办

先给结论：bufio.Reader.Peek(n) 返回的切片长度不足时，不要直接按 n 取下标。先判断错误，再区分两类边界：如果请求长度超过 Reader 的内部缓冲容量，通常得到 bufio.ErrBufferFull；如果底层输入已经结束或返回其他错误，可能只得到当前可用字节。需要“预览后再完整消费”时，复制 Peek 结果保存判断依据，随后用 io.ReadFull 读取固定长度。

本文讨论的官方接口是 bufio.Reader。标准库地址：https://pkg.go.dev/bufio#Reader.Peek。

先明确 Peek 的行为范围

Peek(n) 只观察 Reader 中接下来的字节，不推进读取位置。它返回的切片由 Reader 管理，后续读操作、丢弃操作或再次填充缓冲区后，切片内容就不应继续当作稳定副本使用。

这里最容易混淆的是“请求长度”和“当前可用长度”。当 n 大于内部缓冲区容量时，Reader 无法一次提供所需窗口，会返回 ErrBufferFull；当 n 没超过容量但输入在填满前结束，返回切片可能短于 n，此时应结合错误决定是正常结束、输入不完整还是底层故障。

把返回长度与错误放在同一张变更表里

表格中的“通常”很重要：短切片本身不是错误类型，错误值才说明 Reader 为什么没有给出完整窗口。旧代码若只写 if len(data) ，会把 EOF、缓冲区边界和底层故障混成一个结果，后续排查会缺少依据。

旧写法为什么会在短读时出问题

package main

import (
    "bufio"
    "fmt"
    "io"
    "strings"
)

func inspect(r *bufio.Reader, n int) error {
    data, err := r.Peek(n)
    // 先处理错误，再使用返回切片，避免把 ErrBufferFull 当成普通短读。
    if err != nil {
        return fmt.Errorf("预览 %d 字节失败: %w", n, err)
    }
    if len(data) != n {
        // 没有完整窗口时不继续按 n 取下标，保留短读语义。
        return io.ErrUnexpectedEOF
    }

    // Peek 的切片由 Reader 管理，复制后再交给业务逻辑保存。
    preview := append([]byte(nil), data...)
    fmt.Printf("预览内容=%q\\n", preview)

    // Peek 不推进位置；这里显式消费同一段数据，避免重复读取。
    consumed := make([]byte, n)
    if _, err := io.ReadFull(r, consumed); err != nil {
        return fmt.Errorf("消费预览数据失败: %w", err)
    }
    return nil
}

func main() {
    // 使用短输入演示：完整窗口不足时应返回错误，而不是越界访问。
    r := bufio.NewReader(strings.NewReader("go"))
    if err := inspect(r, 4); err != nil {
        fmt.Println("检查结果:", err)
    }
}

这个示例没有把代码运行输出当成文章证据；关键是处理顺序：先读取 data, err，再判断 err 和 len(data)，最后才复制或消费。若把 data[n-1] 写在错误判断之前，短切片就可能触发越界。

用 Peek、复制和 ReadFull 组合处理

需要预览固定前缀时，推荐把职责拆开：Peek 只负责观察，复制负责保存稳定副本，ReadFull 负责在确认输入足够后消费固定长度。这样不会误以为 Peek 已经移动了 Reader 的位置，也不会把底层短读误判成完整消息。

如果协议允许的前缀长度可能超过 Reader 的缓冲容量，不能靠增大 Peek 参数硬取完整消息。应把读取拆成可管理的小块，或者直接使用流式解析；ErrBufferFull 表示观察窗口设计超出了当前缓冲边界，不表示输入内容一定损坏。

回归检查与迁移清单

• 把所有 Peek 调用的错误判断移到切片索引之前。
• 为 len(data) 明确区分 EOF、ErrBufferFull 和其他错误。
• 需要跨越后续读操作保存内容时，立即复制 Peek 返回切片。
• Peek 后不要再假设读取位置已经移动；消费固定长度时使用 ReadFull 或明确的 Read。
• 为“输入刚好 n 字节”“输入少于 n 字节”“n 超过缓冲容量”和“底层 Reader 报错”分别准备回归用例。

相关问题

Peek 返回短切片一定是 ErrBufferFull 吗？

不一定。ErrBufferFull 重点表示请求窗口超过内部缓冲能力；输入在 EOF 前结束时，也可能返回较短切片并带 EOF。必须同时看 error。

Peek 后可以直接保存返回的 []byte 吗？

不建议。后续读取或缓冲区变化可能使这段切片失效；需要保存就用 append([]byte(nil), data...) 复制。

为什么 Peek 后还要 ReadFull？

Peek 只观察不消费，ReadFull 才负责把确认需要的固定长度从 Reader 中完整取走；二者职责不同。