如何使用Golang实现ast节点过滤功能-Golang-PHP中文网

直接遍历 ast.node 会漏掉嵌套节点，因为 ast.inspect 默认在子节点返回 false 时终止当前分支，必须显式返回 true 才继续深入遍历。

如何使用golang实现ast节点过滤功能

为什么直接遍历 `ast.Node` 会漏掉嵌套节点

Go 的 ast 包里没有内置“深度过滤”函数，ast.Inspect 是唯一通用遍历方式，但它默认在子节点返回 false 时就终止当前分支——很多人误以为它像递归函数一样自动往下走，结果只拿到顶层节点。

必须显式返回 true 才继续进入子节点，否则遍历提前退出
ast.Walk 是纯访问器，不支持中途收集或中断，也不返回值，不适合“过滤”语义
自定义结构体 + ast.Visitor 接口实现容易因忘记 Visit 返回值逻辑而失效

用 `ast.Inspect` 实现带条件的节点收集

最轻量、最可控的方式是闭包捕获结果切片，配合布尔返回值控制遍历深度。比如只收集所有 ast.CallExpr：

var calls []ast.Node
ast.Inspect(fset.File, func(n ast.Node) bool {
    if call, ok := n.(*ast.CallExpr); ok {
        calls = append(calls, call)
    }
    // 继续遍历子树：返回 true；跳过子树：返回 false
    return true
})

注意 fset.File 是 *ast.File 类型，不是 ast.Node；传错会 panic
类型断言用 *ast.CallExpr 而非 ast.CallExpr，因为 AST 节点都是指针
若只想收集某一层（如仅函数体内的调用），可在进入 *ast.FuncDecl 时设标志位，出时重置

如何安全地排除特定节点类型（比如跳过注释和空白）

ast 树本身不包含注释节点（ast.CommentGroup）作为子节点，它们被挂在 ast.File.Comments 或各节点的 Doc/Comment 字段里。所以“过滤注释”不是遍历跳过，而是避免误读这些字段：

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

不要对 n.Doc 或 n.Comment 做 ast.Inspect —— 它们不是 ast.Node，而是 *ast.CommentGroup，其内部字段 List 是 []*ast.Comment，不参与 AST 遍历
如果目标是“忽略带特定注释的函数”，需先提取 funcDecl.Doc.Text()，再字符串匹配，而非试图在 AST 中过滤注释节点
空白由 lexer 处理，AST 层无对应节点，无需“过滤”

性能敏感场景下避免重复遍历的技巧

一次 ast.Inspect 只能做一件事（比如只找变量声明），但实际常需多条件并行判断。反复调用 Inspect 开销大，尤其处理大文件时：

把多个收集目标合并进一个闭包：用多个 if 分支分别 append 到不同切片
若需“排除某类节点后再处理其余”，别用两次遍历，改用布尔标记 + 单次遍历中跳过逻辑（例如：遇到 *ast.BadStmt 就设 skip = true，后续子节点检查该标记）
不要在 Inspect 回调里做 heavy 操作（如正则匹配、IO），它们会在每次节点访问时执行

AST 过滤真正的难点不在语法，而在理解哪些信息根本不在树上（比如注释位置、原始缩进、预处理宏）——这些得靠 token.FileSet 和源码字符串协同获取。