通过gin.Context 进行数据绑定报错EOF

1.背景

由于需要通过接口请求参数进行鉴权,需要提取其中的参数。
假设该参数为uid,其既可能出现在get请求:

curl --location --request GET 'http://127.0.0.1:8080/add?uid=222'

也可能出现在post请求中以json方式传递:

curl --location --request POST 'http://127.0.0.1:8080/add' \
--header 'Content-Type: application/json' \
--data-raw '{
  "uid":222
}'

项目采用的web框架为gin

2.问题描述

定义了一个中间件,里面包含了如下代码:
(1)提取get请求携带的参数

uid := c.Query("uid") 
uID, _ := strconv.ParseUint(uidStr, 10, 64)

(2)提取post请求中json结构体的参数

if uID == 0 {//如果请求的url没有携带参数,尝试解析body里面的
    u := &User{}
    err := c.Bind(u)
    if err != nil {
        log.Fatal("err", err)
    }
    uID = u.UID
}

type User struct {
    UID uint64 `json:"uid"`
}

然而,发现post请求返回的结果是这样的:

{
    "message": "EOF"
}

但没有报错信息

3.原因

ctx.Bind重复调用了。bind一次后,再调用bind就会失败。
可以看到bind方法底层是这样的,decode会去把请求body的内容读空:

func decodeJSON(r io.Reader, obj interface{}) error {
    decoder := json.NewDecoder(r)
         //...其他代码
    if err := decoder.Decode(obj); err != nil {
        return err
    }
        //...
}

而上述代码在中间件里面进行了一次bind, 而还有一个中间件也会进行一次bind,所以导致报错EOF
跟之前遇到的这个问题类似:https://www.jianshu.com/p/610c3f27d2ca

4.解决

如果想要可以重复调用数据绑定,可以选择ShouldBindBodyWith方法:

u := &User{}
c.ShouldBindBodyWith(&u, binding.JSON)

该方法底层进行了处理,会把首次读取的body存在context里面,下次如果context里面的值不为空,会从context里面取值,而不会再去读取body:

func (c *Context) ShouldBindBodyWith(obj interface{}, bb binding.BindingBody) (err error) {
    var body []byte
    if cb, ok := c.Get(BodyBytesKey); ok {
        if cbb, ok := cb.([]byte); ok {
            body = cbb
        }
    }
    if body == nil {
        body, err = ioutil.ReadAll(c.Request.Body)
        if err != nil {
            return err
        }
        c.Set(BodyBytesKey, body)
    }
    return bb.BindBody(body, obj)
}

参考:
[1]https://juejin.cn/post/6844903830669262855
[2]https://github.com/gin-gonic/gin/issues/439

你可能感兴趣的:(通过gin.Context 进行数据绑定报错EOF)