base-golang/pkg/validator/validator.go

56 lines
1.1 KiB
Go
Raw Normal View History

2024-07-23 10:23:43 +08:00
package validator
import (
"reflect"
"sync"
"github.com/gin-gonic/gin/binding"
"github.com/go-playground/validator/v10"
)
type DefaultValidator struct {
once sync.Once
validate *validator.Validate
}
var Validator binding.StructValidator = &DefaultValidator{}
func (v *DefaultValidator) ValidateStruct(obj any) error {
if kindOfData(obj) == reflect.Struct {
v.lazyinit()
if err := v.validate.Struct(obj); err != nil {
return err
}
}
return nil
}
func (v *DefaultValidator) Engine() any {
v.lazyinit()
return v.validate
}
func (v *DefaultValidator) lazyinit() {
v.once.Do(func() {
v.validate = validator.New()
v.validate.SetTagName("validate")
_ = v.validate.RegisterValidation("chinesePhone", chinesePhone)
_ = v.validate.RegisterValidation("isoPhone", isoPhone)
_ = v.validate.RegisterValidation("idCard", idCard)
_ = v.validate.RegisterValidation("chineseName", chineseName)
_ = v.validate.RegisterValidation("year", year)
})
}
func kindOfData(data any) reflect.Kind {
value := reflect.ValueOf(data)
valueType := value.Kind()
if valueType == reflect.Ptr {
valueType = value.Elem().Kind()
}
return valueType
}