Gin中间件打印request.Body

之前用Gin开发的时候,使用中间件打印日志,如果向获取body的内容,做法就是写一个中间件方法,返回HandlerFunc,在next()方法调用之前读取request.Body的内容,然后以字符串的方式输出到日志。如下面一个简单的测试:

func LoggerMiddleWare() gin.HandlerFunc {
	return func(c *gin.Context) {
		// PUT或者POST方法时打印body
		if c.Request.Method == http.MethodPut || c.Request.Method == http.MethodPost {
			data, err := ioutil.ReadAll(c.Request.Body)
			if err != nil {
				glog.Errorf("read request body failed,err = %s.", err)
				return
			}
			glog.Infof("request body = %s.", string(data))
		}
		c.Next()
	}
}

这样写之后,会导致c.Next()之后的handler都读取不到body的值。原因就是request.Body时一个ReadCloser接口,同时实现了Reader接口和Closer接口。当进行一次读操作之后会执行关闭接口功能,第二读的时候检测到已经关闭就会直接返回EOF,起始数据本身还在,只是不能读取了。解决方法就是通过NopCloser重新给Body赋值,只有一个Reader接口,无关闭的Closer接口相当与重置Closer接口。

// 重新赋值
c.Request.Body = ioutil.NopCloser(bytes.NewBuffer(data))

参考:https://stackoverflow.com/questions/47186741/how-to-get-the-json-from-the-body-of-a-request-on-go/47295689#47295689

Gin 框架中处理 "invalid request body" 的错误通常涉及以下几个步骤: 1. **中间件**:首先,你可以使用 Gin 提供的 `gin.Recovery()` 中间件来捕获整个请求处理过程中的所有错误,包括 JSON 解析失败。这个中间件会尝试恢复并返回一个默认的错误响应。 ```go router.Use(gin.Recovery()) ``` 2. **自定义验证**:如果你需要更具体的错误处理,可以在路由级别添加一个校验函数,检查请求体是否有效。例如,你可以使用 `binding.BindJSON` 或 `binding.Unmarshal` 来尝试解析 JSON,并在发生错误时抛出异常。 ```go func validateRequestBody(c *gin.Context) { var requestBody YourStruct if err := binding.UnmarshalJSON(c.Body, &requestBody); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request body"}) c.Abort() // 立即终止当前请求 } } ``` 3. **错误处理器**:还可以创建一个专门的错误处理器函数,处理 `body ValidationError` 类型的错误。这可以让你提供更友好的错误信息和状态码。 ```go func invalidRequestBodyHandler(c *gin.Context) { c.JSON(http.StatusBadRequest, gin.H{"error": "Request body is not valid", "details": err.Error()}) } ``` 然后,在路由器上注册这个处理器作为处理特定错误的回调: ```go router.Use(validationRequestBody) router.Any("/your-endpoint", func(c *gin.Context) { ... }) // 如果绑定失败,将调用 invalidRequestBodyHandler router.P rescuedWith(invalidRequestBodyHandler) ```
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

空间法则

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值