Update context.go

This commit is contained in:
thinkerou 2020-05-03 23:44:50 +08:00 committed by GitHub
parent 285d15eb70
commit 4dc9902a97
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -36,7 +36,8 @@ const (
MIMEYAML = binding.MIMEYAML MIMEYAML = binding.MIMEYAML
) )
const bodyBytesKey = "_gin-gonic/gin/bodybyteskey" // BodyBytesKey indicates a default body bind key.
const BodyBytesKey = "_gin-gonic/gin/bodybyteskey"
const abortIndex int8 = math.MaxInt8 / 2 const abortIndex int8 = math.MaxInt8 / 2
@ -687,7 +688,7 @@ func (c *Context) ShouldBindWith(obj interface{}, b binding.Binding) error {
// ShouldBindWith for better performance if you need to call only once. // ShouldBindWith for better performance if you need to call only once.
func (c *Context) ShouldBindBodyWith(obj interface{}, bb binding.BindingBody) (err error) { func (c *Context) ShouldBindBodyWith(obj interface{}, bb binding.BindingBody) (err error) {
var body []byte var body []byte
if cb, ok := c.Get(bodyBytesKey); ok { if cb, ok := c.Get(BodyBytesKey); ok {
if cbb, ok := cb.([]byte); ok { if cbb, ok := cb.([]byte); ok {
body = cbb body = cbb
} }
@ -697,7 +698,7 @@ func (c *Context) ShouldBindBodyWith(obj interface{}, bb binding.BindingBody) (e
if err != nil { if err != nil {
return err return err
} }
c.Set(bodyBytesKey, body) c.Set(BodyBytesKey, body)
} }
return bb.BindBody(body, obj) return bb.BindBody(body, obj)
} }