gin自定义中间件解决requestBody不可重复读问题(最新推荐)
短信预约 -IT技能 免费直播动态提醒
先直接上代码
r := gin.Default()
// 注册中间件，使body可以重复读取
r.Use(func(context *gin.Context) {
all, err := context.GetRawData() // 读取body的内容
if err != nil {
log.Fatal(err)
}
// 重写 GetBody 方法，以便后续的其他操作
context.Request.GetBody = func() (io.ReadCloser, error) {
context.Request.Body = io.NopCloser(bytes.NewBuffer(all))
buffer := bytes.NewBuffer(all)
closer := io.NopCloser(buffer)
return closer, nil
}
body, _ := context.Request.GetBody() // 每次调用GetBody方法，都会新生成一个io.ReadCloser，但是底层的byte数据，都是all变量缓存的。
context.Request.Body = body
context.Next()
})注意，上面的中间件，需要在第一个执行。
分析
在gin中，context.Request.Body 是一个io.ReadCloser的接口，如下图
查看io.ReadCloser接口定义
type ReadCloser interface {
Reader
Closer
}
type Reader interface {
Read(p []byte) (n int, err error)
}
type Closer interface {
Close() error
}我们发现io.ReaderCloser接口的本质就是Read(p []byte) (n int, err error) 和 Close() error 的组合。
所以我们只需要自己编写实现Read(p []byte) (n int, err error) 和 Close() error 这两个方法的结构体即可赋值给context.Request.Body，在我们自己实现的方法中实现可重复读取即可达到我们的目的。
到此这篇关于gin自定义中间件解决requestBody不可重复读问题的文章就介绍到这了,更多相关requestBody不可重复读内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
