Gin框架中如何在中间件与处理器中重复绑定JSON请求体？

Gin的请求体是io.ReadCloser类型，读取一次后游标会移到末尾，后续无法再次读取。要解决这个问题，核心是缓存请求体内容并重置到context中，或者仅解析需要的字段避免全量绑定。以下是两种可行方案：

方案一：缓存请求体并重置 #

在中间件中读取请求体后，将内容缓存，然后重新构造一个可读的io.ReadCloser设置回context，这样处理器就能再次读取。

修改你的中间件代码如下：

import (
    "bytes"
    "encoding/json"
    "io"
    "github.com/gin-gonic/gin"
)

type RequestWithCompany struct {
    CompanyId string `json:"company_id"`
}

func LoadCompany() gin.HandlerFunc {
    return func(context *gin.Context) {
        // 读取原始请求体
        bodyBytes, err := io.ReadAll(context.Request.Body)
        if err != nil {
            context.Next()
            return
        }
        // 关闭原始body，避免资源泄漏
        _ = context.Request.Body.Close()

        // 将读取到的内容重新设置回Request.Body，让后续处理器可以再次读取
        context.Request.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))

        // 解析company_id字段
        var requestBody RequestWithCompany
        err = json.Unmarshal(bodyBytes, &requestBody)
        if err != nil || requestBody.CompanyId == "" {
            context.Next()
            return
        }

        // 查询公司信息
        company, err := companyModel.Company{}.GetOneCompanyById(requestBody.CompanyId)
        if err != nil {
            context.Next()
            return
        }
        context.Set("company", company)
        context.Next()
    }
}

方案二：仅解析指定字段（更高效） #

如果只需要company_id字段，没必要绑定整个结构体，可以直接解析单个字段，减少内存开销：

import (
    "bytes"
    "encoding/json"
    "io"
    "github.com/gin-gonic/gin"
)

func LoadCompany() gin.HandlerFunc {
    return func(context *gin.Context) {
        // 读取原始请求体并缓存
        bodyBytes, err := io.ReadAll(context.Request.Body)
        if err != nil {
            context.Next()
            return
        }
        _ = context.Request.Body.Close()
        context.Request.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))

        // 仅解析company_id字段
        var temp struct {
            CompanyId string `json:"company_id"`
        }
        err = json.Unmarshal(bodyBytes, &temp)
        if err != nil || temp.CompanyId == "" {
            context.Next()
            return
        }

        company, err := companyModel.Company{}.GetOneCompanyById(temp.CompanyId)
        if err != nil {
            context.Next()
            return
        }
        context.Set("company", company)
        context.Next()
    }
}

关键注意事项 #

• 必须关闭原始的context.Request.Body，避免资源泄漏
• 重置body时要用io.NopCloser包装bytes.Buffer，因为Request.Body要求实现io.ReadCloser接口
• 处理company_id为空或查询失败的情况，确保中间件不阻塞正常请求

内容的提问来源于stack exchange，提问作者Oussama