starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-03-26 02:51:07 +01:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/api/handler.go
View File

@@ -58,4 +58,4 @@ func checkHost(f http.HandlerFunc) http.HandlerFunc {
LogDebug(r).Interface("headers", r.Header).Msg("API request")
f(w, r)
}
}
}

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

@@ -34,7 +34,7 @@ var (
)
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()
return gphttp.IsStatusCodeValid(int(statusCode))
})

5
internal/notif/webhook.go
View File

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

7
internal/utils/validation.go
View File

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