Huangzj
60 lines
1014 B
Go
60 lines
1014 B
Go
/*
|
||
* @Author : huangzj
|
||
* @Time : 2020/6/15 16:16
|
||
* @Description：
|
||
*/
|
||
|
||
package set
|
||
|
||
import "sync"
|
||
|
||
type ISet interface {
|
||
Add(i interface{})
|
||
AddAll(list ...interface{})
|
||
Remove(i interface{})
|
||
Contains(i interface{}) bool
|
||
IsEmpty() bool
|
||
GetAllSet() []interface{}
|
||
}
|
||
|
||
type Set struct {
|
||
setMap sync.Map //用来存储元素的map
|
||
}
|
||
|
||
func (set *Set) Add(i interface{}) {
|
||
set.setMap.Store(i, nil)
|
||
}
|
||
|
||
func (set *Set) AddAll(list ...interface{}) {
|
||
for _, item := range list {
|
||
set.Add(item)
|
||
}
|
||
}
|
||
|
||
func (set *Set) Remove(i interface{}) {
|
||
set.setMap.Delete(i)
|
||
}
|
||
|
||
func (set *Set) Contains(i interface{}) bool {
|
||
_, has := set.setMap.Load(i)
|
||
return has
|
||
}
|
||
|
||
func (set *Set) IsEmpty() bool {
|
||
length := 0
|
||
set.setMap.Range(func(key, value interface{}) bool {
|
||
length = length + 1
|
||
return true
|
||
})
|
||
return length == 0
|
||
}
|
||
|
||
func (set *Set) GetAllSet() []interface{} {
|
||
list := make([]interface{}, 0)
|
||
set.setMap.Range(func(key, value interface{}) bool {
|
||
list = append(list, key)
|
||
return true
|
||
})
|
||
return list
|
||
}
|