You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
850 B
51 lines
850 B
8 years ago
|
package main
|
||
|
|
||
|
import (
|
||
|
"reflect"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/gin-gonic/gin/binding"
|
||
|
"gopkg.in/go-playground/validator.v9"
|
||
|
)
|
||
|
|
||
|
type defaultValidator struct {
|
||
|
once sync.Once
|
||
|
validate *validator.Validate
|
||
|
}
|
||
|
|
||
|
var _ binding.StructValidator = &defaultValidator{}
|
||
|
|
||
|
func (v *defaultValidator) ValidateStruct(obj interface{}) error {
|
||
|
|
||
|
if kindOfData(obj) == reflect.Struct {
|
||
|
|
||
|
v.lazyinit()
|
||
|
|
||
|
if err := v.validate.Struct(obj); err != nil {
|
||
|
return error(err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (v *defaultValidator) lazyinit() {
|
||
|
v.once.Do(func() {
|
||
|
v.validate = validator.New()
|
||
|
v.validate.SetTagName("binding")
|
||
|
|
||
|
// add any custom validations etc. here
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func kindOfData(data interface{}) reflect.Kind {
|
||
|
|
||
|
value := reflect.ValueOf(data)
|
||
|
valueType := value.Kind()
|
||
|
|
||
|
if valueType == reflect.Ptr {
|
||
|
valueType = value.Elem().Kind()
|
||
|
}
|
||
|
return valueType
|
||
|
}
|