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

Posted 2023-04-29 pdsoloz

先直接上代码

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

go的web框架-gin

1 gin的helloworld体验
2 使用New和Default初始化路由器的区别

new 没有错误捕捉

3 gin的路由分组

4 获取url中的变量

5 获取get和post表单信息

6 gin返回protobuf

7 登录的表单验证

8 注册表单的验证

9 表单验证错误翻译成中文
10 表单中文翻译的json格式化细节
11 自定义gin中间件
12 通过abort终止中间件后续逻辑的执行
13 gin的中间件原理源码分析
14 gin返回html
15 加载多个html文件
16 static静态文件的处理
17 gin的优雅退出