通过读取 a.go 文件反映结构类型

Question

我正在发现生成器（go generate），并且正在尝试为我的结构生成验证 function。

这个想法是我不希望我的程序在运行时使用反射，我宁愿让生成器使用反射来生成我想要使用的实际方法。

问题是我无法在生成器代码中导入我的结构，到目前为止我发现的唯一方法是从生成器中读取 .go 文件并使用正则表达式手动解析那里定义的类型

我有类似的东西

型号/型号.go：

package models

//go:generate go run ../generator.go -file models.go

type MyStruct struct {
    ...
}

发电机.go：

package main

func main() {
    f, err := ioutil.ReadFile(fileName) // I read filename from the flag provided
    ...
    // I parse f to generate my stuff
}

我非常希望有一个自省 package 将 go 代码作为字符串并给我一些关于那里定义的结构的信息

或者也许有一种方法可以导入调用 go:generate 的文件以直接访问类型

Answer 1

无需指定文件名，此代码也是如此：

//go:generate go run ../generator.go -file $GOFILE

在文本/模板包的帮助下，您无需解析文件。 一个非常简单的例子就是这样的。 这将为您提供线索：

package main

import (
    "flag"
    "os"
    "text/template"
)

//go:generate go run main.go -name=A
//go:generate go run main.go -name=B
//go:generate go run main.go -name=C

var name = flag.String("name", "test", "name of struct")

var code = `
package main

type Struct{{.}} struct {}

func (s *Struct{{.}} ) Vailadte() bool {
return true
}
`

func main() {
    flag.Parse()
    file, _ := os.Create(*name + ".go")
    defer file.Close()

    tmpl, _ := template.New("test").Parse(code)
    tmpl.Execute(file, *name)
}

Answer 2

也许你可以在你的生成器中使用go/parser和go/ast 。