Go语言中通过字符串名称获取reflect.Type的策略与实践

package main

import (
    "fmt"
    "reflect"
)

// 定义一些示例结构体
type User struct {
    ID   int
    Name string
}

type Product struct {
    SKU   string
    Price float64
}

// typeRegistry 是一个全局的类型注册表，将类型名称映射到 reflect.Type
var typeRegistry = make(map[string]reflect.Type)

// RegisterType 用于注册类型。
// 传入一个类型的零值实例或指针实例，其类型信息将被注册。
func RegisterType(obj interface{}) {
    t := reflect.TypeOf(obj)
    // 使用 t.String() 作为键，它会返回完整的类型名称，如 "main.User" 或 "*main.User"
    typeRegistry[t.String()] = t
    fmt.Printf("Registered type: %s\n", t.String())
}

// GetTypeByName 根据完整的类型名称字符串获取对应的 reflect.Type。
// 如果找到，返回 reflect.Type 和 true；否则返回 nil 和 false。
func GetTypeByName(name string) (reflect.Type, bool) {
    t, ok := typeRegistry[name]
    return t, ok
}

// init 函数在包被导入时自动执行，用于初始化类型注册表。
func init() {
    // 注册需要通过名称查找的类型。
    // 通常注册值类型，如果需要创建指针实例，可以通过 reflect.New(type.Elem())。
    RegisterType(User{})
    RegisterType(Product{})
    // 如果需要通过指针类型名查找，也可以注册指针类型
    RegisterType(&User{})
}

func main() {
    fmt.Println("--- 查找并使用值类型 ---")
    // 尝试通过名称获取类型
    typeName1 := "main.User" // 注意：这里需要完整的包名 + 类型名
    if userType, ok := GetTypeByName(typeName1); ok {
        fmt.Printf("Found type '%s': %v\n", typeName1, userType)
        // 使用反射创建 User 类型的实例。
        // reflect.New(userType) 返回一个指向 User 零值的 reflect.Value (Kind() == Ptr)。
        // .Elem() 获取其指向的元素（即 User 值本身），再通过 .Interface().(User) 转换为 User 类型。
        newUser := reflect.New(userType).Elem().Interface().(User)
        newUser.ID = 1
        newUser.Name = "Alice"
        fmt.Printf("Created user: %+v\n", newUser)
    } else {
        fmt.Printf("Type '%s' not found.\n", typeName1)
    }

    fmt.Println("\n--- 查找并使用指针类型 ---")
    typeName2 := "main.Product"
    if productType, ok := GetTypeByName(typeName2); ok {
        fmt.Printf("Found type '%s': %v\n", typeName2, productType)
        // 使用反射创建 Product 类型的指针实例。
        // reflect.New(productType) 直接返回一个指向 Product 零值的 reflect.Value (Kind() == Ptr)。
        // .Interface().(*Product) 将其转换为 *Product 类型。
        newProductPtr := reflect.New(productType).Interface().(*Product)
        newProductPtr.SKU = "P001"
        newProductPtr.Price = 99.99
        fmt.Printf("Created product (ptr): %+v\n", newProductPtr)
    } else {
        fmt.Printf("Type '%s' not found.\n", typeName2)
    }

    fmt.Println("\n--- 查找不存在的类型 ---")
    typeName3 := "main.NonExistentType"
    if _, ok := GetTypeByName(typeName3); !ok {
        fmt.Printf("Type '%s' not found, as expected.\n", typeName3)
    }

    fmt.Println("\n--- 通过指针类型名查找 ---")
    typeName4 := "*main.User" // 查找注册的指针类型
    if userPtrType, ok := GetTypeByName(typeName4); ok {
        fmt.Printf("Found type '%s': %v\n", typeName4, userPtrType)
        // 如果注册的是指针类型 (*User)，而我们想创建 *User 实例，
        // 同样可以使用 reflect.New(userPtrType.Elem()) 来获取其元素类型（User），
        // 然后创建 User 的指针实例。
        newUserPtr := reflect.New(userPtrType.Elem()).Interface().(*User)
        newUserPtr.ID = 2
        newUserPtr.Name = "Bob"
        fmt.Printf("Created user (ptr from ptr type): %+v\n", newUserPtr)
    }
}