如何创建从文件读取的运行时自动生成结构体的实例

问题根源 #

你当前的StructExtract函数存在核心错误：通过reflect.New(reflect.TypeOf(structType)).Elem().Interface()创建的是AST语法树节点*ast.StructType的实例，而非你解析的业务结构体实例。Gorm Gen的FieldRelateModel需要的是目标结构体的运行时类型实例，AST节点完全不符合要求。

解决方案 #

Go的reflect包无法直接从AST生成运行时类型，必须将AST对应的结构体代码编译为可加载的代码。以下是通过生成临时Go文件、编译为插件并加载的实现方案：

修正后的StructExtract实现 #

package main

import (
	"bytes"
	"go/format"
	"os"
	"os/exec"
	"path/filepath"
	"plugin"
	"text/template"

	"go/ast"
	"go/parser"
	"go/token"
)

func StructExtract(filePath string) (any, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return nil, err
	}
	defer file.Close()

	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, filePath, file, parser.AllErrors)
	if err != nil {
		return nil, err
	}

	var typeSpec *ast.TypeSpec
	var structType *ast.StructType

	// 提取第一个结构体的TypeSpec
	for _, decl := range f.Decls {
		genDecl, ok := decl.(*ast.GenDecl)
		if !ok || genDecl.Tok != token.TYPE {
			continue
		}
		for _, spec := range genDecl.Specs {
			ts, ok := spec.(*ast.TypeSpec)
			if !ok {
				continue
			}
			st, ok := ts.Type.(*ast.StructType)
			if !ok {
				continue
			}
			typeSpec = ts
			structType = st
			break
		}
		if typeSpec != nil {
			break
		}
	}

	if typeSpec == nil {
		return nil, os.ErrNotExist
	}

	// 创建临时目录
	tmpDir, err := os.MkdirTemp("", "struct-gen")
	if err != nil {
		return nil, err
	}
	defer os.RemoveAll(tmpDir) // 自动清理临时文件

	// 结构体代码模板
	const codeTemplate = `
package main

type {{.StructName}} struct {
	{{- range .Fields}}
	{{.Name}} {{.Type}} {{.Tag}}
	{{- end}}
}

// 导出函数，返回结构体实例
func NewInstance() interface{} {
	return &{{.StructName}}{}
}
`

	// 解析结构体字段信息
	type FieldInfo struct {
		Name string
		Type string
		Tag  string
	}
	var fields []FieldInfo
	for _, field := range structType.Fields.List {
		fieldName := field.Names[0].Name
		fieldType := fset.Position(field.Type.Pos()).String()
		fieldTag := ""
		if field.Tag != nil {
			fieldTag = field.Tag.Value
		}
		fields = append(fields, FieldInfo{fieldName, fieldType, fieldTag})
	}

	// 渲染模板生成代码
	tmpl, err := template.New("struct").Parse(codeTemplate)
	if err != nil {
		return nil, err
	}
	var codeBuf bytes.Buffer
	err = tmpl.Execute(&codeBuf, map[string]interface{}{
		"StructName": typeSpec.Name.Name,
		"Fields":     fields,
	})
	if err != nil {
		return nil, err
	}

	// 格式化代码
	formattedCode, err := format.Source(codeBuf.Bytes())
	if err != nil {
		return nil, err
	}

	// 写入临时文件
	tmpFile := filepath.Join(tmpDir, "struct.go")
	err = os.WriteFile(tmpFile, formattedCode, 0644)
	if err != nil {
		return nil, err
	}

	// 编译为插件
	pluginPath := filepath.Join(tmpDir, "struct.so")
	cmd := exec.Command("go", "build", "-buildmode=plugin", "-o", pluginPath, tmpFile)
	cmd.Stderr = os.Stderr
	err = cmd.Run()
	if err != nil {
		return nil, err
	}

	// 加载插件并获取实例
	p, err := plugin.Open(pluginPath)
	if err != nil {
		return nil, err
	}
	newInstanceFunc, err := p.Lookup("NewInstance")
	if err != nil {
		return nil, err
	}
	instance := newInstanceFunc.(func() interface{})()
	return instance, nil
}

适配FieldRelateModel的使用方式 #

现在可以直接将StructExtract返回的实例传入FieldRelateModel：

relModel, err := StructExtract("path/to/your/generated/struct.go")
if err != nil {
	// 处理错误逻辑
}

// 传入FieldRelateModel使用
opt := FieldRelateModel(field.HasOne, "User", relModel, nil)

注意事项 #

• 编译插件时，Go版本、GOARCH/GOOS环境变量必须与主程序完全一致，否则会加载失败。
• 若生成的结构体依赖外部包，需确保编译时这些包能被正确解析。
• 临时目录会自动清理，但若程序异常退出可能残留文件，可根据需求添加额外清理逻辑。

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