这篇文章主要介绍了gin自定义中间件解决requestBody不可重复读问题,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
先直接上代码
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不可重复读内容请搜索编程学习网以前的文章希望大家以后多多支持编程学习网!
织梦狗教程
本文标题为:gin自定义中间件解决requestBody不可重复读问题(最新推荐)


基础教程推荐
猜你喜欢
- R语言 ggplot2改变柱状图的顺序操作 2022-11-17
- ruby on rails validates 2023-09-22
- Ruby3多线程并行Ractor使用方法详解 2023-07-23
- 浅析ELF转二进制允许把 Binary 文件加载到任意位置 2023-07-06
- R语言-修改(替换)因子变量的元素操作 2022-11-26
- R语言绘制折线图实例分析 2022-11-21
- ruby-on-rails-为使用Rails 4,nginx和乘客的用户设置自定义域 2023-09-21
- Swift中重写和重载的使用与对比总结 2023-07-05
- win10下使用virtualbox + vagrant配置ruby开发机环境 2023-07-23
- Swift初始化器与可选链的使用方法介绍 2023-07-08