动态获取结构体所有字段的指针

golang学习网今天将给大家带来《动态获取结构体所有字段的指针》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

我正在尝试为 golang 构建一个简单的 orm 层。 这将采用一个结构体并生成 cols [] ，然后可以将其传递给 sql 函数 rows.scan(cols...) 获取结构中与结果集中找到的每一列相对应的字段的指针

这是我的示例结构

type examplestruct struct {
    id        int64          `sql:"id"`
    aid    string         `sql:"a_id"`
    userid    int64          `sql:"user_id"`

这是我的通用 orm 函数

func getsqlcolumntofieldmap(model *examplestruct) map[string]interface{} {
    
    typeofmodel := reflect.typeof(*model)
    valueofmodel := reflect.valueof(*model)
    columntodatapointermap := make(map[string]interface{})
    for i := 0; i < valueofmodel.numfield(); i++ {
        sql_column := typeofmodel.field(i).tag.get("sql")
        structvalue := valueofmodel.field(i)
        columntodatapointermap[sql_column] = structvalue.addr()
    }
    return columntodatapointermap
}

一旦这个方法工作正常，我就可以使用它生成的映射根据我在 rows() 对象中获得的column_names 创建一个有序的 sql 指针列表 但是我在 .addr() 方法调用上遇到以下错误

panic: reflect.Value.Addr of unaddressable value [recovered]
    panic: reflect.Value.Addr of unaddressable value

这不可能吗？ 另外，在理想的情况下，我希望该方法采用接口而不是 *examplestruct，以便它可以在不同的数据库模型之间重用。

正确答案

该错误表示您要获取的地址的值无法寻址。这是因为，即使您将指针传递给 getsqlcolumntofieldmap()，您也会立即取消引用它并稍后使用非指针值。

当传递给 reflect.valueof() 时，该值被包装在 interface{} 中，并且包装在接口中的值不可寻址。

您不得取消引用指针，而应使用 type.elem() 和 value.elem() 来获取元素类型和指向的值。

类似这样的事情：

func getsqlcolumntofieldmap(model *examplestruct) map[string]interface{} {
    t := reflect.typeof(model).elem()
    v := reflect.valueof(model).elem()
    columntodatapointermap := make(map[string]interface{})
    for i := 0; i < v.numfield(); i++ {
        sql_column := t.field(i).tag.get("sql")
        structvalue := v.field(i)
        columntodatapointermap[sql_column] = structvalue.addr()
    }
    return columntodatapointermap
}

通过这个简单的改变就可以了！并且它不依赖于参数类型，您可以将其更改为 interface{} 并传递任何结构体指针。

func getsqlcolumntofieldmap(model interface{}) map[string]interface{} {
    // ...
}

测试它：

type examplestruct struct {
    id     int64  `sql:"id"`
    aid    string `sql:"a_id"`
    userid int64  `sql:"user_id"`
}

type point struct {
    x int `sql:"x"`
    y int `sql:"y"`
}

func main() {
    fmt.println(getsqlcolumntofieldmap(&examplestruct{}))
    fmt.println(getsqlcolumntofieldmap(&point{}))
}

输出（在 Go Playground 上尝试）：

map[a_id:<*string value> id:<*int64 value> user_id:<*int64 value>]
map[x:<*int value> y:<*int value>]

请注意，Value.Addr() 返回包装在 reflect.value 中的地址。要“展开”指针，请使用 Value.Interface()：

func getsqlcolumntofieldmap(model interface{}) map[string]interface{} {
    t := reflect.typeof(model).elem()
    v := reflect.valueof(model).elem()
    m := make(map[string]interface{})
    for i := 0; i < v.numfield(); i++ {
        colname := t.field(i).tag.get("sql")
        field := v.field(i)
        m[colname] = field.addr().interface()
    }
    return m
}

这将输出（在 Go Playground 上尝试）：

map[a_id:0xc00007e008 id:0xc00007e000 user_id:0xc00007e018]
map[x:0xc000018060 y:0xc000018068]

有关反射的深入介绍，请阅读博文：The Laws of Reflection

本篇关于《动态获取结构体所有字段的指针》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！