TyleoDvDelaware3217/GoDoxy
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2025-07-21 20:04:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

small refactor

Browse source
This commit is contained in:
yusing 2025-01-09 14:07:11 +08:00
parent 2ce1ceb460
commit 8109c9ac4f
4 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
internal/net/http/middleware/cidr_whitelist.go
View file

@ -34,7 +34,7 @@ var (
) )
func init() { func init() {
utils.Validator().RegisterValidation("status_code", func(fl validator.FieldLevel) bool { utils.MustRegisterValidation("status_code", func(fl validator.FieldLevel) bool {
statusCode := fl.Field().Int() statusCode := fl.Field().Int()
return gphttp.IsStatusCodeValid(int(statusCode)) return gphttp.IsStatusCodeValid(int(statusCode))
}) })

5
internal/notif/webhook.go
View file

@ -49,10 +49,7 @@ func jsonIfTemplateNotUsed(fl validator.FieldLevel) bool {
func init() { func init() {
utils.RegisterDefaultValueFactory(DefaultValue) utils.RegisterDefaultValueFactory(DefaultValue)
err := utils.Validator().RegisterValidation("jsonIfTemplateNotUsed", jsonIfTemplateNotUsed) utils.MustRegisterValidation("jsonIfTemplateNotUsed", jsonIfTemplateNotUsed)
if err != nil {
panic(err)
}
} }
// Name implements Provider. // Name implements Provider.

7
internal/utils/validation.go
View file

@ -12,3 +12,10 @@ var ErrValidationError = E.New("validation error")
func Validator() *validator.Validate { func Validator() *validator.Validate {
return validate return validate
} }
func MustRegisterValidation(tag string, fn validator.Func) {
err := validate.RegisterValidation(tag, fn)
if err != nil {
panic(err)
}
}