本文介绍使用 golang.org/x/tools/go/ast/astutil 包中的 Cursor 机制，对 Go 抽象语法树（AST）进行精准节点删除或替换，特别适用于移除表达式语句（ast.ExprStmt）等无副作用的冗余节点。
本文介绍使用 `golang.org/x/tools/go/ast/astutil` 包中的 `cursor` 机制，对 go 抽象语法树（ast）进行精准节点删除或替换，特别适用于移除表达式语句（*ast.exprstmt）等无副作用的冗余节点。
在 Go 的 AST 操作中，*直接修改 `ast.File或其子节点切片（如file.Decls、f.Body.List）是不安全的**——因为ast.Inspect是只读遍历，且 AST 节点间存在父子引用关系，手动删改切片易导致 panic 或结构损坏。正确方式是借助官方维护的astutil工具包，它提供了基于游标（astutil.Cursor`）的安全重写机制。
✅ 推荐方案：使用 astutil.Apply + Cursor.Delete
astutil.Apply 会深度遍历 AST，并在每个节点进入（Enter）和退出（Leave）时提供 astutil.Cursor，通过 cursor.Delete() 可安全移除当前节点（要求该节点位于可变切片中，如函数体语句列表、文件声明列表等）：
import (
"go/ast"
"go/parser"
"go/token"
"golang.org/x/tools/go/ast/astutil"
)
func removeExprStmts(fset *token.FileSet, file *ast.File) *ast.File {
astutil.Apply(file, nil, func(cursor *astutil.Cursor) bool {
// 检查当前节点是否为 *ast.ExprStmt，且父节点是 *ast.BlockStmt（即函数/方法体）
if exprStmt, ok := cursor.Node().(*ast.ExprStmt); ok {
if block, ok := cursor.Parent().(*ast.BlockStmt); ok {
// 确保该 ExprStmt 在 BlockStmt.List 中（典型场景）
for i, stmt := range block.List {
if stmt == exprStmt {
cursor.Delete() // 安全删除：自动从 block.List 中移除
return false // 不再向下遍历该节点子树（已删除）
}
}
}
}
return true
})
return file
}
// 使用示例
func main() {
fset := token.NewFileSet()
file, err := parser.ParseFile(fset, "example.go", `package main
func main() {
42 // 表达式语句（无副作用）
"hello" // 同上
println("ok")
}`, parser.ParseComments)
if err != nil {
panic(err)
}
modified := removeExprStmts(fset, file)
// 此时 modified 的 main 函数体中已移除孤立表达式语句
}⚠️ 注意事项
- cursor.Delete() 仅对“可编辑上下文”有效：即当前节点必须是父节点某个切片字段（如 *ast.BlockStmt.List、*ast.File.Decls）中的元素；若节点是单个字段（如 *ast.ReturnStmt.Results），则需用 cursor.Replace(...) 替换为 nil 或空切片。
- 删除后需确保语法合法性：例如删除函数体中所有语句会导致空 {}，这是合法的；但删除 if 条件表达式则会破坏结构，需谨慎判断。
- astutil.Apply 是不可逆操作，建议在修改前深拷贝 AST（可用 ast.Copy）或保留原始 *token.FileSet 以支持错误定位。
- 若需批量替换（如将 *ast.ExprStmt 替换为空语句 *ast.EmptyStmt），可调用 cursor.Replace(&ast.EmptyStmt{})。
总结
不要手动操作 AST 切片，而应统一使用 astutil.Apply 配合 Cursor.Delete / Cursor.Replace 实现安全、可维护的 AST 转换。该模式已被 gofmt、goimports 等工具广泛采用，是 Go 生态中处理 AST 修改的事实标准。
