Go从入门到精通——接口(interface)——示例:使用空接口实现可以保存任意值的字典

示例:使用空接口实现可以保存任意值的字典

  空接口可以保存任何类型这个特性可以方便地用于容器的设计。

  下面例子使用 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
posted @ 2022-06-04 22:47  左扬  阅读(307)  评论(0编辑  收藏  举报
levels of contents