2.08 Go之使用空接口实现保存任意值的字典

2.08 Go之使用空接口实现保存任意值的字典

需求描述

使用map和interface{}实现一个字典，可以存取任意类型的key和value

声明字典，设计设置值、获取值的函数

/* 声明一个字典结构 */
type Dictionary struct {
    // 设置一个map属性，key和value都是interface{}类型
    data map[interface{}]interface{}
}
​
/* 定义该类型的函数 */
func (d *Dictionary) Get(key interface{}) interface{} {
    return d.data[key]
}
​
func (d *Dictionary) Set(key interface{}, value interface{}) {
    d.data[key] = value 
}

遍历字段的所有键值函数

设置回调函数，回调返回需要遍历的数据。将回调的返回值设置为bool类型，外部逻辑在回调中不需要遍历时直接返回false即可终止遍历。

/* 设置回调函数，返回需要遍历的数据 */
// 定义回调，类型为 func(k,v interface{})bool，意思是返回键值数据（k、v）。bool 表示遍历流程控制，返回 true 时继续遍历，返回 false 时终止遍历。
func (d *Dictionary) Visit(callback func(k, v interface{}) bool) {
    // 回调函数返回值为空，直接返回
    if callback == nil {
        return
    }
​
    // 循环取值
    for k, v := range d.data {
        // 根据 callback 的返回值，决定是否继续遍历
        if !callback(k, v) {
            return
        }
    }
}

初始化和清除

字典结构包含有map，需要在创建Dictionary实例时初始化map。

NewDictionary中调用Clear()方法避免了map初始化过程的代码重复问题。

/* 清空数据函数 */
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("大锤", 80)
    dict.Set("小锤", 60)
    dict.Set("瞎几把锤", 24)
​
    // 获取值并打印
    favorite := dict.Get("小锤")
    fmt.Println("价格：", favorite)
​
    // 遍历所有字典元素
    dict.Visit(func(key, value interface{}) bool {
        // 转换值得类型
        if value.(int) > 50 {
            // 输出内容
            fmt.Println(key, "贵的一批!")
            return true
        }
​
        // 默认输出
        fmt.Println(key, "不是很贵!")
​
        return true
    })
}

完整的代码

package main
​
import "fmt"
​
/* 声明一个字典结构 */
type Dictionary struct {
    // 设置一个map属性，key和value都是interface{}类型
    data map[interface{}]interface{}
}
​
/* 定义该类型的函数 */
func (d *Dictionary) Get(key interface{}) interface{} {
    return d.data[key]
}
​
func (d *Dictionary) Set(key interface{}, value interface{}) {
    d.data[key] = value
}
​
/* 设置回调函数，返回需要遍历的数据 */
// 定义回调，类型为 func(k,v interface{})bool，意思是返回键值数据（k、v）。bool 表示遍历流程控制，返回 true 时继续遍历，返回 false 时终止遍历。
func (d *Dictionary) Visit(callback func(k, v interface{}) bool) {
    // 回调函数返回值为空，直接返回
    if callback == nil {
        return
    }
​
    // 循环取值
    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("大锤", 80)
    dict.Set("小锤", 60)
    dict.Set("瞎几把锤", 24)
​
    // 获取值并打印
    favorite := dict.Get("小锤")
    fmt.Println("价格：", favorite)
​
    // 遍历所有字典元素
    dict.Visit(func(key, value interface{}) bool {
        // 转换值得类型
        /*
        遍历返回的 key 和 value 的类型都是 interface{}，这里确认 value 只有 int 类型，所以将 value 转换为 int 类型判断是否大于 40
        */
        if value.(int) > 50 {
            // 输出内容
            fmt.Println(key, "贵的一批!")
            return true
        }
​
        // 默认输出
        fmt.Println(key, "不是很贵!")
​
        return true
    })
}