wireguard-dashboard/component/validator.go

package component

import (
	"errors"
	"gitee.ltd/lxh/logger/log"
	"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"
	zhTranslations "github.com/go-playground/validator/v10/translations/zh"
	"reflect"
	"strings"
)

var validatorTrans ut.Translator

func init() {
	initValidatorTranslator()
}

func Error(err error) string {
	var errs validator.ValidationErrors
	ok := errors.As(err, &errs)
	if !ok {
		return err.Error()
	}

	errMap := errs.Translate(validatorTrans)

	var errMsg string
	for _, v := range errMap {
		errMsg = v
	}

	return errMsg
}

// Validate
// @description: 校验
// @param data
// @return string
func Validate(data any) error {
	if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
		return v.Struct(data)
	}

	return nil
}

// initValidatorTranslator
// @description: 初始化翻译机
// @receiver vli
func initValidatorTranslator() {
	//修改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("label"), ",", 2)[0]
			if name == "-" {
				return ""
			}
			return name
		})

		zhT := zh.New() //中文翻译器
		enT := en.New() //英文翻译器

		// 第一个参数是备用（fallback）的语言环境
		// 后面的参数是应该支持的语言环境（支持多个）
		// uni := ut.New(zhT, zhT) 也是可以的
		uni := ut.New(enT, zhT, enT)

		// locale 通常取决于 http 请求头的 'Accept-Language'
		var ok bool
		// 也可以使用 uni.FindTranslator(...) 传入多个locale进行查找
		validatorTrans, ok = uni.GetTranslator("zh")
		if !ok {
			log.Errorf("获取翻译机失败")
			return
		}

		err := overrideTranslator(v, validatorTrans)
		if err != nil {
			log.Errorf("覆盖原有翻译失败: %v", err.Error())
			return
		}
	}

}

// overrideTranslator
// @description: 覆盖原有翻译
// @param v
// @param translator
// @return error
func overrideTranslator(v *validator.Validate, translator ut.Translator) error {
	err := zhTranslations.RegisterDefaultTranslations(v, translator)
	if err != nil {
		return err
	}

	return nil
}