-
Notifications
You must be signed in to change notification settings - Fork 458
/
set.go
120 lines (99 loc) · 1.95 KB
/
set.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package selector
import (
"bytes"
"github.com/spiffe/spire/proto/spire/common"
)
type Set interface {
Raw() []*common.Selector
Array() []*Selector
Power() <-chan Set
Equal(otherSet Set) bool
Includes(selector *Selector) bool
IncludesSet(s2 Set) bool
Add(selector *Selector)
Remove(selector *Selector) *Selector
String() string
Size() int
}
type set map[Selector]*Selector
func NewSet(selectors ...*Selector) Set {
set := set{}
for _, cs := range selectors {
set.Add(cs)
}
return &set
}
func NewSetFromRaw(c []*common.Selector) Set {
set := set{}
for _, cs := range c {
s := &Selector{
Type: cs.Type,
Value: cs.Value,
}
set.Add(s)
}
return &set
}
func (s *set) Raw() []*common.Selector {
c := []*common.Selector{}
for _, selector := range *s {
cs := &common.Selector{
Type: selector.Type,
Value: selector.Value,
}
c = append(c, cs)
}
return c
}
// Array returns an array with the elements of the set in any order.
func (s *set) Array() []*Selector {
c := []*Selector{}
for _, selector := range *s {
c = append(c, selector)
}
return c
}
func (s *set) Power() <-chan Set {
return PowerSet(s)
}
func (s *set) Equal(otherSet Set) bool {
return EqualSet(s, otherSet.(*set))
}
func (s *set) Includes(selector *Selector) bool {
return Includes(s, selector)
}
func (s *set) IncludesSet(s2 Set) bool {
return IncludesSet(s, s2.(*set))
}
func (s *set) Add(selector *Selector) {
(*s)[*selector] = selector
}
func (s *set) Remove(selector *Selector) *Selector {
key := *selector
if removed, ok := (*s)[key]; ok {
delete(*s, key)
return removed
}
return nil
}
func (s *set) Size() int {
return len(*s)
}
func (s *set) String() string {
var b bytes.Buffer
b.WriteString("[")
if len(*s) > 0 {
i := 0
for _, selector := range *s {
if i > 0 {
b.WriteString(" ")
}
b.WriteString(selector.Type)
b.WriteString(":")
b.WriteString(selector.Value)
i++
}
}
b.WriteString("]")
return b.String()
}