Golang go.ast 包详解

go/ast 包是Go语言的一个标准库，它提供了对Go语言的抽象语法树(AST)的访问。AST是源代码的内存表示，可以用来进行静态分析、代码生成、代码转换等。

以下是一些使用go/ast包的常见方法：

1. 解析源代码生成AST：

package main
 
import (
    "fmt"
    "go/ast"
    "go/parser"
    "go/token"
)
 
func main() {
    expr := `func add(x, y int) int { return x + y }`
    fset := token.NewFileSet()
    exprAst, err := parser.ParseExpr(fset, "", expr)
    if err != nil {
        panic(err)
    }
    ast.Print(fset, exprAst)
}

在这个例子中，我们使用go/parser包来解析一个字符串表达式，然后使用go/ast包的Print函数来打印这个表达式的AST。

2. 遍历AST：

package main
 
import (
    "go/ast"
    "go/token"
    "log"
    "strings"
)
 
func main() {
    expr := `func add(x, y int) int { return x + y }`
    fset := token.NewFileSet()
    exprAst, err := parser.ParseExpr(fset, "", expr)
    if err != nil {
        log.Fatal(err)
    }
 
    ast.Inspect(exprAst, func(n ast.Node) bool {
        if n == nil {
            return false
        }
        var name string
        switch n := n.(type) {
        case *ast.Ident:
            name = "Ident"
        case *ast.BasicLit:
            name = "BasicLit"
        case *ast.FuncLit:
            name = "FuncLit"
        default:
            name = "Other"
        }
        log.Printf("%s: %s\n", name, strings.ReplaceAll(fmt.Sprint(n), "\n", " "))
        return true
    })
}

在这个例子中，我们使用ast.Inspect函数来遍历AST，并打印出每个节点的类型和内容。

3. 修改AST：

package main
 
import (
    "go/ast"
    "go/token"
    "log"
)
 
func main() {
    expr := `func add(x, y int) int { return x + y }`
    fset := token.NewFileSet()
    exprAst, err := parser.ParseExpr(fset, "", expr)
    if err != nil {
        log.Fatal(err)
    }
 
    ast.Inspect(exprAst, func(n ast.Node) bool {
        if ident, ok := n.(*ast.Ident); ok && ident.Name == "x" {
            ident.Name = "a"
        }
        return true
    })
 
    ast.Inspect(exprAst, func(n ast.Node) bool {
        if call, ok := n.(*ast.CallExpr); ok {
            log.Printf("CallExpr: %s\n", ast.Print(call))
        }
        return true
    })
}

在这个例子中，我们使用ast.Inspect来找到所有的Ident节点，并将名字为"x"的改为"a"。然后我们又使用ast.Inspect来打印出所有的CallExpr节点，此时应该是将"x + y"改为"a + y"。

以上就是go/ast包的一些基本使用方法。它非常有