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