feat: validator add RegisterValidation for extend (#32)

validator/extend.go

@@ -1,9 +1,5 @@
 package validator
 
-import (
-	govalidator "github.com/go-playground/validator/v10"
-)
-
-var extendValidators = map[string]govalidator.Func{
+var extendValidators = map[string]Func{
 	// add your extend extend validators
 }

validator/validator.go

@@ -13,6 +13,7 @@ var (
 
 type (
 	Validator interface {
+		RegisterValidation(tag string, fn Func, callValidationEvenIfNull ...bool) error
 		Struct(s interface{}) error
 		Var(field interface{}, tag string) error
 	}
@@ -22,18 +23,27 @@ type (
 	}
 
 	// alias
+	FieldLevel             = govalidator.FieldLevel
+	Func                   = func(fl FieldLevel) bool
 	InvalidValidationError = govalidator.InvalidValidationError
 	ValidationErrors       = govalidator.ValidationErrors
 )
 
 func New() Validator {
-	validate := govalidator.New()
-	for k, val := range extendValidators {
-		_ = validate.RegisterValidation(k, val)
+	v := &defaultValidator{
+		Validate: govalidator.New(),
 	}
-	return &defaultValidator{
-		Validate: validate,
+
+	for k, val := range extendValidators {
+		_ = v.RegisterValidation(k, val)
 	}
+
+	return v
+}
+
+func RegisterValidation(tag string, fn Func, callValidationEvenIfNull ...bool) error {
+	initGValidator()
+	return gValidator.RegisterValidation(tag, fn, callValidationEvenIfNull...)
 }
 
 func Struct(s interface{}) error {
@@ -46,6 +56,10 @@ func Var(field interface{}, tag string) error {
 	return gValidator.Var(field, tag)
 }
 
+func (v *defaultValidator) RegisterValidation(tag string, fn Func, callValidationEvenIfNull ...bool) error {
+	return v.Validate.RegisterValidation(tag, fn, callValidationEvenIfNull...)
+}
+
 func (v *defaultValidator) Struct(s interface{}) error {
 	return v.Validate.Struct(s)
 }

validator/validator_test.go

@@ -9,6 +9,36 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestRegisterValidation(t *testing.T) {
+	var (
+		err error
+		ast = assert.New(t)
+	)
+
+	_ = RegisterValidation("streq", func(fl FieldLevel) bool {
+		field := fl.Field()
+		param := fl.Param()
+		if field.Kind() == reflect.String {
+			return field.String() == param
+		}
+		return false
+	})
+
+	type testStruct struct {
+		Data string `validate:"streq=aaa"`
+	}
+	err = Struct(testStruct{})
+	if errs, ok := err.(govalidator.ValidationErrors); ast.True(ok) {
+		ast.Len(errs, 1)
+	}
+	err = Struct(testStruct{Data: "aa"})
+	if errs, ok := err.(govalidator.ValidationErrors); ast.True(ok) {
+		ast.Len(errs, 1)
+	}
+	err = Struct(testStruct{Data: "aaa"})
+	ast.NoError(err)
+}
+
 func TestStruct(t *testing.T) {
 	var (
 		err error
@@ -56,8 +86,8 @@ func TestVar(t *testing.T) {
 }
 
 func TestExtendValidators(t *testing.T) {
-	stubs := gostub.Stub(&extendValidators, map[string]govalidator.Func{
-		"streq": func(fl govalidator.FieldLevel) bool {
+	stubs := gostub.Stub(&extendValidators, map[string]Func{
+		"streq": func(fl FieldLevel) bool {
 			field := fl.Field()
 			param := fl.Param()
 			if field.Kind() == reflect.String {