Go实现集合set

package main
//支持任意类型添加
import (
	"fmt"
	"sync"
)

type inter interface{}

type Set struct {
	m map[inter]bool
	sync.RWMutex
}

func New() *Set {
	return &Set{
		m: map[inter]bool{},
	}
}
func (s *Set) Add(item inter) {
	s.Lock()
	defer s.Unlock()
	s.m[item] = true
}
func (s *Set) Remove(item inter) {
	s.Lock()
	s.Unlock()
	delete(s.m, item)
}
func (s *Set) Has(item inter) bool {
	s.RLock()
	defer s.RUnlock()
	_, ok := s.m[item]
	return ok
}
func (s *Set) Len() int {
	return len(s.List())
}
func (s *Set) Clear() {
	s.Lock()
	defer s.Unlock()
	s.m = map[inter]bool{}
}
func (s *Set) IsEmpty() bool {
	if s.Len() == 0 {
		return true
	}
	return false
}
func (s *Set) List() []inter {
	s.RLock()
	defer s.RUnlock()
	list := []inter{}
	for item := range s.m {
		list = append(list, item)
	}
	return list
}

func main() {
	s := New()
	s.Add(11)
	s.Add("aa")
	for i, k := range s.List() {
		fmt.Println(i, k)
	}
}

  

原文地址:https://www.cnblogs.com/huangxiaohen/p/4281325.html