88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
package validation
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/go-playground/validator/v10"
|
|
)
|
|
|
|
// 初始化一个验证器实例
|
|
var validate = validator.New()
|
|
|
|
// 自定义验证规则
|
|
func init() {
|
|
_ = validate.RegisterValidation("telephone", func(fl validator.FieldLevel) bool {
|
|
if err := IsValidPhone(fl.Field().String()); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
})
|
|
_ = validate.RegisterValidation("dateonly", func(fl validator.FieldLevel) bool {
|
|
_, err := time.ParseInLocation("2006-01-02", fl.Field().String(), time.Local)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
})
|
|
}
|
|
|
|
func ValidateForm(s any) error {
|
|
// 验证结构体数据
|
|
if err := validate.Struct(s); err != nil {
|
|
var invalidValidationError *validator.InvalidValidationError
|
|
if errors.As(err, &invalidValidationError) {
|
|
return errors.New("验证器配置错误")
|
|
}
|
|
|
|
// 获取结构体的反射类型
|
|
t := reflect.TypeOf(s)
|
|
|
|
errorMessages := make([]string, 0)
|
|
for _, err := range err.(validator.ValidationErrors) {
|
|
// 获取字段名
|
|
fieldName := err.Field()
|
|
// 获取标签名
|
|
tag := err.Tag()
|
|
// 获取验证失败的字段值
|
|
// value := err.Value()
|
|
|
|
// 通过反射获取字段
|
|
field, found := t.Elem().FieldByName(fieldName)
|
|
errorMsg := fmt.Sprintf("[%s] 验证失败: %s", fieldName, translate(tag))
|
|
if found {
|
|
commentTag := field.Tag.Get("comment")
|
|
errorMsg = fmt.Sprintf("[%s] 验证失败: %s", commentTag, translate(tag))
|
|
}
|
|
|
|
errorMessages = append(errorMessages, errorMsg)
|
|
}
|
|
return errors.New(strings.Join(errorMessages, "; "))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func translate(s string) string {
|
|
switch s {
|
|
case "required":
|
|
return "不能为空"
|
|
case "min":
|
|
return "不能小于"
|
|
case "max":
|
|
return "不能大于"
|
|
case "email":
|
|
return "不是有效的邮箱地址"
|
|
case "telephone":
|
|
return "不是有效的手机号"
|
|
case "datetime":
|
|
return "不是有效的日期时间"
|
|
case "dateonly":
|
|
return "不是有效的日期"
|
|
}
|
|
return s
|
|
}
|