如何通过结构体名字符串在运行时创建Go空结构体适配GORM AutoMigrate

要实现通过结构体名称字符串创建对应空实例并传入GORM的AutoMigrate，核心是利用Go的反射机制结合类型映射表——因为Go是静态类型语言，运行时无法直接通过字符串查找未注册的类型。以下是具体实现步骤：

1. 构建结构体名称到类型的映射表 #

首先需要把Protobuf生成的所有结构体类型提前注册到一个map中，key是结构体名称字符串，value是对应的reflect.Type。根据结构体数量多少，有两种维护方式：

手动维护示例（少量结构体） #

假设Protobuf生成的结构体在pb包下：

package yourpackage

import (
    "reflect"
    "your/project/path/pb"
)

// 结构体名称到类型的映射
var modelTypeMap = map[string]reflect.Type{
    "Screen": reflect.TypeOf(&pb.Screen{}),
    "User":   reflect.TypeOf(&pb.User{}),
    // 其他Protobuf结构体依次添加...
}

自动生成映射表（大量结构体） #

如果结构体数量多，手动维护太繁琐，可以用go generate结合AST解析自动生成映射表：

1. 写一个生成工具generate_model_map.go：

package main

import (
    "fmt"
    "go/ast"
    "go/parser"
    "go/token"
    "os"
)

func main() {
    fset := token.NewFileSet()
    // 替换为你的Protobuf生成代码所在的包路径
    pkgPath := "./pb"
    files, err := parser.ParseDir(fset, pkgPath, nil, parser.AllErrors)
    if err != nil {
        panic(fmt.Sprintf("parse pb package failed: %v", err))
    }

    // 生成pb包下的ModelTypeMap代码
    fmt.Println("package pb")
    fmt.Println()
    fmt.Println("import (")
    fmt.Println("\t\"reflect\"")
    fmt.Println(")")
    fmt.Println()
    fmt.Println("var ModelTypeMap = map[string]reflect.Type{")
    for _, pkg := range files {
        for _, file := range pkg.Files {
            for _, decl := range file.Decls {
                genDecl, ok := decl.(*ast.GenDecl)
                if !ok || genDecl.Tok != token.TYPE {
                    continue
                }
                for _, spec := range genDecl.Specs {
                    typeSpec, ok := spec.(*ast.TypeSpec)
                    if !ok {
                        continue
                    }
                    // 只处理结构体类型
                    if _, isStruct := typeSpec.Type.(*ast.StructType); isStruct {
                        fmt.Printf("\t\"%s\": reflect.TypeOf(&%s{}),\n", typeSpec.Name.Name, typeSpec.Name.Name)
                    }
                }
            }
        }
    }
    fmt.Println("}")
}

2. 在Protobuf生成的包文件中添加生成指令：
   在pb包下任意一个.go文件顶部添加：

//go:generate go run ../path/to/generate_model_map.go

3. 执行生成命令：
   在项目根目录运行go generate ./...，即可在pb包下自动生成包含所有结构体类型的ModelTypeMap。

2. 通过名称创建实例并执行迁移 #

拿到AST解析出的结构体名称列表后，从映射表中取出类型，用反射创建空实例指针，再传入AutoMigrate：

import (
    "log"
    "reflect"
    "your/project/path/pb"
    "gorm.io/gorm"
)

func MigrateAll(db *gorm.DB, structNames []string) error {
    var dst []interface{}
    for _, name := range structNames {
        // 从映射表获取类型
        t, ok := pb.ModelTypeMap[name]
        if !ok {
            log.Printf("skip unknown struct: %s", name)
            continue
        }
        // 创建空实例指针（比如&Screen{}）
        instance := reflect.New(t.Elem()).Interface()
        dst = append(dst, instance)
    }
    // 执行自动迁移
    return db.AutoMigrate(dst...)
}

这样就能批量把所有Protobuf生成的结构体迁移到数据库了。

内容的提问来源于stack exchange，提问作者yAnGwAwA