一文初探Go语言中的reflect反射包

时间:2022-12-05 陈明勇人气:0

reflect 反射包

针对反射，Go 提供了 reflect 包，使用这个包里的函数可以在程序运行时获取和更新未知变量的值，操作未知变量的方法等。

reflect 包核心的两个重要类型：

reflect.Type：Type 是一个接口，不同数据类型有着不同的结构体实现。这个接口用于操作变量的类型信息，类型的信息只能读取。
reflect.Value：Value 是一个结构体，通过这个结构体可以操作变量的值。

TypeOf(i) 和 ValueOf(i)

reflect.TypeOf(i any) Type：获取变量的类型，返回一个 reflect.Type 类型。
reflect.ValueOf(i any) Value：获取变量的值，返回 reflect.Value 类型，通过 Value 可以对获取变量更多的信息。

案例1：获取变量的类别和类型信息

import (
    "fmt"
    "reflect"
)

type User struct {
    Name string
}

func main() {
    user := User{
            Name: "cmy",
    }
    func4Reflect(user)
}

func func4Reflect(data any) {
    typ := reflect.TypeOf(data)
    fmt.Println("类别：", typ.Kind()) // 类别： struct
    fmt.Println("类型：", typ.Name()) // 类型： User
}

通过 TypeOf() 函数获取 data 的类型信息，然后调用 Kind() 和 Name() 方法分别获取 data 变量的类别和类型信息。
根据返回结果可知， Kind() 返回的是 Go 的数据类型，而 Name() 返回的是我们自定义的数据类型。
根据 Kind() 返回值的特点，可以用于判断变量属于 Go 的哪种数据类型，用于类型限制等场景。

案例2：修改基本数据类型变量的值

import (
    "fmt"
    "reflect"
)

func main() {
    num1 := 666
    fmt.Println("num1 原值：", num1)
    func4Reflect(&num1)
    fmt.Println("num1 修改后的值：", num1)

    num2 := 0.5
    fmt.Println("num2 原值：", num2)
    func4Reflect(&num2)
    fmt.Println("num2 修改后的值：", num2)

    str := "go"
    fmt.Println("str 原值：", str)
    func4Reflect(&str)
    fmt.Println("str 修改后的值：", str)
}

func func4Reflect(data any) {
    typ := reflect.TypeOf(data)
    val := reflect.ValueOf(data)
    switch typ.Elem().Kind() {
    case reflect.Int:
            val.Elem().SetInt(888)
    case reflect.Float64:
            val.Elem().SetFloat(3.14)
    case reflect.String:
            val.Elem().SetString("Golang")
    }
}

通过 ValueOf() 函数获取 data 变量的值信息，然后结合 reflect.Type.Kind() 方法，对不同类型的变量的值进行修改操作（只举三种类型的例子）：

int 类型 → 使用 SetInt(val) 方法对值进行修改。
float64 → 使用 SetFloat(val) 方法对值进行修改。
string 类型 → 使用 SetString(val) 方法对值进行修改。

data 必须是指针类型，否则无法通过反射修改。

由于是指针类型，因此需要调用 Elem() 方法获取到指针指向的变量，才能修改变量的值。

案例3：通过反射获取结构体的字段名、字段类型和字段的值

import (
    "fmt"
    "reflect"
)

type User struct {
    Name string
    Age  int
}

func main() {
    user := User{
            Name: "cmy",
            Age:  18,
    }
    func4Reflect(user)
}

func func4Reflect(data any) {
    typ := reflect.TypeOf(data)
    val := reflect.ValueOf(data)
    // 获取结构体字段的数量
    numField := val.NumField()
    for i := 0; i < numField; i++ {
            fmt.Println("字段名称：", typ.Field(i).Name)
            fmt.Println("字段类型：", typ.Field(i).Type.Name())
            fmt.Println("字段值：", val.Field(i).Interface())
            fmt.Println("----------------------------")
    }
}

首先通过 TypeOf() 和 ValueOf() 获取到结构体的类型信息和值信息。
其次通过 Value.NumField() 方法获取到结构体字段的数量。
接着遍历结构体的字段，通过 Type.Field(i) 方法，传入索引，获取到对应字段的类型信息，通过 Name 属性获取字段名，Type.Name() 获取字段类型。
最后通过 Value.Field(i) 方法，传入索引，获取到对应字段的值信息，通过 Interface() 方法获取字段实际的值。

小结

本文首先介绍了 reflect 包里两个重要的类型 reflect.Type 和 reflect.Value，简单说明了它们的作用；其次介绍了TypeOf(i) 和 ValueOf(i) 两个函数；最后通过三个案例介绍了它们的使用场景。

加载全部内容