113 lines
3.2 KiB
Go
113 lines
3.2 KiB
Go
package validator
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/go-playground/validator/v10"
|
|
"regexp"
|
|
"strconv"
|
|
)
|
|
|
|
var validate *validator.Validate
|
|
|
|
// 初始化自定义校验器
|
|
func init() {
|
|
validate = validator.New()
|
|
|
|
if err := validate.RegisterValidation("validName", validName); err != nil {
|
|
panic(fmt.Sprintf("注册 validName 验证器时发生错误: %v", err))
|
|
}
|
|
|
|
// 注册自定义验证器 validMobileNo
|
|
if err := validate.RegisterValidation("validMobileNo", validMobileNo); err != nil {
|
|
panic(fmt.Sprintf("注册 validMobileNo 验证器时发生错误: %v", err))
|
|
}
|
|
|
|
// 注册自定义验证器 validDate
|
|
if err := validate.RegisterValidation("validDate", validDate); err != nil {
|
|
panic(fmt.Sprintf("注册 validDate 验证器时发生错误: %v", err))
|
|
}
|
|
|
|
// 注册自定义验证器 validIDCard
|
|
if err := validate.RegisterValidation("validIDCard", validIDCard); err != nil {
|
|
panic(fmt.Sprintf("注册 validIDCard 验证器时发生错误: %v", err))
|
|
}
|
|
|
|
// 注册自定义验证器 validTimeRange
|
|
if err := validate.RegisterValidation("validTimeRange", validTimeRange); err != nil {
|
|
panic(fmt.Sprintf("注册 validTimeRange 验证器时发生错误: %v", err))
|
|
}
|
|
}
|
|
|
|
// ValidateAndParse 封装了解密、解析和校验逻辑
|
|
func ValidateAndParse(decryptData []byte, req interface{}) error {
|
|
// 解析解密后的 JSON 数据
|
|
if err := json.Unmarshal(decryptData, req); err != nil {
|
|
return errors.New("解密后的数据格式不正确")
|
|
}
|
|
|
|
// 校验
|
|
v := GetValidator()
|
|
if err := v.Struct(req); err != nil {
|
|
for _, validationErr := range err.(validator.ValidationErrors) {
|
|
field := validationErr.StructField()
|
|
tag := validationErr.Tag()
|
|
return errors.New(GetErrorMessage(field, tag))
|
|
}
|
|
}
|
|
|
|
// 如果没有错误，返回 nil
|
|
return nil
|
|
}
|
|
|
|
// 获取验证器实例
|
|
func GetValidator() *validator.Validate {
|
|
return validate
|
|
}
|
|
|
|
// 自定义的名称验证
|
|
func validName(fl validator.FieldLevel) bool {
|
|
name := fl.Field().String()
|
|
validNamePattern := `^[\p{Han}]+$`
|
|
matched, _ := regexp.MatchString(validNamePattern, name)
|
|
return matched
|
|
}
|
|
|
|
// 自定义的手机号验证
|
|
func validMobileNo(fl validator.FieldLevel) bool {
|
|
phone := fl.Field().String()
|
|
validMobileNoPattern := `^1[3-9]\d{9}$`
|
|
matched, _ := regexp.MatchString(validMobileNoPattern, phone)
|
|
return matched
|
|
}
|
|
|
|
// 自定义正则表达式校验 yyyyMMdd 格式
|
|
func validDate(fl validator.FieldLevel) bool {
|
|
date := fl.Field().String()
|
|
validDatePattern := `^\d{4}(0[1-9]|1[0-2])(0[1-9]|[12][0-9]|3[01])$`
|
|
matched, _ := regexp.MatchString(validDatePattern, date)
|
|
return matched
|
|
}
|
|
|
|
// 自定义身份证校验
|
|
func validIDCard(fl validator.FieldLevel) bool {
|
|
id := fl.Field().String()
|
|
validIDPattern := `^\d{15}$|^\d{18}$|^\d{17}(\d|X|x)$` // 匹配15位或18位身份证
|
|
matched, _ := regexp.MatchString(validIDPattern, id)
|
|
return matched
|
|
}
|
|
|
|
// 自定义 time_range 校验 (1-5年)
|
|
func validTimeRange(fl validator.FieldLevel) bool {
|
|
timeRange := fl.Field().String()
|
|
if timeRange == "" {
|
|
return true // 如果为空，认为是有效的，因为是非必填项
|
|
}
|
|
value, err := strconv.Atoi(timeRange)
|
|
if err != nil || value < 1 || value > 5 {
|
|
return false
|
|
}
|
|
return true
|
|
}
|