45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
package validator
|
||
|
||
import (
|
||
"errors"
|
||
"github.com/go-playground/locales/zh"
|
||
ut "github.com/go-playground/universal-translator"
|
||
"github.com/go-playground/validator/v10"
|
||
zhTranslations "github.com/go-playground/validator/v10/translations/zh"
|
||
"reflect"
|
||
)
|
||
|
||
var zhLocales = zh.New()
|
||
var validate = validator.New()
|
||
var uni = ut.New(zhLocales)
|
||
var trans, _ = uni.GetTranslator("zh")
|
||
|
||
// 验证器注册翻译器
|
||
var _ = zhTranslations.RegisterDefaultTranslations(validate, trans)
|
||
|
||
func init() {
|
||
//注册一个函数，获取struct tag里自定义的label作为字段名
|
||
validate.RegisterTagNameFunc(func(fld reflect.StructField) string {
|
||
if label := fld.Tag.Get("label"); len(label) > 0 {
|
||
return label
|
||
}
|
||
|
||
if jsonField := fld.Tag.Get("json"); len(jsonField) > 0 {
|
||
return jsonField
|
||
}
|
||
|
||
return fld.Name
|
||
})
|
||
}
|
||
|
||
func ValidateStruct(data interface{}) (err error) {
|
||
err = validate.Struct(data)
|
||
if err != nil {
|
||
for _, e := range err.(validator.ValidationErrors) {
|
||
err = errors.New(e.Translate(trans))
|
||
break
|
||
}
|
||
}
|
||
return
|
||
}
|