Go 怎样为 ReadAll 设置输入大小上限

直接调用 io.ReadAll(r) 会一直读到 EOF 或底层错误，输入来自网络、上传或不可信 Reader 时，内存上限并不会自动出现。最小改法是在它前面包一层限制器；如果业务必须拒绝超大输入，就把读取窗口设为“允许大小加 1”，再根据结果判断。

一、先把上限放在 ReadAll 之前

io.ReadAll 本身没有大小参数。对于只允许读取前 maxBytes 字节的内部输入，可以先用 io.LimitReader 包装：

package main

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

func main() {
    const maxBytes = 8
    source := strings.NewReader("0123456789abcdef")

    // 限制器让 ReadAll 最多看到 maxBytes 个字节。
    limited := io.LimitReader(source, maxBytes)
    data, err := io.ReadAll(limited)
    if err != nil {
        // 底层读取错误仍然要向上返回，不能把它当成正常截断。
        panic(err)
    }
    fmt.Printf("读取 %d 字节：%s\n", len(data), data)
}

这段代码会得到 8 个字节。需要注意：LimitReader 到达限制后返回 EOF，所以它只保证“最多读多少”，并不会告诉你原始输入是否还有剩余内容。适合预览、截取固定前缀或协议本来就允许截断的场景。

二、需要拒绝超限时多读一个字节

解析 JSON、配置文件或签名载荷时，静默截断通常会掩盖数据错误。此时把限制值增加一个字节：长度等于 maxBytes+1 就说明输入超限；长度更小或相等才进入后续解析。

package main

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

var errTooLarge = errors.New("input exceeds limit")

func readAtMost(r io.Reader, maxBytes int64) ([]byte, error) {
    if maxBytes  maxBytes {
        return nil, errTooLarge
    }
    return data, nil
}

func main() {
    data, err := readAtMost(strings.NewReader("0123456789"), 8)
    if err != nil {
        fmt.Println("拒绝输入：", err)
        return
    }
    fmt.Println(string(data))
}

这里的 maxBytes+1 只增加一个字节的探测成本，仍然把内存控制在上限附近。若输入还要做 JSON 解码，应先完成这个边界判断，再把 data 交给解码器；不要先把无限输入全部读进内存。

三、HTTP 请求体优先使用 MaxBytesReader

在 HTTP Handler 中，r.Body 来自客户端，建议在第一次读取前设置 http.MaxBytesReader。它在读到上限后返回 *http.MaxBytesError，比普通 LimitReader 更适合向客户端返回 413。

func upload(w http.ResponseWriter, r *http.Request) {
    const maxBody = 1 

示例中的错误分支很重要：超限、客户端断开和底层 I/O 失败不应共用一个成功响应。对真正可能很大的文件，则应改用分块或流式处理，大小限制只是护栏，不是把大文件变成一次性字节切片的理由。

四、按输入来源选择方案

所以，“给 ReadAll 设置上限”不是给函数加一个不存在的参数，而是把边界放到 Reader 层，并先决定超限时要截断还是拒绝。只要这个决策明确，代码和错误处理就不会混在一起。

相关问题

LimitReader 到达上限会返回错误吗？

它会在限制范围结束后表现为 EOF；如果业务要识别超限，应使用 max+1 探测或改用 MaxBytesReader。

为什么不只检查 Content-Length？

请求可能没有可靠的长度信息，且实际读取仍可能超过预期；最终边界必须落在真正消费 Body 的 Reader 上。

ReadAll 之后还能节省内存吗？

不能。ReadAll 已经构造了完整字节切片；如果输入可能很大，应在读取前限制，或直接改为流式解析。