Go net/http MaxBytesReader 限制上传体积后如何判断读取失败

Go 服务端接收上传内容时，不能只依赖客户端声明的 Content-Length。更稳妥的做法是在 handler 内用 http.MaxBytesReader 包装 r.Body，让读取动作本身带有上限。关键判断是：读取成功不等于一定没有超限；当读取真正越过上限时，应该用 errors.As 判断 *http.MaxBytesError，不要只比较错误字符串。

官方地址：https://pkg.go.dev/net/http#MaxBytesReader

这篇只处理请求体读取、超限分支和资源关闭。文件类型、鉴权、业务字段校验仍然属于后续环节。

先看 MaxBytesReader 到底限制了什么

MaxBytesReader 接收响应写入器、原始请求体和最大字节数，返回一个新的 io.ReadCloser。它和 io.LimitReader 都能限制读取量，但前者面向 HTTP 请求体，并且在读穿上限时返回 *http.MaxBytesError；调用返回对象的 Close 时，也会关闭底层读取器。

因此，限制器解决的是“最多读多少”的输入边界，不会替代 JSON 解析、字段校验或鉴权。把它放在 handler 的入口附近，后续所有读取都使用包装后的 body，边界才不会被绕开。

用 errors.As 区分三种读取结果

下面的函数把请求体读取结果归类为成功、超限和其它错误。示例中的注释说明了判断顺序和关闭责任：

package upload

import (
    "errors"
    "io"
    "net/http"
)

func readLimitedBody(w http.ResponseWriter, r *http.Request, limit int64) ([]byte, int, error) {
    // 包装后的 body 才是后续读取的唯一入口，避免直接读取原始请求体。
    limited := http.MaxBytesReader(w, r.Body, limit)
    defer limited.Close() // 结束读取后释放底层请求体资源。

    data, err := io.ReadAll(limited)
    if err == nil {
        // 没有错误表示内容在上限内读完；业务格式仍需继续校验。
        return data, http.StatusOK, nil
    }

    var maxErr *http.MaxBytesError
    if errors.As(err, &maxErr) {
        // MaxBytesError 提供触发本次限制的字节上限，适合记录结构化日志。
        return nil, http.StatusRequestEntityTooLarge, maxErr
    }

    // 网络中断、底层读取失败等情况不能伪装成“上传超限”。
    return nil, http.StatusBadRequest, err
}

这里用 errors.As 而不是 err.Error() 字符串比较，是因为错误可能被包装；类型判断能保留 MaxBytesError.Limit 等结构化信息。HTTP 状态码只是示例策略，项目也可以统一映射为自己的错误响应格式。

在 handler 中把读取分支变成清晰响应

读取函数返回后，handler 再决定响应内容。超限是客户端输入不符合接口约束，普通读取失败则可能是连接中断或服务端读取链路问题，两者应该分开记录：

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    // 这里的 8 MiB 是接口约定，生产环境应按文件类型和业务场景配置。
    const maxUploadSize = 8 

如果接口接收的是 JSON，可以在读取成功后再调用解码器；如果接收 multipart 文件，则要把大小边界、临时文件策略和文件类型检查一起设计。不要因为已经限制了字节数，就把任意内容当成可信文件。

别把读取上限和业务校验混为一谈

上限值也不应只看单个文件大小。还要结合反向代理、服务器超时、内存预算和并发量设置，避免应用层允许的值高于入口层实际承受能力。

最后复查 Close 和错误边界

上线前可以按这四项复查：第一，是否所有读取都经过 MaxBytesReader；第二，是否使用 errors.As 识别 *http.MaxBytesError；第三，是否为包装后的 body 配置了 Close；第四，成功读取后是否仍然执行格式和业务校验。

这样处理后，“上传太大”“客户端中途断开”和“内容格式不对”不会被混成一个模糊的读取失败，日志、响应和后续排查都会更清楚。

相关问题

MaxBytesReader 和 io.LimitReader 有什么区别？

两者都能限制读取量，但 MaxBytesReader 面向 HTTP 请求体，返回 ReadCloser，并在读穿限制时返回 *http.MaxBytesError；它还会在 Close 时关闭底层读取器。

为什么不能直接比较错误字符串？

错误可能经过包装，字符串也可能随实现变化。使用 errors.As 判断 *http.MaxBytesError 才能稳定区分超限，并读取它的 Limit 字段。

读取成功后还需要校验文件或 JSON 吗？

需要。MaxBytesReader 只回答“读取大小是否超过上限”，不回答内容是否是合法 JSON、允许的文件类型或当前用户有无权限。