package set
|
|
// StringSet 定义一个字符串集合类型，使用 map 实现
|
type StringSet map[string]struct{}
|
|
func NewStringSet() StringSet {
|
return make(StringSet)
|
}
|
|
// Add 添加元素到集合中
|
func (s StringSet) Add(elements ...string) {
|
for _, element := range elements {
|
s[element] = struct{}{}
|
}
|
}
|
|
// Remove 删除集合中的元素
|
func (s StringSet) Remove(element string) {
|
delete(s, element)
|
}
|
|
// Contains 检查元素是否存在于集合中
|
func (s StringSet) Contains(element string) bool {
|
_, exists := s[element]
|
return exists
|
}
|
|
// Size 返回集合的大小
|
func (s StringSet) Size() int {
|
return len(s)
|
}
|
|
// Elements 返回集合中的所有元素
|
func (s StringSet) Elements() []string {
|
var elements []string
|
for key := range s {
|
elements = append(elements, key)
|
}
|
return elements
|
}
|
|
// Iterate 遍历集合中的所有元素
|
func (s StringSet) Iterate(f func(string)) {
|
for key := range s {
|
f(key)
|
}
|
}
|
