Go反射实现动态修改对象

反射是 Go 语言中一项强大的特性，它允许我们在运行时检查和修改对象。借助反射，我们可以：迭代对象的字段并动态修改字段值。根据动态输入创建结构体，即使我们不知道输入的键名。在不修改源代码的情况下实现定制解析器和序列化器。

用 Go 反射动态修改对象

简介

反射是 Go 语言中一项强大的特性，它允许程序在运行时检查和修改数据结构。借助反射，我们可以在不修改源代码的情况下动态地操纵对象。这在实现自定义解析器、序列化器和其他需要根据输入动态调整行为的场景中非常有用。

代码

为了演示 Go 反射的强大功能，让我们创建一个名为 Person 的结构体并使用反射动态地更改它的字段：

package main

import (
    "fmt"
    "reflect"
)

type Person struct {
    Name string
    Age  int
}

func main() {
    // 创建一个 Person 结构体的实例
    p := Person{"John", 30}

    // 获取反射类型
    rt := reflect.TypeOf(p)

    // 迭代结构体字段
    for i := 0; i < rt.NumField(); i++ {
        // 获取字段反射值
        fv := reflect.ValueOf(p).Field(i)

        // 修改字段值
        switch fv.Kind() {
        case reflect.String:
            fv.SetString("Jane")
        case reflect.Int:
            fv.SetInt(35)
        }
    }

    // 打印修改后的结构体
    fmt.Println(p)
}

实战案例

让我们考虑一个实际的例子，其中我们需要动态地解析 JSON 输入，该输入具有不一致的键值。使用反射，我们可以动态地创建具有正确类型和值的结构体，即使我们对输入的键一无所知。

以下是执行此操作的代码示例：

package main

import (
    "encoding/json"
    "fmt"
    "reflect"
)

func main() {
    // 输入 JSON 字符串
    input := `{"name": "John", "age": 30}`

    // 解析 JSON 为 map[string]interface{}
    data := make(map[string]interface{})
    if err := json.Unmarshal([]byte(input), &data); err != nil {
        panic(err)
    }

    // 创建 Person 反射类型
    pType := reflect.TypeOf(Person{})

    // 创建 Person 反射值
    pValue := reflect.New(pType)

    // 迭代 JSON 对象的键值对
    for key, val := range data {
        // 获取结构体字段的反射值
        fv := pValue.Elem().FieldByName(key)

        // 检查字段是否存在
        if !fv.IsValid() {
            continue
        }

        // 根据字段类型设置字段值
        switch fv.Kind() {
        case reflect.String:
            fv.SetString(val.(string))
        case reflect.Int:
            fv.SetInt(int64(val.(float64)))
        }
    }

    // 打印修改后的结构体
    fmt.Println(pValue.Elem().Interface())
}

正如您所看到的，通过使用反射，我们可以灵活地操纵对象并根据动态输入创建所需的结构。