golang高效Set集合实现插件库goset的使用

为什么选择goset？

Go语言本身没有内置Set集合实现，目前常用的set实现是golang-set，但它的API设计不够友好。例如从切片创建Set时：

import "github.com/deckarep/golang-set"

func main() {
    ints := []int{1, 2, 3, 4}
    mapset.NewSet(ints...)  // 不能工作
    mapset.NewSetFromSlice(ints)  // 不能工作
    mapset.NewSetWith(ints...)  // 不能工作
}

会出现错误：

cannot use ints (type []int) as type []interface{}

必须手动将[]int转换为[]interface{}，这很不优雅。因此goset提供了更好的解决方案。

goset使用示例

基本用法

import "github.com/zoumo/goset"

func main() {
    // 从多个元素创建Set
    goset.NewSet(1, 2, 3, 4)
    
    // 从切片创建Set
    goset.NewSetFrom([]int{1, 2, 3, 4})

    // 字符串Set
    goset.NewSet("1", "2", "3")
    goset.NewSetFrom([]string{"1", "2", "3"})
}

完整API示例

package main

import (
    "fmt"
    "github.com/zoumo/goset"
)

func main() {
    // 创建Set
    set1 := goset.NewSet(1, 2, 3, 4)
    set2 := goset.NewSet(3, 4, 5, 6)

    // 添加元素
    set1.Add(5)
    
    // 删除元素
    set1.Remove(1)
    
    // 检查元素是否存在
    fmt.Println(set1.Contains(2))  // true
    fmt.Println(set1.Contains(10)) // false
    
    // 集合运算
    union := set1.Unite(set2)      // 并集
    intersect := set1.Intersect(set2) // 交集
    diff := set1.Diff(set2)        // 差集
    
    // 转换为切片
    ints := set1.ToInts()
    fmt.Println(ints)
    
    // 遍历元素
    set1.Range(func(index int, elem interface{}) bool {
        fmt.Printf("%d: %v\n", index, elem)
        return true
    })
    
    // 线程安全/不安全转换
    safeSet := set1.ToThreadSafe()
    unsafeSet := safeSet.ToThreadUnsafe()
}

Set接口完整定义

type Set interface {
    // 添加元素
    Add(elem ...interface{}) error
    
    // 从切片/数组添加元素
    Extend(b interface{}) error
    
    // 删除元素
    Remove(elem ...interface{})
    
    // 检查元素是否存在
    Contains(elem interface{}) bool
    ContainsAll(elems ...interface{}) bool
    ContainsAny(elems ...interface{}) bool
    
    // 集合操作
    Diff(b Set) Set             // 差集
    SymmetricDiff(b Set) Set    // 对称差集
    Unite(b Set) Set            // 并集
    Intersect(b Set) Set        // 交集
    
    // 转换
    ToThreadSafe() Set         // 转为线程安全
    ToThreadUnsafe() Set       // 转为非线程安全
    
    // 其他方法
    Copy() Set                 // 复制
    Len() int                  // 元素数量
    String() string            // 字符串表示
    Range(foreach func(index int, elem interface{}) bool) // 遍历
    
    // 转换为切片
    ToStrings() []string
    ToInts() []int
    Elements() []interface{}
}

goset提供了线程安全和非线程安全两种实现，可以方便地进行集合运算和类型转换，是Go语言中处理集合问题的好选择。