Go io.Reader 如何限制单次读取的最大字节数

上传请求、压缩包或文本流时，最怕上游数据没有边界：调用方以为“读一次”就只会拿到一小段，结果却把整个 Reader 读进内存。Go 的 io.LimitReader 可以给 Reader 套一个总量上限，但它解决的不是每次 Read 的缓冲区大小问题。

先把“读取上限”和“每次 Read 大小”分开

io.Reader 的约定是把数据写入调用方提供的 p []byte，一次最多写入 len(p) 个字节。io.LimitReader 则是在 Reader 外面增加一个“还剩多少字节”的计数器。两者叠加时，真正的一次读取大小取决于缓冲区和剩余上限中较小的那个。

用 io.LimitReader 给流加上总量边界

最小用法是先包装，再把包装后的 Reader 交给下游。下面的例子把一段文本的可读范围限制为 8 个字节；原始字符串更长，但 io.ReadAll 只能看到前 8 个字节。

package main

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

func main() {
    source := strings.NewReader("0123456789abcdef")
    limited := io.LimitReader(source, 8) // 给上游流设置总字节上限

    data, err := io.ReadAll(limited) // 读取包装后的 Reader，不会超过 8 字节
    if err != nil {
        fmt.Println("读取失败：", err) // 非 EOF 错误才需要按失败处理
        return
    }
    fmt.Printf("%q\\n", data) // 结果只包含 01234567
}

这里的 8 是整个包装 Reader 的剩余额度，不是每次调用的 buffer 长度。官方实现会在额度不足时缩短传给上游的切片，并在每次成功读取后扣减剩余值；额度耗尽时返回 0, io.EOF。

读取循环里先处理 n，再处理 err

如果不用 io.ReadAll，就要遵循 Reader 的通用约定：一次 Read 可能同时返回已经读到的数据和错误。因此不能写成“只要 err 不为空就丢弃 n”。

buf := make([]byte, 4) // 4 只控制单次 Read 的接收空间
for {
    n, err := limited.Read(buf)
    if n > 0 {
        consume(buf[:n]) // 先消费本次已经拿到的有效字节
    }
    if err != nil {
        if err == io.EOF {
            break // 到达上限或上游自然结束，都属于正常结束
        }
        return err // 其他错误说明读取过程异常
    }
}

示例中的 consume 代表业务处理函数。注意，io.LimitReader 不会替你关闭底层资源；如果上游是 io.ReadCloser，关闭责任仍在创建或接收它的那一层。

需要判断是否超限时，给上限多读一个字节

“读到 limit 个字节”只能说明结果被截断到上限，不能证明源数据刚好只有 limit 个字节。要区分“刚好等于上限”和“超过上限”，可把包装额度设为 limit+1。最多多读一个字节，既能得到明确判定，也不会把整个大输入装进内存。

func readAtMost(r io.Reader, limit int64) ([]byte, bool, error) {
    if limit  limit {
        return data[:len(data)-1], true, nil // 丢掉探测字节，返回上限内数据并标记超限
    }
    return data, false, nil // 等于 limit 也可能是刚好读完，不能误报
}

生产代码还要考虑 limit == math.MaxInt64 时的加一溢出；如果上限来自外部配置，应先做范围检查。若只是边界读取而不需要把内容汇总成切片，可以继续使用固定缓冲区循环，避免 ReadAll 带来的聚合内存。

常见问题

io.LimitReader 会限制每次 Read 的字节数吗？

不会。每次最多读多少仍受 len(p) 影响；它限制的是整个包装 Reader 的剩余额度。

到达上限时返回 EOF 是错误吗？

对这个包装 Reader 来说是正常结束信号。只有上游返回的其他非空错误，才应按读取失败处理。

为什么 limit+1 能判断超限？

若只读 limit 个字节，刚好等于上限和超过上限会得到同样长度。多读一个字节后，结果长度大于 limit 就能区分两者。

包装 ReadCloser 后需要调用 Close 吗？

需要关闭真正持有资源的底层 ReadCloser。LimitReader 只提供 Reader 包装，不负责替底层对象释放文件、连接或请求体。