forked from src-d/borges
/
set.go
63 lines (50 loc) · 733 Bytes
/
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
package tool
import (
"sort"
"sync"
)
type Set struct {
l map[string]struct{}
m *sync.RWMutex
}
func NewSet(threadSafe bool) *Set {
var m *sync.RWMutex
if threadSafe {
m = new(sync.RWMutex)
}
return &Set{
l: make(map[string]struct{}),
m: m,
}
}
func (s *Set) Add(name string) {
if s.m != nil {
s.m.Lock()
defer s.m.Unlock()
}
s.l[name] = struct{}{}
}
func (s *Set) Contains(name string) bool {
if s.m != nil {
s.m.RLock()
defer s.m.RUnlock()
}
_, ok := s.l[name]
return ok
}
func (s *Set) List() []string {
l := make([]string, len(s.l))
if s.m != nil {
s.m.RLock()
}
var i int
for k := range s.l {
l[i] = k
i++
}
if s.m != nil {
s.m.RUnlock()
}
sort.Strings(l)
return l
}