-
Notifications
You must be signed in to change notification settings - Fork 34
/
validator_helper_map.go
55 lines (49 loc) · 1.07 KB
/
validator_helper_map.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
package validator
import (
"context"
"fmt"
"github.com/machinefi/w3bstream/pkg/depends/kit/validator/rules"
)
func MapRuleParam(ctx context.Context, r *Rule) (key, elem Validator, err error) {
if r.Params == nil {
return
}
if len(r.Params) != 2 {
err = fmt.Errorf(
"map should only 2 parameter, but got %d", len(r.Params),
)
return
}
compiler := FactoryFromContext(ctx)
for i, param := range r.Params {
switch rule := param.(type) {
case *rules.Rule:
switch i {
case 0:
key, err = compiler.Compile(ctx, rule.RAW, r.Type.Key())
if err != nil {
err = fmt.Errorf("map key %s", err)
return
}
case 1:
elem, err = compiler.Compile(ctx, rule.RAW, r.Type.Elem())
if err != nil {
err = fmt.Errorf("map elem %s", err)
return
}
}
case *rules.Lit:
raw := rule.Bytes()
if len(raw) > 0 {
err = fmt.Errorf("map parameter should be a valid rule")
return
}
elem, err = compiler.Compile(ctx, raw, r.Type.Elem())
if err != nil {
err = fmt.Errorf("map elem %s", err)
return
}
}
}
return
}