gin自定义中间件解决requestBody不可重复读问题

打印 上一主题 下一主题

主题 914|帖子 914|积分 2742

先直接上代码
  1.         r := gin.Default()
  2.         // 注册中间件,使body可以重复读取
  3.         r.Use(func(context *gin.Context) {
  4.                 all, err := context.GetRawData() // 读取body的内容
  5.                 if err != nil {
  6.                         log.Fatal(err)
  7.                 }
  8.                 // 重写 GetBody 方法,以便后续的其他操作
  9.                 context.Request.GetBody = func() (io.ReadCloser, error) {
  10.                         context.Request.Body = io.NopCloser(bytes.NewBuffer(all))
  11.                         buffer := bytes.NewBuffer(all)
  12.                         closer := io.NopCloser(buffer)
  13.                         return closer, nil
  14.                 }
  15.                 body, _ := context.Request.GetBody()  // 每次调用GetBody方法,都会新生成一个io.ReadCloser,但是底层的byte数据,都是all变量缓存的。
  16.                 context.Request.Body = body
  17.                 context.Next()
  18.         })
复制代码
注意,上面的中间件,需要在第一个执行。
分析

在gin中,context.Request.Body 是一个io.ReadCloser的接口,如下图

查看io.ReadCloser接口定义
  1. type ReadCloser interface {
  2.         Reader
  3.         Closer
  4. }
  5. type Reader interface {
  6.         Read(p []byte) (n int, err error)
  7. }
  8. type Closer interface {
  9.         Close() error
  10. }
复制代码
我们发现io.ReaderCloser接口的本质就是Read(p []byte) (n int, err error) 和 Close() error 的组合。
所以我们只需要自己编写实现Read(p []byte) (n int, err error) 和 Close() error 这两个方法的结构体即可赋值给context.Request.Body,在我们自己实现的方法中实现可重复读取即可达到我们的目的。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

知者何南

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表