Golang ast怎么修改变量赋值语句-Golang-PHP中文网

用 ast.Inspect 定位赋值语句需区分 *ast.AssignStmt（对应 =、+= 等）和 *ast.ValueSpec（对应 :=），前者检查 stmt.Tok == token.ASSIGN，后者需在 *ast.GenDecl 中遍历 Specs 并确认 Values 非空；修改须用 astutil.Apply 替换节点，不可用 ast.Inspect 直接编辑。

golang ast怎么修改变量赋值语句

如何用 `ast.Inspect` 定位赋值语句

Go 的 AST 修改不是“就地编辑”，而是先遍历找到目标节点，再构造新节点替换。关键前提是准确识别变量赋值语句——它对应的是 *ast.AssignStmt，且 tok 字段为 token.ASSIGN（=）、token.ADD_ASSIGN（+=）等。

常见误判点：把 := 当作赋值语句。其实 := 是 *ast.AssignStmt 吗？不是——它是 *ast.DeclStmt 里的 *ast.GenDecl + *ast.ValueSpec，必须单独处理。

实操建议：

用 ast.Inspect 遍历时，优先检查 node 是否为 *ast.AssignStmt，再判断 stmt.Tok == token.ASSIGN
若需同时覆盖 := 场景，得额外进入 *ast.GenDecl，遍历 Specs 中的 *ast.ValueSpec，并确认 Values 非空
别依赖 ast.Print 输出肉眼判断——它不显示 Tok 值，容易漏掉 += 类型

用 `astutil.Apply` 替换赋值右值（`Rhs`）

ast.Inspect 只读，无法修改树；真正做替换得靠 astutil.Apply，它支持在遍历中返回新节点来替换旧节点。

典型需求：把所有 x = 42 改成 x = 100。重点在构造合法的新右值节点：

42 是 *ast.BasicLit，100 也要用 ast.NewIdent("100") ❌ 错！数字字面量必须用 &ast.BasicLit{Kind: token.INT, Value: "100"}
若右值是表达式（如 y + 1），不能直接改字符串，得用 ast.NewIdent("z") 或重构整个 ast.BinaryExpr
替换后务必调用 astutil.Apply 的 replace 函数返回新 *ast.AssignStmt，否则修改不生效

示例片段（仅示意结构）：

astutil.Apply(fset, file, nil, func(cursor *astutil.Cursor) bool {
    if stmt, ok := cursor.Node().(*ast.AssignStmt); ok && stmt.Tok == token.ASSIGN {
        if len(stmt.Lhs) == 1 && len(stmt.Rhs) == 1 {
            ident, ok := stmt.Lhs[0].(*ast.Ident)
            if ok && ident.Name == "x" {
                // 构造新右值：100
                newRhs := &ast.BasicLit{Kind: token.INT, Value: "100"}
                newStmt := &ast.AssignStmt{
                    Lhs: stmt.Lhs,
                    Tok: stmt.Tok,
                    Rhs: []ast.Expr{newRhs},
                }
                cursor.Replace(newStmt)
                return false // 阻止继续深入该子树
            }
        }
    }
    return true
})