GoFrame gset-基本使用

基本使用

package main

import (
 "github.com/gogf/gf/v2/container/gset"
 "fmt"
)

func main() {
 // 创建一个并发安全的集合对象
 s := gset.New(true)

 // 添加数据项
 s.Add(1)

 // 批量添加数据项
 s.Add([]interface{}{1, 2, 3}...)

 // 集合数据项大小
 fmt.Println(s.Size())

 // 集合中是否存在指定数据项
 fmt.Println(s.Contains(2))

 // 返回数据项slice
 fmt.Println(s.Slice())

 // 删除数据项
 s.Remove(3)

 // 遍历数据项
 s.Iterator(func(v interface{}) bool {
  fmt.Println("Iterator:", v)
  return true
 })

 // 将集合转换为字符串
 fmt.Println(s.String())

 // 并发安全写锁操作
 s.LockFunc(func(m map[interface{}]struct{}) {
  m[4] = struct{}{}
 })

 // 并发安全读锁操作
 s.RLockFunc(func(m map[interface{}]struct{}) {
  fmt.Println(m)
 })

 // 清空集合
 s.Clear()
 fmt.Println(s.Size())
}

执行后，输出结果为：

3
true
[1 2 3]
Iterator: 1
Iterator: 2
[1 2]
map[1:{} 2:{} 4:{}]
0

交差并补集

我们可以使用以下方法实现交差并补集，并返回一个新的结果集合

func (set *Set) Intersect(others ...*Set) (newSet *Set)
func (set *Set) Diff(others ...*Set) (newSet *Set)
func (set *Set) Union(others ...*Set) (newSet *Set)
func (set *Set) Complement(full *Set) (newSet *Set)

Intersect: 交集，属于set且属于others的元素为元素的集合。
Diff: 差集，属于set且不属于others的元素为元素的集合。
Union: 并集，属于set或属于others的元素为元素的集合。
Complement: 补集，(前提: set应当为full的子集)属于全集full不属于集合set的元素组成的集合。如果给定的full集合不是set的全集时，返回full与set的差集.

通过集合方法我们可以发现，交差并集方法支持多个集合参数进行计算。以下为简化示例，只使用一个参数集合。

package main

import (
 "fmt"
 "github.com/gogf/gf/v2/frame/g"
 "github.com/gogf/gf/v2/container/gset"
)

func main() {
 s1 := gset.NewFrom(g.Slice{1, 2, 3})
 s2 := gset.NewFrom(g.Slice{4, 5, 6})
 s3 := gset.NewFrom(g.Slice{1, 2, 3, 4, 5, 6, 7})

 // 交集
 fmt.Println(s3.Intersect(s1).Slice())
 // 差集
 fmt.Println(s3.Diff(s1).Slice())
 // 并集
 fmt.Println(s1.Union(s2).Slice())
 // 补集
 fmt.Println(s1.Complement(s3).Slice())
}

执行后，输出结果为：

[1 2 3]
[4 5 6 7]
[1 2 3 4 5 6]
[7 4 5 6]

Contains/ContainsI包含判断

package main

import (
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
)

func main() {
 var set gset.StrSet
 set.Add("a")
 fmt.Println(set.Contains("a"))
 fmt.Println(set.Contains("A"))
 fmt.Println(set.ContainsI("A"))

 // Output:
 // true
 // false
 // true
}

Pop/Pops集合项出栈

package main

import (
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
)

func main() {
 var set gset.Set
 set.Add(1, 2, 3, 4)
 fmt.Println(set.Pop())
 fmt.Println(set.Pops(2))
 fmt.Println(set.Size())

 // May Output:
 // 1
 // [2 3]
 // 1
}

Join集合项串连

package main

import (
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
)

func main() {
 var set gset.Set
 set.Add("a", "b", "c", "d")
 fmt.Println(set.Join(","))

 // May Output:
 // a,b,c,d
}

IsSubsetOf子集判断

package main

import (
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
 "github.com/gogf/gf/v2/frame/g"
)

func main() {
 var s1, s2 gset.Set
 s1.Add(g.Slice{1, 2, 3}...)
 s2.Add(g.Slice{2, 3}...)
 fmt.Println(s1.IsSubsetOf(&s2))
 fmt.Println(s2.IsSubsetOf(&s1))

 // Output:
 // false
 // true
}

AddIfNotExist*判断性写入

判断性写入是指当指定的数据项不存在时则写入并且方法返回true，否则忽略吸入并且方法返回false。相关方法如下：

AddIfNotExist
AddIfNotExistFunc
AddIfNotExistFuncLock

方法具体描述请查看接口文档或源码注释。

package main

import (
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
)

func main() {
 var set gset.Set
 fmt.Println(set.AddIfNotExist(1))
 fmt.Println(set.AddIfNotExist(1))
 fmt.Println(set.Slice())

 // Output:
 // true
 // false
 // [1]
}

Walk遍历修改

package main

import (
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
 "github.com/gogf/gf/v2/frame/g"
)

func main() {
 var (
  set    gset.StrSet
  names  = g.SliceStr{"user", "user_detail"}
  prefix = "gf_"
 )
 set.Add(names...)
 // Add prefix for given table names.
 set.Walk(func(item string) string {
  return prefix + item
 })
 fmt.Println(set.Slice())

 // May Output:
 // [gf_user gf_user_detail]
}

JSON序列化/反序列

gset模块下的所有容器类型均实现了标准库json数据格式的序列化/反序列化接口。

Marshal

package main

import (
 "encoding/json"
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
)

func main() {
 type Student struct {
  Id     int
  Name   string
  Scores *gset.IntSet
 }
 s := Student{
  Id:     1,
  Name:   "john",
  Scores: gset.NewIntSetFrom([]int{100, 99, 98}),
 }
 b, _ := json.Marshal(s)
 fmt.Println(string(b))
}

执行后，终端输出：

{"Id":1,"Name":"john","Scores":[100,99,98]}

Unmarshal

package main


import (
 "encoding/json"
 "fmt"
 "github.com/gogf/gf/v2/container/gset"
)


func main() {
 b := []byte(`{"Id":1,"Name":"john","Scores":[100,99,98]}`)
 type Student struct {
  Id     int
  Name   string
  Scores *gset.IntSet
 }
 s := Student{}
 json.Unmarshal(b, &s)
 fmt.Println(s)
}

执行后，输出结果：