/
validator.go
85 lines (70 loc) · 2.21 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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
// Copyright (c) 2016 Yandex LLC. All rights reserved.
// Author: Vladimir Skipor <skipor@yandex-team.ru>
package config
import (
"github.com/pkg/errors"
validator "gopkg.in/bluesuncorp/validator.v9"
)
var validations = []struct {
key string
val validator.Func
}{
{"min-time", MinTimeValidation},
{"max-time", MaxTimeValidation},
{"min-size", MinSizeValidation},
{"max-size", MaxSizeValidation},
}
var stringValidations = []struct {
key string
val StringValidation
}{
{"endpoint", EndpointStringValidation},
{"url-path", URLPathStringValidation},
}
var defaultValidator = newValidator()
func Validate(value interface{}) error {
return errors.WithStack(defaultValidator.Struct(value))
}
func newValidator() *validator.Validate {
validate := validator.New()
validate.SetTagName("validate")
for _, val := range validations {
_ = validate.RegisterValidation(val.key, val.val)
}
for _, val := range stringValidations {
_ = validate.RegisterValidation(val.key, StringToAbstractValidation(val.val))
}
return validate
}
// RegisterCustom used to set custom validation check hooks on specific types,
// that will be called on such type validation, even if it is nested field.
func RegisterCustom(v CustomValidation, types ...interface{}) (_ struct{}) {
if len(types) < 1 {
panic("should be registered for at least one type")
}
defaultValidator.RegisterStructValidation(func(sl validator.StructLevel) {
v(structLevelHandle{sl})
}, types...)
return
}
type StringValidation func(value string) bool
// StringToAbstractValidation wraps StringValidation into validator.Func.
func StringToAbstractValidation(sv StringValidation) validator.Func {
return func(fl validator.FieldLevel) bool {
if strVal, ok := fl.Field().Interface().(string); ok {
return sv(strVal)
}
return false
}
}
type ValidateHandle interface {
Value() interface{}
ReportError(field, reason string)
}
type CustomValidation func(h ValidateHandle)
type structLevelHandle struct{ validator.StructLevel }
var _ ValidateHandle = structLevelHandle{}
func (sl structLevelHandle) Value() interface{} { return sl.Current().Interface() }
func (sl structLevelHandle) ReportError(field, reason string) {
sl.StructLevel.ReportError(nil, field, "", reason, "")
}