Go语言中高效处理HTTP请求体（详解net/http包之请求体的读取与复用）

基础：如何读取请求体？

最简单的读取方式是使用 io.ReadAll（Go 1.16+）或 ioutil.ReadAll（旧版本）：

package mainimport (    "fmt"    "io"    "net/http")func handler(w http.ResponseWriter, r *http.Request) {    body, err := io.ReadAll(r.Body)    if err != nil {        http.Error(w, "无法读取请求体", http.StatusBadRequest)        return    }    defer r.Body.Close()    fmt.Fprintf(w, "接收到的请求体: %s", string(body))}func main() {    http.HandleFunc("/", handler)    http.ListenAndServe(":8080", nil)}

这段代码能正常工作，但如果你在函数中多次调用 io.ReadAll(r.Body)，第二次及之后都会返回空字节，因为流已经被消费完了。

解决方案：如何复用请求体？

要实现 HTTP请求复用，核心思路是：先将原始请求体读入内存（如字节切片），然后用这个副本创建一个新的可重复读取的 io.ReadCloser。

以下是推荐做法：

package mainimport (    "bytes"    "fmt"    "io"    "net/http")func handler(w http.ResponseWriter, r *http.Request) {    // 第一步：读取原始请求体    originalBody, err := io.ReadAll(r.Body)    if err != nil {        http.Error(w, "读取请求体失败", http.StatusBadRequest)        return    }    r.Body.Close() // 关闭原始 Body    // 第二步：将字节切片包装为新的 ReadCloser    r.Body = io.NopCloser(bytes.NewBuffer(originalBody))    // 现在你可以多次读取 r.Body 了！    body1, _ := io.ReadAll(r.Body)    r.Body = io.NopCloser(bytes.NewBuffer(originalBody)) // 重置    body2, _ := io.ReadAll(r.Body)    fmt.Fprintf(w, "第一次读取: %s\n第二次读取: %s", string(body1), string(body2))}func main() {    http.HandleFunc("/reuse", handler)    http.ListenAndServe(":8080", nil)}

关键点在于：bytes.NewBuffer(originalBody) 创建了一个支持多次读取的缓冲区，再通过 io.NopCloser 将其转换为 io.ReadCloser 类型，从而替换掉原始的 r.Body。

进阶技巧：封装成中间件

如果你希望在整个应用中都能复用请求体（例如用于日志、验证、业务逻辑等），可以写一个中间件：

func reusableBodyMiddleware(next http.Handler) http.Handler {    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {        if r.Body != nil {            bodyBytes, err := io.ReadAll(r.Body)            if err != nil {                http.Error(w, "读取请求体失败", http.StatusBadRequest)                return            }            r.Body.Close()            // 保存原始字节到上下文（可选）            ctx := context.WithValue(r.Context(), "bodyBytes", bodyBytes)            // 替换 Body 为可重读版本            r.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))            r = r.WithContext(ctx)        }        next.ServeHTTP(w, r)    })}// 使用中间件func main() {    mux := http.NewServeMux()    mux.HandleFunc("/api", apiHandler)    http.ListenAndServe(":8080", reusableBodyMiddleware(mux))}

注意事项

• 不要对大文件请求体使用此方法，因为会将全部内容加载到内存，可能导致 OOM。
• 确保在读取后关闭原始 r.Body，避免资源泄漏。
• 在生产环境中，建议限制请求体大小（通过 http.MaxBytesReader）。

总结

通过本文，你已经掌握了在 Go语言 中使用 net/http 包读取请求体的基本方法，并学会了如何实现HTTP请求复用。这些技巧对于构建健壮、可维护的 Web 服务至关重要。

记住关键词：Go语言、net/http、请求体读取、HTTP请求复用——它们是你深入学习 Go Web 开发的基石。

现在，快去试试吧！遇到问题欢迎留言讨论。