blob: 97f70bc3d8d2be93ef032b18a58e3df90b92aac1 (
plain)
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
|
// Copyright 2024 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package collections
type Set[T comparable] map[T]struct{}
func (s Set[T]) Add(el ...T) {
for _, el := range el {
s[el] = struct{}{}
}
}
func (s Set[T]) Remove(el T) bool {
if !s.Contains(el) {
return false
}
delete(s, el)
return true
}
func (s Set[T]) Contains(el T) bool {
_, ok := s[el]
return ok
}
func (s Set[T]) ToSlice() []T {
sl := []T{}
for el := range s {
sl = append(sl, el)
}
return sl
}
func (s Set[T]) Clone() Set[T] {
newSet := Set[T]{}
for el := range s {
newSet.Add(el)
}
return newSet
}
func (s Set[T]) Equals(rhs Set[T]) bool {
if len(s) != len(rhs) {
return false
}
rhsClone := rhs.Clone()
for el := range s {
rhsClone.Remove(el)
}
return len(rhsClone) == 0
}
func SliceToSet[T comparable](sl []T) Set[T] {
set := Set[T]{}
for _, el := range sl {
set.Add(el)
}
return set
}
|
