-
Notifications
You must be signed in to change notification settings - Fork 0
/
enum_pii.go
77 lines (61 loc) · 1.52 KB
/
enum_pii.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
package client
import "fmt"
type piiSet struct {
Block PII
Replace PII
}
// PIIs represents the set of PIIs that can be used.
var PIIs = piiSet{
Block: newPII("block"),
Replace: newPII("replace"),
}
// Parse parses the string value and returns a PII if one exists.
func (piiSet) Parse(value string) (PII, error) {
pii, exists := piis[value]
if !exists {
return PII{}, fmt.Errorf("invalid pii %q", value)
}
return pii, nil
}
// MustParse parses the string value and returns a pii if one exists. If
// an error occurs the function panics.
func (piiSet) MustParse(value string) PII {
pii, err := PIIs.Parse(value)
if err != nil {
panic(err)
}
return pii
}
// =============================================================================
// Set of known PIIs.
var piis = make(map[string]PII)
// PII represents a PII in the system.
type PII struct {
value string
}
func newPII(pii string) PII {
p := PII{pii}
piis[pii] = p
return p
}
// String returns the name of the PII.
func (p PII) String() string {
return p.value
}
// UnmarshalText implement the unmarshal interface for JSON conversions.
func (p *PII) UnmarshalText(data []byte) error {
pii, err := PIIs.Parse(string(data))
if err != nil {
return err
}
p.value = pii.value
return nil
}
// MarshalText implement the marshal interface for JSON conversions.
func (p PII) MarshalText() ([]byte, error) {
return []byte(p.value), nil
}
// Equal provides support for the go-cmp package and testing.
func (p PII) Equal(p2 PII) bool {
return p.value == p2.value
}