gin自定义中间件解决requestBody不可重复读问题
  i86oKZWg3QMr 2023年11月02日 55 0
Go

先直接上代码

	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的接口,如下图
image

查看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,在我们自己实现的方法中实现可重复读取即可达到我们的目的。

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
  uGYzDadp0Cs7   2024年04月18日   78   0   0 Go
  hyrB1Ag4eVs8   2024年04月15日   69   0   0 Go
  dHUS172Lkv6A   2024年05月08日   174   0   0 Go
  YFCZjJLTjJgW   2024年05月04日   51   0   0 Go
  YFCZjJLTjJgW   2024年05月17日   59   0   0 Go
  uGYzDadp0Cs7   2024年04月16日   118   0   0 Go
  YFCZjJLTjJgW   2024年05月17日   59   0   0 Go
i86oKZWg3QMr