Go AST：获取所有 struct

发布于07月13日

我希望能够获得所有 struct .例如，假设我们有:

type SomeType struct {
    // ..
}

type someType2 struct {
    //..
}

我们的代码.

structs := getAllStructs(srcPath)  //gets SomeType and someType2

我有一些代码可以找到所有内容.转到srcPath中的文件，并在其上执行parser.ParseFile.

有没有办法用ast、parser、packages等...我可以在任何范围内同时获得所有 struct ？如果有一个 struct 不是包范围的呢？如何在函数中也获得 struct 声明？例如:

func main() {
    
    type someType3 struct {
        //..
    }

}

推荐答案

Parse每个感兴趣的文件.Inspect查找 struct 类型的文件.

fset := token.NewFileSet()
f, err := parser.ParseFile(fset, fileName, nil, 0)
if err != nil {
    log.Fatal(err)
}

// Collect the struct types in this slice.
var structTypes []*ast.StructType

// Use the Inspect function to walk AST looking for struct
// type nodes.
ast.Inspect(f, func(n ast.Node) bool {
    if n, ok := n.(*ast.StructType); ok {
         structTypes = append(structTypes, n)
    }
    return true
})

// The slice structTypes contains all *ast.StructTypes in the file.