go-zero读取请求体出现EOF错误的解决方法
作者：PandaMiner
前言
最近自己在搭 go-zero 的脚手架，遇到一个问题，原先的一个 post 请求是执行成功，当我添加了一个过滤器之后执行该请求就会报错请求体读取 EOF 错误，我感到有点奇怪，为什么过滤器中的逻辑会影响到后续的请求呢？在此记录下这次的坑。
问题分析
先上结论，这是由于 go 中 http.Request.Body 请求体是一个流，它只能被读取一次，假如你在过滤器已经进行过流的读取，那么在后续的请求中就读取不到了。
示例的请求逻辑是这样，当请求到达后台时，先经过过滤器检查请求头中是否携带 token，否则则请求失败，成功则继续 post 请求逻辑，本次的 post 请求和过滤器只是做示例使用，逻辑很简单，过滤器中的逻辑就是去获取请求 header 中 token 字段，如果这个字段为空，则请求失败，代码示例如下：
func tokenFilter(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
type Request struct {
Token string `header:"token"`
}
var req Request
err := httpx.Parse(r, &req)
if err != nil || req.Token == "" {
resp, _ := utils.ToJsonString(models.ErrorResponseWithCode(http.StatusUnauthorized, "token required"))
routers.HttpError(w, http.StatusUnauthorized, resp)
return
}
next(w, r)
}
}
按照官方文档推荐，在过滤器我使用了 httpx.Parse(r, &req) 获取请求体数据，导致后续 post 请求逻辑读取请求体失败，也就是说，这个方法中会消耗掉请求体流的数据，查看源码可以发现，这个方法是会读取请求中的 path，form，header，body 数据，赋值到我们的 struct 中：
func Parse(r *http.Request, v any) error {
kind := mapping.Deref(reflect.TypeOf(v)).Kind()
if kind != reflect.Array && kind != reflect.Slice {
if err := ParsePath(r, v); err != nil {
return err
}
if err := ParseForm(r, v); err != nil {
return err
}
if err := ParseHeaders(r, v); err != nil {
return err
}
}
if err := ParseJsonBody(r, v); err != nil {
return err
}
if valid, ok := v.(validation.Validator); ok {
return valid.Validate()
} else if val := validator.Load(); val != nil {
return val.(Validator).Validate(r, v)
}
return nil
}
继续往下查看源码可以发现，其中的 ParseForm 方法和 ParseJsonBody 方法就会消耗掉请求体流的数据，ParsePath 方法 ParseHeaders 方法则不会，而在这个过滤器中，我们只需要获取 header 的数据，body 的数据需要由后续逻辑获取，所以在这里需要修改过滤只获取 header 数据的方法，如果需要的是 path 数据也是同样的道理，修改后的过滤器：
func tokenFilter(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
type Request struct {
Token string `header:"token"`
}
var req Request
err := httpx.ParseHeaders(r, &req)
if err != nil || req.Token == "" {
resp, _ := utils.ToJsonString(models.ErrorResponseWithCode(http.StatusUnauthorized, "token required"))
routers.HttpError(w, http.StatusUnauthorized, resp)
return
}
next(w, r)
}
}
修改后后续节点即可正常获取请求体数据。
完整示例代码请参考：igolang
到此这篇关于go-zero读取请求体出现EOF错误的解决方法的文章就介绍到这了,更多相关解决go-zero读取请求体出现EOF错误内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！