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.
74 lines
1.7 KiB
74 lines
1.7 KiB
package base
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin/binding"
|
|
"github.com/go-playground/locales/zh"
|
|
ut "github.com/go-playground/universal-translator"
|
|
"github.com/go-playground/validator/v10"
|
|
zh_translations "github.com/go-playground/validator/v10/translations/zh"
|
|
"log"
|
|
"reflect"
|
|
"strings"
|
|
)
|
|
|
|
var translator ut.Translator
|
|
|
|
type ErrMsg = map[string]string
|
|
|
|
// 自定义验证器的错误消息映射表
|
|
var customMsg = map[string]string{
|
|
//"exists": "{0}对应的记录不存在",
|
|
}
|
|
|
|
// initValidateMsg 初始验证器消息处理器（中间件)
|
|
func initValidateMsg() {
|
|
uni := ut.New(zh.New())
|
|
translator, _ = uni.GetTranslator("zh")
|
|
v := binding.Validator.Engine().(*validator.Validate)
|
|
_ = zh_translations.RegisterDefaultTranslations(v, translator)
|
|
// 注册自定验证器的翻译
|
|
tf := func(ut ut.Translator, fe validator.FieldError) string {
|
|
msg, err := ut.T(fe.Tag(), fe.Field())
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
return msg
|
|
}
|
|
for tag, text := range customMsg {
|
|
_ = v.RegisterTranslation(tag, translator, func(trans ut.Translator) error {
|
|
_ = trans.Add(tag, text, false)
|
|
return nil
|
|
}, tf)
|
|
}
|
|
|
|
v.RegisterTagNameFunc(func(fld reflect.StructField) string {
|
|
name := ""
|
|
tagJson := fld.Tag.Get("json")
|
|
if len(tagJson) > 0 {
|
|
tmpName := strings.SplitN(tagJson, ",", 2)[0]
|
|
if tmpName != "-" {
|
|
name = tmpName
|
|
}
|
|
}
|
|
return name
|
|
})
|
|
}
|
|
|
|
// Translate 翻译错误消息的方法
|
|
func Translate(err error) map[string]string {
|
|
errs, ok := err.(validator.ValidationErrors)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
errMsg := make(ErrMsg, len(errs))
|
|
for _, e := range errs {
|
|
errMsg[e.Field()] = e.Translate(translator)
|
|
}
|
|
return errMsg
|
|
}
|
|
|
|
func init() {
|
|
//
|
|
initValidateMsg()
|
|
}
|