Go ast.PreorderStack 怎样遍历 AST 并保留父节点栈

做 Go 源码分析时，最麻烦的往往不是“能不能遍历 AST”，而是访问一个调用节点时，怎样知道它属于哪个函数、位于哪一层语法结构。ast.PreorderStack 专门解决这个上下文问题：回调拿到当前节点 n 的同时，还能拿到从根节点到父节点的栈。

下面用一个小例子把源码解析、父节点识别、跳过子树和版本边界串起来。PreorderStack 在 Go 1.25 加入；如果项目仍需支持更早版本，应继续使用 ast.Inspect 或自行维护遍历栈。

先看 PreorderStack 到底多提供了什么

普通的 ast.Preorder 只告诉我们当前节点，适合按深度优先顺序读取所有节点。ast.PreorderStack 的回调签名多了一个 stack []ast.Node：它包含传入的原始栈，以及从根节点到当前节点父级的所有节点，但不包含当前节点本身。

这意味着回调里可以把当前节点和它的语境分开处理。例如当前节点是 *ast.CallExpr，就从栈尾向前寻找最近的 *ast.FuncDecl。如果回调返回 false，当前节点下面的子树不会继续访问；它不像 ast.Inspect 那样再调用一次“离开节点”的回调。

先把 Go 源码解析成 AST

PreorderStack 接收的是 ast.Node，所以第一步仍然是用 go/parser 把源码变成 *ast.File。解析失败时不要继续遍历，否则调用方无法区分输入不是完整 Go 源码，还是遍历结果为空。

package main

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

// nearestFunc 从父节点栈尾部寻找当前节点所属的最近函数。
func nearestFunc(stack []ast.Node) string {
	for i := len(stack) - 1; i >= 0; i-- {
		if fn, ok := stack[i].(*ast.FuncDecl); ok {
			return fn.Name.Name
		}
	}
	return ""
}

func main() {
	src := `package demo

func load() { read() }
func read() { fmt.Println("ok") }
`

	// FileSet 保存源码位置；ParseFile 返回可供 go/ast 遍历的根节点。
	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, "demo.go", src, parser.SkipObjectResolution)
	if err != nil {
		// 解析错误是输入边界，不能把不完整 AST 当成正常结果。
		fmt.Println("parse error:", err)
		return
	}

	// 回调中的 stack 是当前节点的父路径，不包含当前 n。
	ast.PreorderStack(f, nil, func(n ast.Node, stack []ast.Node) bool {
		call, ok := n.(*ast.CallExpr)
		if !ok {
			return true // 非调用节点继续访问它的子节点。
		}
		fmt.Printf("%s -> %s\n", nearestFunc(stack), callName(call))
		return true
	})
}

// callName 只处理最常见的标识符和选择器调用，复杂表达式留给调用方扩展。
func callName(call *ast.CallExpr) string {
	switch fun := call.Fun.(type) {
	case *ast.Ident:
		return fun.Name
	case *ast.SelectorExpr:
		return fun.Sel.Name
	default:
		return ""
	}
}

这里传入 nil 表示从空的外部栈开始。parser.SkipObjectResolution 足够支持语法树遍历；如果还要解析标识符绑定关系，再考虑类型检查，而不是把这篇文章的遍历任务混在一起。

沿栈找到调用属于哪个函数

回调访问 CallExpr 时，栈尾通常是更靠近当前节点的父级，可能依次经过文件、函数声明、函数体和表达式语句。不要假设固定下标，因为加一层括号、闭包或控制流节点都会改变路径长度。反向扫描并用类型断言找最近的 *ast.FuncDecl，比写死层级稳定得多。

上例中，read() 会归到 load，而 fmt.Println 会归到 read。这只是语法归属，不代表已经完成名称解析、方法分派或跨文件调用图构建；如果要回答“这个调用最终绑定到哪个包的函数”，还需要额外的类型信息。

用返回值控制子树，并处理边界

如果只关心顶层函数声明，可以在不需要深入的节点上返回 false。例如遇到某个生成代码区域或不需要分析的复合字面量时，可以直接跳过它的子树。要注意，false 只影响当前节点下面的访问，不会终止整棵 AST 的遍历。

还有三个边界值得写进调用方：

• 传入的 stack 只表达回调当前看到的路径；根节点回调时没有父级节点，不能把空栈当成错误。
• 回调要返回 true 才会继续访问当前节点的子树；非目标节点通常直接返回 true。
• PreorderStack 是 Go 1.25 新增 API。旧版本编译失败时，不要只改 import，应改用 ast.Inspect，并自己处理进入和离开节点的栈。

常见问题

stack 是否包含当前节点？

不包含。它包含从根到当前节点父级的路径，并保留调用时传入的原始栈。当前节点要单独使用回调参数 n。

能不能用 PreorderStack 直接生成完整调用图？

不能直接等同。它能提供语法层级和调用表达式归属；跨文件、接口方法和泛型实例化的绑定仍需类型检查或更完整的分析流程。

把任务收敛到“当前节点 + 父节点栈”后，ast.PreorderStack 很适合做轻量源码检查器：解析阶段负责输入边界，回调负责节点判断，父栈负责语境，返回值负责裁剪无关子树。