Go语言ast如何重写_Go语言源代码自动修改【极客】-Golang-PHP中文网

必须用ast.walk（配合自定义ast.visitor），不能只靠ast.inspect；因inspect仅只读遍历，节点修改无效，而walk可通过visit返回新节点实现安全替换，且需手动维护父节点引用并确保语法合法性。

go语言ast如何重写_go语言源代码自动修改【极客】

用 `ast.Inspect` 还是 `ast.Walk`？选错就改不动节点

直接结论：必须用 ast.Walk（配合自定义 ast.Visitor），不能只靠 ast.Inspect。因为 ast.Inspect 是只读遍历，它传入的节点指针是 AST 结构的副本视图，原地修改 node.XXX 字段完全无效——Go 的 AST 节点是不可变结构体，且父节点对子节点的引用不会自动更新。

真正能安全替换节点的方式，是在 Visit 方法中返回替换后的节点：

func (v *rewriter) Visit(node ast.Node) ast.Node {
    if expr, ok := node.(*ast.BasicLit); ok && expr.Kind == token.STRING {
        return &ast.BasicLit{
            Kind:  token.STRING,
            Value: `"rewritten"`,
        }
    }
    return node // 不匹配则透传
}

必须返回新节点才能生效；返回 nil 表示删除该节点（注意：仅对子节点列表中的位置有效）
若想插入多个节点（比如在函数体开头加日志），需用 *ast.BlockStmt 的 List 字段手动拼接切片，再整体替换
ast.Inspect 只适合调试打印或收集信息，别指望它改代码

修改后怎么写回源文件？`gofmt` 和 `go/format` 别混用

AST 修改完只是内存里的结构，不落地等于没改。关键点在于：不能直接用 fmt.Print 输出 AST 节点，那会输出调试格式（含字段名、大括号），不是合法 Go 代码。

正确流程是：printer.Fprint → go/format.Source → 写文件：

fset := token.NewFileSet()
astFile := parseFile(fset, "main.go") // 你已修改好的 *ast.File

var buf bytes.Buffer
err := printer.Fprint(&buf, fset, astFile)
if err != nil { panic(err) }

// 格式化为标准 gofmt 风格（修复缩进、空行等）
formatted, err := format.Source(buf.Bytes())
if err != nil { panic(err) }

os.WriteFile("main.go", formatted, 0644)

printer.Fprint 输出的是“原始 AST 映射”，可能缺空格、换行错乱，必须过 go/format.Source
不要自己调 exec.Command("gofmt", "-w") —— 多余且跨平台行为不一致，go/format 就是标准库提供的等价能力
如果源文件含非 UTF-8 BOM 或混合换行符（CRLF/LF），format.Source 可能报 invalid character，需提前 Normalize

重命名变量或函数时，为什么 `ast.Ident` 改了但没生效？

常见错误：遍历到 *ast.Ident 就直接改 ident.Name = "newName"，结果编译失败——符号引用没同步更新，或类型检查崩了。

Go语言（Golang）1.26.0

Go语言（Golang）1.26.0版本官方下载，版本号 1.26.0，适合旧项目维护、兼容性测试和指定版本开发环境搭建。

下载

根本原因：Go 的 AST 不维护作用域和绑定关系。ast.Ident 本身只是个名字字符串，它是否代表同一个变量，取决于它在作用域树中的位置和声明点。单纯改 Name，无法保证所有引用都指向新名字，更不会更新 *ast.Object。

如需安全重命名，必须配合 types.Info（通过 types.NewChecker 类型检查获取），定位所有绑定到同一对象的 Ident，再批量替换
简单脚本场景（如只改某个函数内局部变量），可用“先找声明语句（*ast.AssignStmt 或 *ast.TypeSpec），再扫同层 ast.Ident”做启发式匹配，但不保准
别碰 import 别名和方法接收者名——它们出现在 AST 中的位置特殊，容易漏改

为什么跑完重写，`go build` 报 “undefined: xxx”？

大概率是节点替换破坏了语法结构合法性。AST 重写不是字符串替换，每个节点类型有严格字段约束。例如：

把 *ast.CallExpr 替换成 *ast.BasicLit，但该位置语法要求必须是表达式（ok），可如果它原本是语句（如单独一行的函数调用），就得包一层 *ast.ExprStmt
删掉 return 语句后，函数末尾没返回值，但签名声明了 int，就会触发编译错误
修改 if 的条件表达式时，忘了同步更新其 Init 或 Post 字段（如果是 *ast.ForStmt），导致结构断裂