gin/binding/form.go

90 lines
2.0 KiB
Go
Raw Normal View History

2015-03-31 15:51:10 +00:00
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
package binding
import (
"mime/multipart"
"net/http"
"reflect"
)
2015-03-31 15:51:10 +00:00
2017-09-04 01:15:50 +00:00
const defaultMemory = 32 * 1024 * 1024
type formBinding struct{}
type formPostBinding struct{}
type formMultipartBinding struct{}
2015-03-31 15:51:10 +00:00
2015-07-10 11:06:01 +00:00
func (formBinding) Name() string {
2015-05-26 14:31:05 +00:00
return "form"
2015-03-31 15:51:10 +00:00
}
2015-07-10 11:06:01 +00:00
func (formBinding) Bind(req *http.Request, obj interface{}) error {
2015-03-31 15:51:10 +00:00
if err := req.ParseForm(); err != nil {
return err
}
2019-01-18 01:32:53 +00:00
if err := req.ParseMultipartForm(defaultMemory); err != nil {
if err != http.ErrNotMultipart {
return err
}
}
2015-03-31 15:51:10 +00:00
if err := mapForm(obj, req.Form); err != nil {
return err
}
2015-05-31 14:30:00 +00:00
return validate(obj)
2015-03-31 15:51:10 +00:00
}
2015-07-10 11:06:01 +00:00
func (formPostBinding) Name() string {
return "form-urlencoded"
}
2015-07-10 11:06:01 +00:00
func (formPostBinding) Bind(req *http.Request, obj interface{}) error {
if err := req.ParseForm(); err != nil {
return err
}
if err := mapForm(obj, req.PostForm); err != nil {
return err
}
return validate(obj)
}
2015-07-10 11:06:01 +00:00
func (formMultipartBinding) Name() string {
return "multipart/form-data"
}
2015-07-10 11:06:01 +00:00
func (formMultipartBinding) Bind(req *http.Request, obj interface{}) error {
2017-09-04 01:15:50 +00:00
if err := req.ParseMultipartForm(defaultMemory); err != nil {
return err
}
if err := mappingByPtr(obj, (*multipartRequest)(req), "form"); err != nil {
return err
}
return validate(obj)
}
type multipartRequest http.Request
var _ setter = (*multipartRequest)(nil)
var (
multipartFileHeaderStructType = reflect.TypeOf(multipart.FileHeader{})
)
// TrySet tries to set a value by the multipart request with the binding a form file
func (r *multipartRequest) TrySet(value reflect.Value, field reflect.StructField, key string, opt setOptions) (isSetted bool, err error) {
if value.Type() == multipartFileHeaderStructType {
_, file, err := (*http.Request)(r).FormFile(key)
if err != nil {
return false, err
}
if file != nil {
value.Set(reflect.ValueOf(*file))
return true, nil
}
}
return setByForm(value, field, r.MultipartForm.Value, key, opt)
}