最新要闻

广告

手机

iphone11大小尺寸是多少？苹果iPhone11和iPhone13的区别是什么？

iphone11大小尺寸是多少？苹果iPhone11和iPhone13的区别是什么？

 警方通报辅警执法直播中被撞飞：犯罪嫌疑人已投案

警方通报辅警执法直播中被撞飞：犯罪嫌疑人已投案

家电

  1. 首页
  2. 移动通信
  3. 内容

最新资讯：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 = bodycontext.Next()})

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

分析

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

查看io.ReadCloser接口定义

type ReadCloser interface {ReaderCloser}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，在我们自己实现的方法中实现可重复读取即可达到我们的目的。

关键词：