要使用go的go/ast库解析代码结构，首先加载和解析go文件，其次通过遍历ast找出函数定义或结构体字段信息。具体步骤如下：1. 使用token.newfileset()创建位置信息记录器，并用parser.parsefile解析源码；2. 利用ast.inspect深度优先遍历ast，判断节点类型以获取函数定义；3. 遍历声明提取结构体字段及其tag信息；需要注意ast结构复杂、位置信息需借助fset.position获取、不能直接修改源码以及包导入不会展开等问题。
用 Go 的 go/ast 库解析代码结构，其实是一个很实用但也容易让人卡住的操作。它能帮助你分析、重构甚至生成 Go 代码，但刚上手时面对 AST（抽象语法树）的结构可能会有点懵。
下面结合几个实际使用场景，讲讲怎么用 go/ast 来做语法树分析，以及一些需要注意的地方。
如何加载和解析一个 Go 文件
第一步是把 Go 源码读进来，然后用 parser.ParseFile 解析成 AST 节点。这个过程看起来简单，但有些细节容易忽略。
立即学习“go语言免费学习笔记（深入）”；
fset := token.NewFileSet()
file, err := parser.ParseFile(fset, "example.go", nil, parser.ParseComments)
if err != nil {
log.Fatal(err)
}几点说明：
-
token.FileSet是用来记录位置信息的，比如行号。 -
parser.ParseFile第三个参数可以传入源码内容，如果是nil就会自动从文件中读取。 - 如果需要处理注释，记得加上
parser.ParseComments标志。
拿到 *ast.File 后，就可以开始遍历整个文件结构了。
遍历 AST 结构：找出函数定义
很多时候我们想找出所有函数名或者某个特定的函数，这时候就需要遍历 AST。
常用的方式是使用 ast.Inspect 函数：
ast.Inspect(file, func(n ast.Node) bool {
if fn, ok := n.(*ast.FuncDecl); ok {
fmt.Println("Found function:", fn.Name.Name)
}
return true
})这里的关键点：
-
ast.Inspect会深度优先遍历整个 AST。 - 判断节点类型时要小心，不是每个节点都是你需要的类型。
- 返回值为
true表示继续深入遍历子节点，返回false则停止当前分支。
如果你想更精细地控制遍历方式，也可以实现 ast.Visitor 接口。
实战案例：提取结构体字段信息
假设你想写个工具，自动生成 JSON tag 或者数据库映射，那你就得从结构体中提取字段信息。
举个例子，结构体如下：
type User struct {
ID int `json:"id"`
Name string `json:"name"`
}你可以这样提取字段和 tag：
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
}
structType, ok := typeSpec.Type.(*ast.StructType)
if !ok {
continue
}
for _, field := range structType.Fields.List {
for _, name := range field.Names {
tag := ""
if field.Tag != nil {
tag = reflect.StructTag(strings.Trim(field.Tag.Value, "`"))
}
fmt.Printf("Field: %s, Tag: %s\n", name.Name, tag)
}
}
}
}这段代码虽然有点长，但逻辑很清晰：
- 先找类型声明（
GenDecl且 Tok 为 TYPE） - 再找结构体定义
- 最后遍历字段，提取名字和 tag
这种方式在写 ORM 工具或数据校验器时特别有用。
常见坑点和注意事项
- AST 结构复杂：刚开始看会觉得结构太多太杂，建议先打印出来看看结构再动手。
-
位置信息不直观：要用
fset.Position(node.Pos())才能拿到具体行号。 -
不能直接修改源码：AST 只是结构表示，要改源码还得配合
printer.Fprint输出回文本。 - 注意包导入问题：如果依赖外部包，在 AST 中不会展开，只保留引用。
基本上就这些，理解 AST 的结构之后，你会发现很多自动化任务都可以通过它来完成。
