GoFrame GSet使用技巧：出栈、子集判断、序列化、遍历修改

2023-09-24 05:28:01

GSet：GoFrame的高级集合类型

简介

GSet 是 GoFrame 提供的一款功能强大的集合类型，它与 Go 标准库中的 set 包类似，但提供了更多特性。本文将深入探讨 GSet 的使用方法，包括出栈、子集判断、序列化和遍历修改等高级技巧。

随机出栈

GSet 提供了 Pop 方法，可从集合中随机删除一个元素并返回该元素。然而，只有 slice、map 和 chan 这三种数据类型支持随机出栈。代码示例：

package main

import (
	"fmt"

	"github.com/goframework/gf/container/gset"
)

func main() {
	set := gset.New(gset.TypeSlice)
	set.Add("a", "b", "c", "d", "e")

	for i := 0; i < 5; i++ {
		v, _ := set.Pop()
		fmt.Println(v)
	}
}

指定顺序出栈

GSet 还提供 Shift 方法，可从集合中删除第一个元素并返回该元素。Shift 方法支持所有数据类型。代码示例：

package main

import (
	"fmt"

	"github.com/goframework/gf/container/gset"
)

func main() {
	set := gset.New(gset.TypeSlice)
	set.Add("a", "b", "c", "d", "e")

	for i := 0; i < 5; i++ {
		v, _ := set.Shift()
		fmt.Println(v)
	}
}

子集判断

GSet 提供 IsSubsetOf 方法，可判断一个集合是否是另一个集合的子集。代码示例：

package main

import (
	"fmt"

	"github.com/goframework/gf/container/gset"
)

func main() {
	set1 := gset.New(gset.TypeSlice)
	set1.Add("a", "b", "c")

	set2 := gset.New(gset.TypeSlice)
	set2.Add("a", "b", "c", "d", "e")

	fmt.Println(set1.IsSubsetOf(set2)) // true
}

序列化

GSet 提供序列化和反序列化方法，可将集合中的数据存储到文件中或从文件中读取数据。代码示例：

package main

import (
	"fmt"

	"github.com/goframework/gf/container/gset"
)

func main() {
	set := gset.New(gset.TypeSlice)
	set.Add("a", "b", "c")

	// 序列化
	data, _ := set.MarshalJSON()

	// 反序列化
	newSet := gset.NewFromJSON(data)

	fmt.Println(newSet) // [a b c]
}

遍历修改

GSet 提供 Walk 方法，可遍历集合中的元素并对每个元素进行修改。代码示例：

package main

import (
	"fmt"

	"github.com/goframework/gf/container/gset"
)

func main() {
	set := gset.New(gset.TypeSlice)
	set.Add("a", "b", "c")

	// 遍历修改
	set.Walk(func(v interface{}) {
		v.(string) += "!"
	})

	fmt.Println(set) // [a! b! c!]
}