gf/g/container/gset/int_set.go

105 lines
1.9 KiB
Go
Raw Normal View History

2017-12-29 16:03:30 +08:00
// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://gitee.com/johng/gf.
//
//
2017-12-31 18:19:58 +08:00
2017-11-23 10:21:28 +08:00
package gset
import (
"fmt"
"gitee.com/johng/gf/g/container/internal/rwmutex"
2017-11-23 10:21:28 +08:00
)
type IntSet struct {
mu *rwmutex.RWMutex
m map[int]struct{}
2017-11-23 10:21:28 +08:00
}
func NewIntSet(safe...bool) *IntSet {
return &IntSet{
m : make(map[int]struct{}),
mu : rwmutex.New(safe...),
}
2017-11-23 10:21:28 +08:00
}
// 给定回调函数对原始内容进行遍历回调函数返回true表示继续遍历否则停止遍历
2018-04-11 16:06:45 +08:00
func (this *IntSet) Iterator(f func (v int) bool) {
this.mu.RLock()
for k, _ := range this.m {
2018-04-11 16:06:45 +08:00
if !f(k) {
break
}
}
this.mu.RUnlock()
}
2017-11-23 10:21:28 +08:00
// 设置键
func (this *IntSet) Add(item int) *IntSet {
this.mu.Lock()
this.m[item] = struct{}{}
this.mu.Unlock()
2017-11-23 10:21:28 +08:00
return this
}
// 批量添加设置键
func (this *IntSet) BatchAdd(items []int) *IntSet {
2018-02-24 16:57:37 +08:00
this.mu.Lock()
for _, item := range items {
this.m[item] = struct{}{}
}
this.mu.Unlock()
2017-11-23 10:21:28 +08:00
return this
}
// 键是否存在
func (this *IntSet) Contains(item int) bool {
this.mu.RLock()
_, exists := this.m[item]
this.mu.RUnlock()
2017-11-23 10:21:28 +08:00
return exists
}
// 删除键值对
func (this *IntSet) Remove(key int) {
this.mu.Lock()
delete(this.m, key)
this.mu.Unlock()
2017-11-23 10:21:28 +08:00
}
// 大小
func (this *IntSet) Size() int {
this.mu.RLock()
l := len(this.m)
this.mu.RUnlock()
2017-11-23 10:21:28 +08:00
return l
}
// 清空set
func (this *IntSet) Clear() {
this.mu.Lock()
this.m = make(map[int]struct{})
this.mu.Unlock()
2017-11-23 10:21:28 +08:00
}
// 转换为数组
func (this *IntSet) Slice() []int {
this.mu.RLock()
ret := make([]int, len(this.m))
2017-11-23 10:21:28 +08:00
i := 0
for item := range this.m {
2017-11-23 10:21:28 +08:00
ret[i] = item
i++
}
this.mu.RUnlock()
2017-11-23 10:21:28 +08:00
return ret
}
// 转换为字符串
func (this *IntSet) String() string {
return fmt.Sprint(this.Slice())
2017-11-23 10:21:28 +08:00
}