示例:使用空接口实现可以保存任意值的字典
空接口可以保存任何类型这个特性可以方便地用于容器的设计。
下面例子使用 map 和 interface{} 实现了一个字典。字典在其他语言中的功能和 map 类似,可以将任意类型的值做成键值对保存,然后进行找回,遍历操作。
代码1-1 实现字典(/dict/dict.go)
1、值设置和获取
字典内部拥有一个 data 字段,其类型为 map。这个 map 的键和值都是 interface{} 类型,也就是实现任意类型关联任意类型。字典的值设置和获取通过 Set() 和 Get() 两个方法来完成,参数都是 interface{}。详细代码如下:
//字典结构
type Dictionary struct {
data map[interface{}]interface{} //键值都为 interface{} 类型
}
//根据键获取值
func (d *Dictionary) Get(key interface{}) (interface{}, error) {
return d.data[key], nil
}
//设置键值
func (d *Dictionary) Set(key interface{}, value interface{}) {
d.data[key] = value
}
2、遍历字段的所有键值关联数据
每个容器都有遍历操作。遍历时,需要提供一个回调返回需要遍历的数据。为了方便在必要时终止遍历操作,可以将回调的返回值设置为 bool 类型,外部逻辑在回调中不需要遍历时直接返回 false 即可终止遍历。
Dictionary 的 Visit() 方法需要传入回调函数,回调函数的类型为 func(k, v interface{}) bool。每次遍历时获得的键值关联数据通过回调函数的 k 和 v 参数返回。 Visit 的详细实现如下:
//遍历所有的键值,如果回调返回值为 false,则停止遍历
//定义回调,类型为func(k,v interface{}) bool,意思是返回键值数据(k、v)。
//bool 表示遍历流程控制,返回 true 时继续遍历,返回 false 时终止遍历
func (d *Dictionary) Visit(callback func(k, v interface{}) bool) {
//当 callback 为空的时候,退出遍历,避免后续代码访问空的 callback 而导致的崩溃
if callback == nil {
return
}
//遍历字典结构的 data 成员,也就是遍历 map 的所有元素
for k, v := range d.data {
//根据 callback 的返回值,决定是否继续遍历,还是退出。
if !callback(k, v) {
return
}
}
}
3、初始化和清除
字典结构含有 map,需要在创建 Dictionary 实例时初始化 map。这个过程通过 Dictionary 的 Clear() 方法完成。在 NewDictionary 中调用 Clear() 方法避免了 map 初始化过程的代码复用问题。代码如下:
//清空所有的数据
func (d *Dictionary) Clear() {
d.data = make(map[interface{}]interface{})
}
//创建一个字典
func NewDdict() *Dictionary {
d := &Dictionary{}
//初始化 map
d.Clear()
return d
}
4、使用字典
字典实现完成后,需要经过一个测试过程,查看这个字典是否存在问题。
将一些字符串和数值组合放入到字典中,然后再从字典中根据键查询出对应的值,接着再遍历一个字典中所有的元素。
func main() {
//创建字典实例
dict := NewDictionary()
//添加游戏数据
dict.Set("My Factory", 60)
dict.Set("Terra Craft", 36)
dict.Set("Don't Hungry", 24)
//获取值及打印值
favorites, err := dict.Get("Terra Craft")
if err != nil {
fmt.Println("favorites:", favorites)
}
//遍历所有的字典元素
dict.Visit(func(key, value interface{}) bool {
//将值转为 int 类型,并判断值是否大于 40
if value.(int) > 40 {
//输出 "很贵"
fmt.Println(key, "很贵!")
return true
}
//默认都是输出 "很便宜"
fmt.Println(key, "很便宜!")
return true
})
}
5、完成程序代码
package main
import "fmt"
//字典结构
type Dictionary struct {
data map[interface{}]interface{} //键值都为 interface{} 类型
}
//根据键获取值
func (d *Dictionary) Get(key interface{}) (interface{}, error) {
return d.data[key], nil
}
//设置键值
func (d *Dictionary) Set(key interface{}, value interface{}) {
d.data[key] = value
}
//遍历所有的键值,如果回调返回值为 false,则停止遍历
//定义回调,类型为func(k,v interface{}) bool,意思是返回键值数据(k、v)。
//bool 表示遍历流程控制,返回 true 时继续遍历,返回 false 时终止遍历
func (d *Dictionary) Visit(callback func(k, v interface{}) bool) {
//当 callback 为空的时候,退出遍历,避免后续代码访问空的 callback 而导致的崩溃
if callback == nil {
return
}
//遍历字典结构的 data 成员,也就是遍历 map 的所有元素
for k, v := range d.data {
//根据 callback 的返回值,决定是否继续遍历,还是退出。
if !callback(k, v) {
return
}
}
}
//清空所有的数据
func (d *Dictionary) Clear() {
d.data = make(map[interface{}]interface{})
}
//创建一个字典
func NewDictionary() *Dictionary {
d := &Dictionary{}
//初始化 map
d.Clear()
return d
}
func main() {
//创建字典实例
dict := NewDictionary()
//添加游戏数据
dict.Set("My Factory", 60)
dict.Set("Terra Craft", 36)
dict.Set("Don't Hungry", 24)
//获取值及打印值
favorites, err := dict.Get("Terra Craft")
if err != nil {
fmt.Println("favorites:", favorites)
}
//遍历所有的字典元素
dict.Visit(func(key, value interface{}) bool {
//将值转为 int 类型,并判断值是否大于 40
if value.(int) > 40 {
//输出 "很贵"
fmt.Println(key, "很贵!")
return true
}
//默认都是输出 "很便宜"
fmt.Println(key, "很便宜!")
return true
})
}
运行代码,输出如下:
GOROOT=C:\Program Files\Go #gosetup
GOPATH=D:\go-testfiles #gosetup
"C:\Program Files\Go\bin\go.exe" build -o C:\Users\zuoyang\AppData\Local\Temp\GoLand\___go_build___go.exe D:\go-testfiles\接口-示例:使用空接口实现可以保存任意值的字典.go #gosetup
C:\Users\zuoyang\AppData\Local\Temp\GoLand\___go_build___go.exe
Terra Craft 很便宜!
Don't Hungry 很便宜!
My Factory 很贵!
进程 已完成,退出代码为 0