/
validator.go
60 lines (50 loc) · 1.44 KB
/
validator.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package validator
import (
"regexp"
"github.com/EduOJ/backend/base/log"
zhTranslations "github.com/EduOJ/backend/base/validator/translations/zh"
zhLocal "github.com/go-playground/locales/zh"
ut "github.com/go-playground/universal-translator"
"github.com/go-playground/validator/v10"
"github.com/pkg/errors"
)
type echoValidator struct {
V *validator.Validate
}
func (cv *echoValidator) Validate(i interface{}) error {
return cv.V.Struct(i)
}
var Trans ut.Translator
func init() {
zh := zhLocal.New()
uni := ut.New(zh, zh)
var found bool
Trans, found = uni.GetTranslator("zh")
if !found {
log.Fatal("could not found zh translator")
panic("could not found zh translator")
}
}
func New() *validator.Validate {
v := validator.New()
// add custom translation here
if err := zhTranslations.RegisterDefaultTranslations(v, Trans); err != nil {
log.Fatal(errors.Wrap(err, "could not register default translations"))
panic(errors.Wrap(err, "could not register default translations"))
}
err := v.RegisterValidation("username", validateUsername)
if err != nil {
log.Fatal(errors.Wrap(err, "could not register validation"))
panic(errors.Wrap(err, "could not register validation"))
}
return v
}
func NewEchoValidator() *echoValidator {
return &echoValidator{
V: New(),
}
}
var usernameRegex = regexp.MustCompile("^[a-zA-Z0-9_]+$")
func validateUsername(fl validator.FieldLevel) bool {
return usernameRegex.MatchString(fl.Field().String())
}