gf/g/container/gset/gset_interface_set.go

// 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.
//

package gset

import (
	"fmt"
	"gitee.com/johng/gf/g/container/internal/rwmutex"
)

type InterfaceSet struct {
	mu *rwmutex.RWMutex
	m  map[interface{}]struct{}
}

func NewInterfaceSet(safe...bool) *InterfaceSet {
	return &InterfaceSet{
		m  : make(map[interface{}]struct{}),
		mu : rwmutex.New(safe...),
    }
}

// 给定回调函数对原始内容进行遍历，回调函数返回true表示继续遍历，否则停止遍历
func (this *InterfaceSet) Iterator(f func (v interface{}) bool) {
    this.mu.RLock()
    defer this.mu.RUnlock()
    for k, _ := range this.m {
		if !f(k) {
			break
		}
    }
}

// 添加
func (this *InterfaceSet) Add(item interface{}) *InterfaceSet {
	this.mu.Lock()
	this.m[item] = struct{}{}
	this.mu.Unlock()
	return this
}

// 批量添加
func (this *InterfaceSet) BatchAdd(items []interface{}) *InterfaceSet {
	this.mu.Lock()
	for _, item := range items {
		this.m[item] = struct{}{}
	}
	this.mu.Unlock()
    return this
}

// 键是否存在
func (this *InterfaceSet) Contains(item interface{}) bool {
	this.mu.RLock()
	_, exists := this.m[item]
	this.mu.RUnlock()
	return exists
}

// 删除键值对
func (this *InterfaceSet) Remove(key interface{}) {
	this.mu.Lock()
	delete(this.m, key)
	this.mu.Unlock()
}

// 大小
func (this *InterfaceSet) Size() int {
	this.mu.RLock()
	l := len(this.m)
	this.mu.RUnlock()
	return l
}

// 清空set
func (this *InterfaceSet) Clear() {
	this.mu.Lock()
	this.m = make(map[interface{}]struct{})
	this.mu.Unlock()
}

// 转换为数组
func (this *InterfaceSet) Slice() []interface{} {
	this.mu.RLock()
	i   := 0
	ret := make([]interface{}, len(this.m))
	for item := range this.m {
		ret[i] = item
		i++
	}
	this.mu.RUnlock()
	return ret
}

// 转换为字符串
func (this *InterfaceSet) String() string {
	return fmt.Sprint(this.Slice())
}

func (this *InterfaceSet) LockFunc(f func(m map[interface{}]struct{})) {
	this.mu.Lock(true)
	defer this.mu.Unlock(true)
	f(this.m)
}

func (this *InterfaceSet) RLockFunc(f func(m map[interface{}]struct{})) {
	this.mu.RLock(true)
	defer this.mu.RUnlock(true)
	f(this.m)
}