本文介绍使用 golang.org/x/tools/go/ast/astutil 包中的 cursor 机制，安全、精准地从 go 抽象语法树（ast）中删除指定类型节点（如 *ast.exprstmt），避免手动修改导致树结构损坏或 panic。
本文介绍使用 golang.org/x/tools/go/ast/astutil 包中的 cursor 机制，安全、精准地从 go 抽象语法树（ast）中删除指定类型节点（如 *ast.exprstmt），避免手动修改导致树结构损坏或 panic。
在 Go 的 AST 操作中，直接赋值为 nil 或替换为空节点（如 &ast.BasicLit{}）是不安全的——因为 AST 是由父节点持有子节点切片（如 file.Decls、func.Body.List）引用的，简单“清空”某个字段不会自动从父容器中移除该节点，反而可能引发运行时 panic 或生成非法语法树。
推荐方案是使用官方维护的工具包：golang.org/x/tools/go/ast/astutil。它提供了基于游标（astutil.Cursor）的安全遍历与编辑能力，其中 Cursor.Delete() 方法可原子性地从父节点的子节点切片中移除当前节点，并自动处理索引偏移，确保遍历过程稳定。
以下是一个完整示例：删除源文件中所有 *ast.ExprStmt（表达式语句，如 fmt.Println("hello")）：
package main
import (
"fmt"
"go/ast"
"go/parser"
"go/token"
"golang.org/x/tools/go/ast/astutil"
)
func deleteExprStmts(src string) (*ast.File, error) {
fset := token.NewFileSet()
file, err := parser.ParseFile(fset, "", src, 0)
if err != nil {
return nil, err
}
// 使用 astutil.Apply 遍历并编辑 AST
astutil.Apply(file, func(cursor *astutil.Cursor) bool {
// 检查当前节点是否为 *ast.ExprStmt
if exprStmt, ok := cursor.Node().(*ast.ExprStmt); ok {
fmt.Printf("Deleting expression statement: %s\n",
astutil.FormatNode(exprStmt.X, fset))
cursor.Delete() // 安全移除，自动更新父节点切片
}
return true
}, nil)
return file, nil
}
func main() {
src := `
package main
import "fmt"
func main() {
fmt.Println("before")
42 // 表达式语句（无副作用），将被删除
fmt.Println("after")
_ = 1 + 1 // 也是表达式语句，将被删除
}
`
file, err := deleteExprStmts(src)
if err != nil {
panic(err)
}
// 此处可调用 go/format 或 ast.Print 进一步验证结果
}✅ 关键要点：
- 必须使用 astutil.Apply（而非 ast.Inspect），因为后者只读，无法触发 cursor.Delete()；
- cursor.Delete() 仅在当前节点是其父节点切片中的直接元素时生效（如 func.Body.List[i]），对非切片字段（如 field.Type）无效；
- 删除后无需手动调整切片索引——astutil 内部已妥善处理；
- 若需替换而非删除（例如用 nil 占位），应改用 cursor.Replace(newNode)，但注意目标字段是否允许 nil（如 Stmt 切片中可为 nil，但 Expr 字段通常不可）。
最后提醒：操作 AST 后，务必通过 go/format.Node 或 go/printer 输出验证语法合法性，并结合 go/types 进行语义检查，确保重构未破坏程序行为。
