124 lines
3.2 KiB
Go
124 lines
3.2 KiB
Go
|
package libs
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"git.apinb.com/bsm-sdk/core/errcode"
|
|||
|
|
"github.com/go-playground/validator/v10"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
validate *validator.Validate
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
validate = validator.New()
|
|||
|
|
// 注册自定义验证规则
|
|||
|
|
registerCustomValidators()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// registerCustomValidators 注册自定义验证器
|
|||
|
|
func registerCustomValidators() {
|
|||
|
|
// 注册手机号验证
|
|||
|
|
validate.RegisterValidation("phone", validatePhone)
|
|||
|
|
// 注册邮箱验证
|
|||
|
|
validate.RegisterValidation("email_format", validateEmail)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// validatePhone 验证手机号（11位数字，以1开头）
|
|||
|
|
func validatePhone(fl validator.FieldLevel) bool {
|
|||
|
|
phone := fl.Field().String()
|
|||
|
|
// 空字符串由omitempty或required处理，这里返回true
|
|||
|
|
if phone == "" {
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
if len(phone) != 11 {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
if len(phone) > 0 && phone[0] != '1' {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
for _, char := range phone {
|
|||
|
|
if char < '0' || char > '9' {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// validateEmail 验证邮箱格式（简单的邮箱格式验证）
|
|||
|
|
func validateEmail(fl validator.FieldLevel) bool {
|
|||
|
|
email := fl.Field().String()
|
|||
|
|
if email == "" {
|
|||
|
|
return true // 允许为空，使用omitempty处理
|
|||
|
|
}
|
|||
|
|
parts := strings.Split(email, "@")
|
|||
|
|
if len(parts) != 2 {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
if len(parts[0]) == 0 || len(parts[1]) == 0 {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
if !strings.Contains(parts[1], ".") {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ValidateStruct 验证结构体，返回格式化的错误信息
|
|||
|
|
func ValidateStruct(s interface{}) error {
|
|||
|
|
if err := validate.Struct(s); err != nil {
|
|||
|
|
return formatValidationError(err)
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// formatValidationError 格式化验证错误信息
|
|||
|
|
func formatValidationError(err error) error {
|
|||
|
|
if validationErrors, ok := err.(validator.ValidationErrors); ok {
|
|||
|
|
var errMsg strings.Builder
|
|||
|
|
errMsg.WriteString("参数验证失败: ")
|
|||
|
|
for i, validationError := range validationErrors {
|
|||
|
|
if i > 0 {
|
|||
|
|
errMsg.WriteString("; ")
|
|||
|
|
}
|
|||
|
|
errMsg.WriteString(getFieldErrorMsg(validationError))
|
|||
|
|
}
|
|||
|
|
// 返回ErrInvalidArgument错误
|
|||
|
|
return errcode.ErrInvalidArgument
|
|||
|
|
}
|
|||
|
|
return errcode.ErrInvalidArgument
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getFieldErrorMsg 获取字段错误信息
|
|||
|
|
func getFieldErrorMsg(fe validator.FieldError) string {
|
|||
|
|
field := fe.Field()
|
|||
|
|
tag := fe.Tag()
|
|||
|
|
|
|||
|
|
switch tag {
|
|||
|
|
case "required":
|
|||
|
|
return fmt.Sprintf("%s 字段为必填项", field)
|
|||
|
|
case "min":
|
|||
|
|
return fmt.Sprintf("%s 字段长度不能小于 %s", field, fe.Param())
|
|||
|
|
case "max":
|
|||
|
|
return fmt.Sprintf("%s 字段长度不能大于 %s", field, fe.Param())
|
|||
|
|
case "len":
|
|||
|
|
return fmt.Sprintf("%s 字段长度必须为 %s", field, fe.Param())
|
|||
|
|
case "email_format":
|
|||
|
|
return fmt.Sprintf("%s 字段邮箱格式不正确", field)
|
|||
|
|
case "phone":
|
|||
|
|
return fmt.Sprintf("%s 字段手机号格式不正确（应为11位数字）", field)
|
|||
|
|
case "gte":
|
|||
|
|
return fmt.Sprintf("%s 字段值必须大于或等于 %s", field, fe.Param())
|
|||
|
|
case "lte":
|
|||
|
|
return fmt.Sprintf("%s 字段值必须小于或等于 %s", field, fe.Param())
|
|||
|
|
case "gt":
|
|||
|
|
return fmt.Sprintf("%s 字段值必须大于 %s", field, fe.Param())
|
|||
|
|
case "lt":
|
|||
|
|
return fmt.Sprintf("%s 字段值必须小于 %s", field, fe.Param())
|
|||
|
|
default:
|
|||
|
|
return fmt.Sprintf("%s 字段验证失败 (%s)", field, tag)
|
|||
|
|
}
|
|||
|
|
}
|