关于go泛型萃取简记

栏目: Go · 发布时间: 5年前

内容简介:最近由于代码需要,需要在一个map中存泛型,例如:然后通过关键字索引查询得到map对应key,大致方法是:当然,上面的pval传入的是指针,这样才能正确将值设置到pval指针中并返回给上层。

最近由于代码需要,需要在一个map中存泛型,例如:

var data map[string] interface{} = make(map[string]interface{})

然后通过关键字索引查询得到map对应key,大致方法是:

func Get(name string, pval interface{}) error

当然,上面的pval传入的是指针,这样才能正确将值设置到pval指针中并返回给上层。

在C++中,经常也有类似的需求,但是C++由于有比较清晰的继承,所以在C++代码中使用泛型可以很容易处理这种需求,在这里不再赘述。

所以在考虑如何优雅的实现上述接口函数,于是翻阅了下golang的reflect,使用映射,可以完美实现,这种方法可以推广到很多其他形式,来处理 go 的泛型化,使代码更加优雅。

实现及注释代码如下:

var m map[string]interface{}

func Get(name string, pval interface{}) error {
    //先判断是否传入的指针
    if reflect.TypeOf(pval).Kind() != reflect.Ptr {
        return errors.New("pval is not a ptr")
    }

    for k, v := range m {
        if k == name {
            //判断类型是否一致
            if reflect.TypeOf(pval).Elem() == reflect.TypeOf(v) {
                //获取Ptr元素值,并Set
                reflect.ValueOf(pval).Elem().Set(reflect.ValueOf(v))
                return nil
            } else {
                return errors.New("value type not equal")
            }
        }
    }
    return errors.New("not find the key" + name)
}

测试代码如下:

package main

import (
    "errors"
    "fmt"
    "reflect"
)

type Human struct {
    name string
    age  int32
}

var m map[string]interface{}

func Get(name string, pval interface{}) error {
    //先判断是否传入的指针
    if reflect.TypeOf(pval).Kind() != reflect.Ptr {
        return errors.New("pval is not a ptr")
    }

    for k, v := range m {
        if k == name {
            //判断类型是否一致
            if reflect.TypeOf(pval).Elem() == reflect.TypeOf(v) {
                //获取Ptr元素值,并Set
                reflect.ValueOf(pval).Elem().Set(reflect.ValueOf(v))
                return nil
            } else {
                return errors.New("value type not equal")
            }
        }
    }
    return errors.New("not find the key" + name)
}

func main() {
    m = make(map[string]interface{})
    m["aaa"] = "abc"
    m["bbb"] = 123
    m["xiaoming"] = Human{name: "xiaoming", age: 32}
    _ = m
    var val Human
    Get("xiaoming", &val)
    fmt.Println("val=", val)
}

打印结果如下:

Get xiaoming= {xiaoming 32}
value type not equal

第二行的错误,是因为类型不匹配,所以没有问题,虽然之前也用过reflect,但是有时候还是记不住,只是看文档时才知道原来这样,写下来比较好

github: https://github.com/jbl19860422/test_extract


以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

数据结构与算法经典问题解析

数据结构与算法经典问题解析

纳拉辛哈·卡鲁曼希 / 骆嘉伟 / 机械工业出版社 / 2016-6-1 / CNY 79.00

本书是一本数据结构方面的优秀教材,以Java为描述语言,介绍了计算机编程中使用的数据结构和算法。本书强调问题及其分析,而非理论阐述,共分为21章,讲述了基本概念、递归和回溯、链表、栈、队列、树、优先队列和堆、并查集DAT、图算法、排序、查找、选择算法(中位数)、符号表、散列、字符串算法、算法设计技术、贪婪算法、分治算法、动态规划算法、复杂度类型等内容。每章首先阐述必要的理论基础,然后给出问题集。全......一起来看看 《数据结构与算法经典问题解析》 这本书的介绍吧!

JSON 在线解析
JSON 在线解析

在线 JSON 格式化工具

MD5 加密
MD5 加密

MD5 加密工具

HEX CMYK 转换工具
HEX CMYK 转换工具

HEX CMYK 互转工具