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

先直接上代码

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()
19.         })

复制代码

注意，上面的中间件，需要在第一个执行。
分析

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

查看io.ReadCloser接口定义

1. type ReadCloser interface {
2.         Reader
3.         Closer
4. }
6. type Reader interface {
7.         Read(p []byte) (n int, err error)
8. }
10. type Closer interface {
11.         Close() error
12. }

复制代码

我们发现io.ReaderCloser接口的本质就是Read(p []byte) (n int, err error) 和 Close() error 的组合。
所以我们只需要自己编写实现Read(p []byte) (n int, err error) 和 Close() error 这两个方法的结构体即可赋值给context.Request.Body，在我们自己实现的方法中实现可重复读取即可达到我们的目的。

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