|
1
|
|
|
package it |
|
2
|
|
|
|
|
3
|
|
|
import ( |
|
4
|
|
|
"context" |
|
5
|
|
|
"strconv" |
|
6
|
|
|
|
|
7
|
|
|
"github.com/muonsoft/validation" |
|
8
|
|
|
) |
|
9
|
|
|
|
|
10
|
|
|
// CountConstraint checks that a given collection's (array, slice or a map) length is between some minimum and |
|
11
|
|
|
// maximum value. |
|
12
|
|
|
type CountConstraint struct { |
|
13
|
|
|
isIgnored bool |
|
14
|
|
|
checkMin bool |
|
15
|
|
|
checkMax bool |
|
16
|
|
|
checkDivisible bool |
|
17
|
|
|
min int |
|
18
|
|
|
max int |
|
19
|
|
|
divisibleBy int |
|
20
|
|
|
groups []string |
|
21
|
|
|
minErr error |
|
22
|
|
|
maxErr error |
|
23
|
|
|
exactErr error |
|
24
|
|
|
divisibleErr error |
|
25
|
|
|
minMessageTemplate string |
|
26
|
|
|
minMessageParameters validation.TemplateParameterList |
|
27
|
|
|
maxMessageTemplate string |
|
28
|
|
|
maxMessageParameters validation.TemplateParameterList |
|
29
|
|
|
exactMessageTemplate string |
|
30
|
|
|
exactMessageParameters validation.TemplateParameterList |
|
31
|
|
|
divisibleByMessageTemplate string |
|
32
|
|
|
divisibleByMessageParameters validation.TemplateParameterList |
|
33
|
1 |
|
} |
|
34
|
|
|
|
|
35
|
|
|
func newCountConstraint() CountConstraint { |
|
36
|
|
|
return CountConstraint{ |
|
37
|
|
|
minErr: validation.ErrTooFewElements, |
|
38
|
|
|
maxErr: validation.ErrTooManyElements, |
|
39
|
|
|
exactErr: validation.ErrNotExactCount, |
|
40
|
|
|
divisibleErr: validation.ErrNotDivisibleCount, |
|
41
|
|
|
minMessageTemplate: validation.ErrTooFewElements.Message(), |
|
42
|
|
|
maxMessageTemplate: validation.ErrTooManyElements.Message(), |
|
43
|
|
|
exactMessageTemplate: validation.ErrNotExactCount.Message(), |
|
44
|
|
|
divisibleByMessageTemplate: validation.ErrNotDivisibleCount.Message(), |
|
45
|
|
|
} |
|
46
|
|
|
} |
|
47
|
|
|
|
|
48
|
|
|
func newCountComparison(min int, max int, checkMin bool, checkMax bool) CountConstraint { |
|
49
|
|
|
c := newCountConstraint() |
|
50
|
1 |
|
c.min = min |
|
51
|
|
|
c.max = max |
|
52
|
|
|
c.checkMin = checkMin |
|
53
|
|
|
c.checkMax = checkMax |
|
54
|
|
|
|
|
55
|
|
|
return c |
|
56
|
1 |
|
} |
|
57
|
|
|
|
|
58
|
|
|
// HasMinCount creates a [CountConstraint] that checks the length of the iterable (slice, array, or map) |
|
59
|
|
|
// is greater than the minimum value. |
|
60
|
|
|
func HasMinCount(min int) CountConstraint { |
|
61
|
|
|
return newCountComparison(min, 0, true, false) |
|
62
|
1 |
|
} |
|
63
|
|
|
|
|
64
|
|
|
// HasMaxCount creates a [CountConstraint] that checks the length of the iterable (slice, array, or map) |
|
65
|
|
|
// is less than the maximum value. |
|
66
|
|
|
func HasMaxCount(max int) CountConstraint { |
|
67
|
|
|
return newCountComparison(0, max, false, true) |
|
68
|
1 |
|
} |
|
69
|
|
|
|
|
70
|
|
|
// HasCountBetween creates a [CountConstraint] that checks the length of the iterable (slice, array, or map) |
|
71
|
|
|
// is between some minimum and maximum value. |
|
72
|
|
|
func HasCountBetween(min int, max int) CountConstraint { |
|
73
|
1 |
|
return newCountComparison(min, max, true, true) |
|
74
|
|
|
} |
|
75
|
|
|
|
|
76
|
|
|
// HasExactCount creates a [CountConstraint] that checks the length of the iterable (slice, array, or map) |
|
77
|
|
|
// has exact value. |
|
78
|
|
|
func HasExactCount(count int) CountConstraint { |
|
79
|
|
|
return newCountComparison(count, count, true, true) |
|
80
|
|
|
} |
|
81
|
|
|
|
|
82
|
|
|
// HasCountDivisibleBy creates a [CountConstraint] that checks the length of the iterable (slice, array, or map) |
|
83
|
|
|
// is divisible by the specific value. |
|
84
|
1 |
|
func HasCountDivisibleBy(divisor int) CountConstraint { |
|
85
|
1 |
|
c := newCountConstraint() |
|
86
|
|
|
c.checkDivisible = true |
|
87
|
|
|
c.divisibleBy = divisor |
|
88
|
|
|
|
|
89
|
|
|
return c |
|
90
|
1 |
|
} |
|
91
|
1 |
|
|
|
92
|
|
|
// When enables conditional validation of this constraint. If the expression evaluates to false, |
|
93
|
|
|
// then the constraint will be ignored. |
|
94
|
|
|
func (c CountConstraint) When(condition bool) CountConstraint { |
|
95
|
|
|
c.isIgnored = !condition |
|
96
|
|
|
return c |
|
97
|
1 |
|
} |
|
98
|
1 |
|
|
|
99
|
|
|
// WhenGroups enables conditional validation of the constraint by using the validation groups. |
|
100
|
|
|
func (c CountConstraint) WhenGroups(groups ...string) CountConstraint { |
|
101
|
|
|
c.groups = groups |
|
102
|
|
|
return c |
|
103
|
|
|
} |
|
104
|
1 |
|
|
|
105
|
1 |
|
// WithMinError overrides default underlying error for violation that will be shown if the |
|
106
|
|
|
// collection length is less than the minimum value. |
|
107
|
|
|
func (c CountConstraint) WithMinError(err error) CountConstraint { |
|
108
|
|
|
c.minErr = err |
|
109
|
|
|
return c |
|
110
|
|
|
} |
|
111
|
1 |
|
|
|
112
|
1 |
|
// WithMaxError overrides default underlying error for violation that will be shown if the |
|
113
|
|
|
// collection length is greater than the maximum value. |
|
114
|
|
|
func (c CountConstraint) WithMaxError(err error) CountConstraint { |
|
115
|
|
|
c.maxErr = err |
|
116
|
|
|
return c |
|
117
|
|
|
} |
|
118
|
|
|
|
|
119
|
|
|
// WithExactError overrides default underlying error for violation that will be shown if minimum and |
|
120
|
|
|
// maximum values are equal and the length of the collection is not exactly this value. |
|
121
|
|
|
func (c CountConstraint) WithExactError(err error) CountConstraint { |
|
122
|
1 |
|
c.exactErr = err |
|
123
|
1 |
|
return c |
|
124
|
1 |
|
} |
|
125
|
|
|
|
|
126
|
|
|
// WithDivisibleError overrides default underlying error for violation that will be shown |
|
127
|
|
|
// the length of the collection is not divisible by specific value. |
|
128
|
|
|
func (c CountConstraint) WithDivisibleError(err error) CountConstraint { |
|
129
|
|
|
c.divisibleErr = err |
|
130
|
|
|
return c |
|
131
|
|
|
} |
|
132
|
|
|
|
|
133
|
|
|
// WithMinMessage sets the violation message that will be shown if the collection length is less than |
|
134
|
1 |
|
// the minimum value. You can set custom template parameters for injecting its values |
|
135
|
1 |
|
// into the final message. Also, you can use default parameters: |
|
136
|
1 |
|
// |
|
137
|
|
|
// {{ count }} - the current collection size; |
|
138
|
|
|
// {{ limit }} - the lower limit. |
|
139
|
|
|
func (c CountConstraint) WithMinMessage(template string, parameters ...validation.TemplateParameter) CountConstraint { |
|
140
|
|
|
c.minMessageTemplate = template |
|
141
|
|
|
c.minMessageParameters = parameters |
|
142
|
|
|
return c |
|
143
|
|
|
} |
|
144
|
|
|
|
|
145
|
|
|
// WithMaxMessage sets the violation message that will be shown if the collection length is greater than |
|
146
|
1 |
|
// the maximum value. You can set custom template parameters for injecting its values |
|
147
|
1 |
|
// into the final message. Also, you can use default parameters: |
|
148
|
1 |
|
// |
|
149
|
|
|
// {{ count }} - the current collection size; |
|
150
|
|
|
// {{ limit }} - the upper limit. |
|
151
|
|
|
func (c CountConstraint) WithMaxMessage(template string, parameters ...validation.TemplateParameter) CountConstraint { |
|
152
|
1 |
|
c.maxMessageTemplate = template |
|
153
|
|
|
c.maxMessageParameters = parameters |
|
154
|
|
|
return c |
|
155
|
|
|
} |
|
156
|
1 |
|
|
|
157
|
1 |
|
// WithExactMessage sets the violation message that will be shown if minimum and maximum values are equal and |
|
158
|
|
|
// the length of the collection is not exactly this value. You can set custom template parameters |
|
159
|
1 |
|
// for injecting its values into the final message. Also, you can use default parameters: |
|
160
|
1 |
|
// |
|
161
|
|
|
// {{ count }} - the current collection size; |
|
162
|
1 |
|
// {{ limit }} - the exact expected collection size. |
|
163
|
1 |
|
func (c CountConstraint) WithExactMessage(template string, parameters ...validation.TemplateParameter) CountConstraint { |
|
164
|
|
|
c.exactMessageTemplate = template |
|
165
|
|
|
c.exactMessageParameters = parameters |
|
166
|
1 |
|
return c |
|
167
|
|
|
} |
|
168
|
|
|
|
|
169
|
|
|
// WithDivisibleMessage sets the violation message that will be shown if the length of the collection |
|
170
|
|
|
// is not divisible by specific value. You can set custom template parameters |
|
171
|
|
|
// for injecting its values into the final message. Also, you can use default parameters: |
|
172
|
|
|
// |
|
173
|
|
|
// {{ count }} - the current collection size; |
|
174
|
|
|
// {{ divisibleBy }} - the divisor for the collection size. |
|
175
|
1 |
|
func (c CountConstraint) WithDivisibleMessage(template string, parameters ...validation.TemplateParameter) CountConstraint { |
|
176
|
1 |
|
c.divisibleByMessageTemplate = template |
|
177
|
1 |
|
c.divisibleByMessageParameters = parameters |
|
178
|
1 |
|
return c |
|
179
|
|
|
} |
|
180
|
|
|
|
|
181
|
1 |
|
func (c CountConstraint) ValidateCountable(ctx context.Context, validator *validation.Validator, count int) error { |
|
182
|
|
|
if c.isIgnored || validator.IsIgnoredForGroups(c.groups...) { |
|
183
|
|
|
return nil |
|
184
|
|
|
} |
|
185
|
|
|
if c.checkDivisible && count%c.divisibleBy != 0 { |
|
186
|
|
|
return c.newNotDivisibleViolation(ctx, validator, count) |
|
187
|
|
|
} |
|
188
|
|
|
if c.checkMax && count > c.max { |
|
189
|
|
|
return c.newViolation(ctx, validator, count, c.max, c.maxErr, c.maxMessageTemplate, c.maxMessageParameters) |
|
190
|
|
|
} |
|
191
|
|
|
if c.checkMin && count < c.min { |
|
192
|
|
|
return c.newViolation(ctx, validator, count, c.min, c.minErr, c.minMessageTemplate, c.minMessageParameters) |
|
193
|
|
|
} |
|
194
|
|
|
|
|
195
|
|
|
return nil |
|
196
|
|
|
} |
|
197
|
|
|
|
|
198
|
|
|
func (c CountConstraint) newViolation( |
|
199
|
|
|
ctx context.Context, |
|
200
|
|
|
validator *validation.Validator, |
|
201
|
|
|
count, limit int, |
|
202
|
|
|
err error, |
|
203
|
|
|
template string, |
|
204
|
|
|
parameters validation.TemplateParameterList, |
|
205
|
|
|
) validation.Violation { |
|
206
|
|
|
if c.checkMin && c.checkMax && c.min == c.max { |
|
207
|
|
|
template = c.exactMessageTemplate |
|
208
|
|
|
parameters = c.exactMessageParameters |
|
209
|
|
|
err = c.exactErr |
|
210
|
|
|
} |
|
211
|
|
|
|
|
212
|
|
|
return validator.BuildViolation(ctx, err, template). |
|
213
|
|
|
WithPluralCount(limit). |
|
214
|
|
|
WithParameters( |
|
215
|
|
|
parameters.Prepend( |
|
216
|
|
|
validation.TemplateParameter{Key: "{{ count }}", Value: strconv.Itoa(count)}, |
|
217
|
|
|
validation.TemplateParameter{Key: "{{ limit }}", Value: strconv.Itoa(limit)}, |
|
218
|
|
|
)..., |
|
219
|
|
|
). |
|
220
|
|
|
Create() |
|
221
|
|
|
} |
|
222
|
|
|
|
|
223
|
|
|
func (c CountConstraint) newNotDivisibleViolation( |
|
224
|
|
|
ctx context.Context, |
|
225
|
|
|
validator *validation.Validator, |
|
226
|
|
|
count int, |
|
227
|
|
|
) validation.Violation { |
|
228
|
|
|
return validator.BuildViolation(ctx, c.divisibleErr, c.divisibleByMessageTemplate). |
|
229
|
|
|
WithPluralCount(c.divisibleBy). |
|
230
|
|
|
WithParameters( |
|
231
|
|
|
c.divisibleByMessageParameters.Prepend( |
|
232
|
|
|
validation.TemplateParameter{Key: "{{ count }}", Value: strconv.Itoa(count)}, |
|
233
|
|
|
validation.TemplateParameter{Key: "{{ divisibleBy }}", Value: strconv.Itoa(c.divisibleBy)}, |
|
234
|
|
|
)..., |
|
235
|
|
|
). |
|
236
|
|
|
Create() |
|
237
|
|
|
} |
|
238
|
|
|
|