79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
|
package utils
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"reflect"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin/binding"
|
|||
|
|
"github.com/go-playground/locales/en"
|
|||
|
|
"github.com/go-playground/locales/zh"
|
|||
|
|
ut "github.com/go-playground/universal-translator"
|
|||
|
|
"github.com/go-playground/validator/v10"
|
|||
|
|
enTranslations "github.com/go-playground/validator/v10/translations/en"
|
|||
|
|
zhTranslations "github.com/go-playground/validator/v10/translations/zh"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var Trans ut.Translator
|
|||
|
|
|
|||
|
|
// InitTrans 初始化翻译器
|
|||
|
|
func InitTrans(locale string) (err error) {
|
|||
|
|
// 修改gin框架中的Validator引擎属性，实现自定制
|
|||
|
|
if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
|
|||
|
|
// 注册一个获取json tag的自定义方法
|
|||
|
|
v.RegisterTagNameFunc(func(fld reflect.StructField) string {
|
|||
|
|
name := strings.SplitN(fld.Tag.Get("columnName"), ",", 2)[0]
|
|||
|
|
if name == "" {
|
|||
|
|
name = strings.SplitN(fld.Tag.Get("json"), ",", 2)[0]
|
|||
|
|
}
|
|||
|
|
if name == "-" {
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
return name
|
|||
|
|
})
|
|||
|
|
// 为SignUpParam注册自定义校验方法
|
|||
|
|
zhT := zh.New() // 中文翻译器
|
|||
|
|
enT := en.New() // 英文翻译器
|
|||
|
|
uni := ut.New(enT, zhT, enT)
|
|||
|
|
var ok bool
|
|||
|
|
// 也可以使用 uni.FindTranslator(...) 传入多个locale进行查找
|
|||
|
|
Trans, ok = uni.GetTranslator(locale)
|
|||
|
|
if !ok {
|
|||
|
|
return fmt.Errorf("uni.GetTranslator(%s) failed", locale)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 注册翻译器
|
|||
|
|
switch locale {
|
|||
|
|
case "en":
|
|||
|
|
err = enTranslations.RegisterDefaultTranslations(v, Trans)
|
|||
|
|
case "zh":
|
|||
|
|
err = zhTranslations.RegisterDefaultTranslations(v, Trans)
|
|||
|
|
default:
|
|||
|
|
err = enTranslations.RegisterDefaultTranslations(v, Trans)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RemoveTopStruct 去除提示信息中的结构体名称
|
|||
|
|
func RemoveTopStruct(fields map[string]string) map[string]string {
|
|||
|
|
res := map[string]string{}
|
|||
|
|
for field, err := range fields {
|
|||
|
|
res[field[strings.Index(field, ".")+1:]] = err
|
|||
|
|
}
|
|||
|
|
return res
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func ValidatorErrorMsg(err error) string {
|
|||
|
|
var errs validator.ValidationErrors
|
|||
|
|
_ = errors.As(err, &errs)
|
|||
|
|
msgMap := RemoveTopStruct(errs.Translate(Trans))
|
|||
|
|
var msg string
|
|||
|
|
for _, v := range msgMap {
|
|||
|
|
msg += v + ";"
|
|||
|
|
}
|
|||
|
|
return msg
|
|||
|
|
}
|